Merge pull request #75 from qiaozhanwei/dev-20190415 (897400e4) · Commits · SUMMER2020 / students / proj-2002251

escheduler-api/src/main/java/cn/escheduler/api/controller/AccessTokenController.java

0 → 100644

+169 −0

Original line number	Diff line number	Diff line
		/*
		* Licensed to the Apache Software Foundation (ASF) under one or more
		* contributor license agreements. See the NOTICE file distributed with
		* this work for additional information regarding copyright ownership.
		* The ASF licenses this file to You under the Apache License, Version 2.0
		* (the "License"); you may not use this file except in compliance with
		* the License. You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*/
		package cn.escheduler.api.controller;


		import cn.escheduler.api.enums.Status;
		import cn.escheduler.api.service.AccessTokenService;
		import cn.escheduler.api.service.UsersService;
		import cn.escheduler.api.utils.Constants;
		import cn.escheduler.api.utils.Result;
		import cn.escheduler.dao.model.User;
		import org.slf4j.Logger;
		import org.slf4j.LoggerFactory;
		import org.springframework.beans.factory.annotation.Autowired;
		import org.springframework.http.HttpStatus;
		import org.springframework.web.bind.annotation.*;

		import java.util.Map;

		import static cn.escheduler.api.enums.Status.*;


		/**
		* user controller
		*/
		@RestController
		@RequestMapping("/access-token")
		public class AccessTokenController extends BaseController{


		private static final Logger logger = LoggerFactory.getLogger(AccessTokenController.class);


		@Autowired
		private AccessTokenService accessTokenService;

		/**
		* create token
		* @param loginUser
		* @return
		*/
		@PostMapping(value = "/create")
		@ResponseStatus(HttpStatus.CREATED)
		public Result createToken(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam(value = "userId") int userId,
		@RequestParam(value = "expireTime") String expireTime,
		@RequestParam(value = "token") String token){
		logger.info("login user {}, create token , userId : {} , token expire time : {} , token : {}", loginUser.getUserName(),
		userId,expireTime,token);

		try {
		Map<String, Object> result = accessTokenService.createToken(userId, expireTime, token);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(CREATE_ACCESS_TOKEN_ERROR.getMsg(),e);
		return error(CREATE_ACCESS_TOKEN_ERROR.getCode(), CREATE_ACCESS_TOKEN_ERROR.getMsg());
		}
		}

		/**
		* create token
		* @param loginUser
		* @return
		*/
		@PostMapping(value = "/generate")
		@ResponseStatus(HttpStatus.CREATED)
		public Result generateToken(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam(value = "userId") int userId,
		@RequestParam(value = "expireTime") String expireTime){
		logger.info("login user {}, generate token , userId : {} , token expire time : {}",loginUser,userId,expireTime);
		try {
		Map<String, Object> result = accessTokenService.generateToken(userId, expireTime);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(GENERATE_TOKEN_ERROR.getMsg(),e);
		return error(GENERATE_TOKEN_ERROR.getCode(), GENERATE_TOKEN_ERROR.getMsg());
		}
		}

		/**
		* query access token list paging
		*
		* @param loginUser
		* @param pageNo
		* @param searchVal
		* @param pageSize
		* @return
		*/
		@GetMapping(value="/list-paging")
		@ResponseStatus(HttpStatus.OK)
		public Result queryAccessTokenList(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam("pageNo") Integer pageNo,
		@RequestParam(value = "searchVal", required = false) String searchVal,
		@RequestParam("pageSize") Integer pageSize){
		logger.info("login user {}, list access token paging, pageNo: {}, searchVal: {}, pageSize: {}",
		loginUser.getUserName(),pageNo,searchVal,pageSize);
		try{
		Map<String, Object> result = checkPageParams(pageNo, pageSize);
		if(result.get(Constants.STATUS) != Status.SUCCESS){
		return returnDataListPaging(result);
		}
		result = accessTokenService.queryAccessTokenList(loginUser, searchVal, pageNo, pageSize);
		return returnDataListPaging(result);
		}catch (Exception e){
		logger.error(QUERY_ACCESSTOKEN_LIST_PAGING_ERROR.getMsg(),e);
		return error(QUERY_ACCESSTOKEN_LIST_PAGING_ERROR.getCode(),QUERY_ACCESSTOKEN_LIST_PAGING_ERROR.getMsg());
		}
		}

		/**
		* delete access token by id
		* @param loginUser
		* @param id
		* @return
		*/
		@PostMapping(value = "/delete")
		@ResponseStatus(HttpStatus.OK)
		public Result delAccessTokenById(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam(value = "id") int id) {
		logger.info("login user {}, delete access token, id: {},", loginUser.getUserName(), id);
		try {
		Map<String, Object> result = accessTokenService.delAccessTokenById(loginUser, id);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(DELETE_USER_BY_ID_ERROR.getMsg(),e);
		return error(Status.DELETE_USER_BY_ID_ERROR.getCode(), Status.DELETE_USER_BY_ID_ERROR.getMsg());
		}
		}


		/**
		* update token
		* @param loginUser
		* @return
		*/
		@PostMapping(value = "/update")
		@ResponseStatus(HttpStatus.CREATED)
		public Result updateToken(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam(value = "id") int id,
		@RequestParam(value = "userId") int userId,
		@RequestParam(value = "expireTime") String expireTime,
		@RequestParam(value = "token") String token){
		logger.info("login user {}, update token , userId : {} , token expire time : {} , token : {}", loginUser.getUserName(),
		userId,expireTime,token);

		try {
		Map<String, Object> result = accessTokenService.updateToken(id,userId, expireTime, token);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(CREATE_ACCESS_TOKEN_ERROR.getMsg(),e);
		return error(CREATE_ACCESS_TOKEN_ERROR.getCode(), CREATE_ACCESS_TOKEN_ERROR.getMsg());
		}
		}

		}

escheduler-api/src/main/java/cn/escheduler/api/controller/SchedulerController.java

+0 −1

Original line number	Diff line number	Diff line
		@@ -46,7 +46,6 @@ public class SchedulerController extends BaseController{
		private static final Logger logger = LoggerFactory.getLogger(SchedulerController.class);
		public static final String DEFAULT_WARNING_TYPE = "NONE";
		public static final String DEFAULT_NOTIFY_GROUP_ID = "1";
		public static final String DEFAULT_MAX_TRY_TIMES = "0";
		public static final String DEFAULT_FAILURE_POLICY = "CONTINUE";

escheduler-api/src/main/java/cn/escheduler/api/controller/TaskRecordController.java

+33 −1

Original line number	Diff line number	Diff line
		@@ -68,7 +68,39 @@ public class TaskRecordController extends BaseController{
		try{
		logger.info("query task record list, task name:{}, state :{}, taskDate: {}, start:{}, end:{}",
		taskName, state, taskDate, startTime, endTime);
		Map<String, Object> result = taskRecordService.queryTaskRecordListPaging(taskName, startTime, taskDate, sourceTable, destTable, endTime,state, pageNo, pageSize);
		Map<String, Object> result = taskRecordService.queryTaskRecordListPaging(false, taskName, startTime, taskDate, sourceTable, destTable, endTime,state, pageNo, pageSize);
		return returnDataListPaging(result);
		}catch (Exception e){
		logger.error(QUERY_TASK_RECORD_LIST_PAGING_ERROR.getMsg(),e);
		return error(QUERY_TASK_RECORD_LIST_PAGING_ERROR.getCode(), QUERY_TASK_RECORD_LIST_PAGING_ERROR.getMsg());
		}

		}

		/**
		* query history task record list paging
		*
		* @param loginUser
		* @return
		*/
		@GetMapping("/history-list-paging")
		@ResponseStatus(HttpStatus.OK)
		public Result queryHistoryTaskRecordListPaging(@RequestAttribute(value = Constants.SESSION_USER) User loginUser,
		@RequestParam(value = "taskName", required = false) String taskName,
		@RequestParam(value = "state", required = false) String state,
		@RequestParam(value = "sourceTable", required = false) String sourceTable,
		@RequestParam(value = "destTable", required = false) String destTable,
		@RequestParam(value = "taskDate", required = false) String taskDate,
		@RequestParam(value = "startDate", required = false) String startTime,
		@RequestParam(value = "endDate", required = false) String endTime,
		@RequestParam("pageNo") Integer pageNo,
		@RequestParam("pageSize") Integer pageSize
		){

		try{
		logger.info("query hisotry task record list, task name:{}, state :{}, taskDate: {}, start:{}, end:{}",
		taskName, state, taskDate, startTime, endTime);
		Map<String, Object> result = taskRecordService.queryTaskRecordListPaging(true, taskName, startTime, taskDate, sourceTable, destTable, endTime,state, pageNo, pageSize);
		return returnDataListPaging(result);
		}catch (Exception e){
		logger.error(QUERY_TASK_RECORD_LIST_PAGING_ERROR.getMsg(),e);

escheduler-api/src/main/java/cn/escheduler/api/controller/UsersController.java

+8 −6

Original line number	Diff line number	Diff line
		@@ -64,13 +64,14 @@ public class UsersController extends BaseController{
		@RequestParam(value = "userName") String userName,
		@RequestParam(value = "userPassword") String userPassword,
		@RequestParam(value = "tenantId") int tenantId,
		@RequestParam(value = "queue") String queue,
		@RequestParam(value = "email") String email,
		@RequestParam(value = "phone", required = false) String phone) {
		logger.info("login user {}, create user, userName: {}, email: {}, tenantId: {}, userPassword: {}, phone: {}, proxyUsers: {}",
		loginUser.getUserName(), userName, email, tenantId, Constants.PASSWORD_DEFAULT, phone);
		logger.info("login user {}, create user, userName: {}, email: {}, tenantId: {}, userPassword: {}, phone: {}, user queue: {}",
		loginUser.getUserName(), userName, email, tenantId, Constants.PASSWORD_DEFAULT, phone,queue);

		try {
		Map<String, Object> result = usersService.createUser(loginUser, userName, userPassword, email, tenantId, phone);
		Map<String, Object> result = usersService.createUser(loginUser, userName, userPassword,email,tenantId, phone,queue);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(CREATE_USER_ERROR.getMsg(),e);
		@@ -127,13 +128,14 @@ public class UsersController extends BaseController{
		@RequestParam(value = "id") int id,
		@RequestParam(value = "userName") String userName,
		@RequestParam(value = "userPassword") String userPassword,
		@RequestParam(value = "queue") String queue,
		@RequestParam(value = "email") String email,
		@RequestParam(value = "tenantId") int tenantId,
		@RequestParam(value = "phone", required = false) String phone) {
		logger.info("login user {}, updateProcessInstance user, userName: {}, email: {}, tenantId: {}, userPassword: {}, phone: {}, proxyUsers: {}",
		loginUser.getUserName(), userName, email, tenantId, Constants.PASSWORD_DEFAULT, phone);
		logger.info("login user {}, updateProcessInstance user, userName: {}, email: {}, tenantId: {}, userPassword: {}, phone: {}, user queue: {}",
		loginUser.getUserName(), userName, email, tenantId, Constants.PASSWORD_DEFAULT, phone,queue);
		try {
		Map<String, Object> result = usersService.updateUser(id,userName,userPassword,email,tenantId,phone);
		Map<String, Object> result = usersService.updateUser(id,userName,userPassword,email,tenantId,phone,queue);
		return returnDataList(result);
		}catch (Exception e){
		logger.error(UPDATE_USER_ERROR.getMsg(),e);

escheduler-api/src/main/java/cn/escheduler/api/enums/Status.java

+6 −0

Original line number	Diff line number	Diff line
		@@ -199,6 +199,12 @@ public enum Status {


		HDFS_NOT_STARTUP(60001,"hdfs not startup"),


		CREATE_ACCESS_TOKEN_ERROR(70001,"create access token error"),
		GENERATE_TOKEN_ERROR(70002,"generate token error"),
		QUERY_ACCESSTOKEN_LIST_PAGING_ERROR(70003,"query access token list paging error"),

		;

		private int code;