xuxueli
5 years ago
parent
9767174c12
commit
2ced601035
@ -1,34 +1,97 @@
|
|||||||
package com.xxl.job.admin.controller;
|
package com.xxl.job.admin.controller;
|
||||||
|
|
||||||
import com.xxl.job.admin.controller.annotation.PermissionLimit;
|
import com.xxl.job.admin.controller.annotation.PermissionLimit;
|
||||||
import com.xxl.job.admin.core.scheduler.XxlJobScheduler;
|
import com.xxl.job.admin.core.conf.XxlJobAdminConfig;
|
||||||
import com.xxl.job.core.biz.AdminBiz;
|
import com.xxl.job.core.biz.AdminBiz;
|
||||||
import org.springframework.beans.factory.InitializingBean;
|
import com.xxl.job.core.biz.model.HandleCallbackParam;
|
||||||
|
import com.xxl.job.core.biz.model.RegistryParam;
|
||||||
|
import com.xxl.job.core.biz.model.ReturnT;
|
||||||
|
import com.xxl.job.core.util.XxlJobRemotingUtil;
|
||||||
import org.springframework.stereotype.Controller;
|
import org.springframework.stereotype.Controller;
|
||||||
|
import org.springframework.web.bind.annotation.RequestBody;
|
||||||
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
||||||
|
import org.springframework.web.bind.annotation.ResponseBody;
|
||||||
|
|
||||||
import javax.servlet.ServletException;
|
import javax.annotation.Resource;
|
||||||
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletRequest;
|
||||||
import javax.servlet.http.HttpServletResponse;
|
import java.util.List;
|
||||||
import java.io.IOException;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Created by xuxueli on 17/5/10.
|
* Created by xuxueli on 17/5/10.
|
||||||
*/
|
*/
|
||||||
@Controller
|
@Controller
|
||||||
public class JobApiController implements InitializingBean {
|
@RequestMapping("/api")
|
||||||
|
public class JobApiController {
|
||||||
|
|
||||||
|
@Resource
|
||||||
|
private AdminBiz adminBiz;
|
||||||
|
|
||||||
@Override
|
|
||||||
public void afterPropertiesSet() throws Exception {
|
|
||||||
|
|
||||||
|
// ---------------------- admin biz ----------------------
|
||||||
|
|
||||||
|
/**
|
||||||
|
* callback
|
||||||
|
*
|
||||||
|
* @param callbackParamList
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping("/callback")
|
||||||
|
@ResponseBody
|
||||||
|
@PermissionLimit(limit=false)
|
||||||
|
public ReturnT<String> callback(HttpServletRequest request, @RequestBody List<HandleCallbackParam> callbackParamList) {
|
||||||
|
|
||||||
|
if (XxlJobAdminConfig.getAdminConfig().getAccessToken()!=null
|
||||||
|
&& XxlJobAdminConfig.getAdminConfig().getAccessToken().trim().length()>0
|
||||||
|
&& !XxlJobAdminConfig.getAdminConfig().getAccessToken().equals(request.getHeader(XxlJobRemotingUtil.XXL_RPC_ACCESS_TOKEN))) {
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "The access token is wrong.");
|
||||||
|
}
|
||||||
|
|
||||||
|
return adminBiz.callback(callbackParamList);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* registry
|
||||||
|
*
|
||||||
|
* @param registryParam
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping("/registry")
|
||||||
|
@ResponseBody
|
||||||
|
@PermissionLimit(limit=false)
|
||||||
|
public ReturnT<String> registry(HttpServletRequest request, @RequestBody RegistryParam registryParam) {
|
||||||
|
|
||||||
|
if (XxlJobAdminConfig.getAdminConfig().getAccessToken()!=null
|
||||||
|
&& XxlJobAdminConfig.getAdminConfig().getAccessToken().trim().length()>0
|
||||||
|
&& !XxlJobAdminConfig.getAdminConfig().getAccessToken().equals(request.getHeader(XxlJobRemotingUtil.XXL_RPC_ACCESS_TOKEN))) {
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "The access token is wrong.");
|
||||||
|
}
|
||||||
|
|
||||||
|
return adminBiz.registry(registryParam);
|
||||||
}
|
}
|
||||||
|
|
||||||
@RequestMapping(AdminBiz.MAPPING)
|
/**
|
||||||
|
* registry remove
|
||||||
|
*
|
||||||
|
* @param registryParam
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
@RequestMapping("/registryRemove")
|
||||||
|
@ResponseBody
|
||||||
@PermissionLimit(limit=false)
|
@PermissionLimit(limit=false)
|
||||||
public void api(HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
|
public ReturnT<String> registryRemove(HttpServletRequest request, @RequestBody RegistryParam registryParam) {
|
||||||
XxlJobScheduler.invokeAdminService(request, response);
|
|
||||||
|
if (XxlJobAdminConfig.getAdminConfig().getAccessToken()!=null
|
||||||
|
&& XxlJobAdminConfig.getAdminConfig().getAccessToken().trim().length()>0
|
||||||
|
&& !XxlJobAdminConfig.getAdminConfig().getAccessToken().equals(request.getHeader(XxlJobRemotingUtil.XXL_RPC_ACCESS_TOKEN))) {
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "The access token is wrong.");
|
||||||
|
}
|
||||||
|
|
||||||
|
return adminBiz.registryRemove(registryParam);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// ---------------------- job biz ----------------------
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -0,0 +1,48 @@
|
|||||||
|
package com.xxl.job.core.biz.client;
|
||||||
|
|
||||||
|
import com.xxl.job.core.biz.AdminBiz;
|
||||||
|
import com.xxl.job.core.biz.model.HandleCallbackParam;
|
||||||
|
import com.xxl.job.core.biz.model.RegistryParam;
|
||||||
|
import com.xxl.job.core.biz.model.ReturnT;
|
||||||
|
import com.xxl.job.core.util.XxlJobRemotingUtil;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* admin api test
|
||||||
|
*
|
||||||
|
* @author xuxueli 2017-07-28 22:14:52
|
||||||
|
*/
|
||||||
|
public class AdminBizClient implements AdminBiz {
|
||||||
|
|
||||||
|
public AdminBizClient() {
|
||||||
|
}
|
||||||
|
public AdminBizClient(String addressUrl, String accessToken) {
|
||||||
|
this.addressUrl = addressUrl;
|
||||||
|
this.accessToken = accessToken;
|
||||||
|
|
||||||
|
// valid
|
||||||
|
if (!this.addressUrl.endsWith("/")) {
|
||||||
|
this.addressUrl = this.addressUrl + "/";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private String addressUrl ;
|
||||||
|
private String accessToken;
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ReturnT<String> callback(List<HandleCallbackParam> callbackParamList) {
|
||||||
|
return XxlJobRemotingUtil.postBody(addressUrl+"api/callback", accessToken, callbackParamList, 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ReturnT<String> registry(RegistryParam registryParam) {
|
||||||
|
return XxlJobRemotingUtil.postBody(addressUrl + "api/registry", accessToken, registryParam, 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ReturnT<String> registryRemove(RegistryParam registryParam) {
|
||||||
|
return XxlJobRemotingUtil.postBody(addressUrl + "api/registryRemove", accessToken, registryParam, 3);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,122 @@
|
|||||||
|
package com.xxl.job.core.util;
|
||||||
|
|
||||||
|
import com.xxl.job.core.biz.model.ReturnT;
|
||||||
|
import com.xxl.registry.client.util.json.BasicJson;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import java.io.BufferedReader;
|
||||||
|
import java.io.DataOutputStream;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author xuxueli 2018-11-25 00:55:31
|
||||||
|
*/
|
||||||
|
public class XxlJobRemotingUtil {
|
||||||
|
private static Logger logger = LoggerFactory.getLogger(XxlJobRemotingUtil.class);
|
||||||
|
public static String XXL_RPC_ACCESS_TOKEN = "XXL_RPC_ACCESS_TOKEN";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* post
|
||||||
|
*
|
||||||
|
* @param url
|
||||||
|
* @param accessToken
|
||||||
|
* @param requestObj
|
||||||
|
* @return
|
||||||
|
*/
|
||||||
|
public static ReturnT<String> postBody(String url, String accessToken, Object requestObj, int timeout) {
|
||||||
|
HttpURLConnection connection = null;
|
||||||
|
BufferedReader bufferedReader = null;
|
||||||
|
try {
|
||||||
|
// connection
|
||||||
|
URL realUrl = new URL(url);
|
||||||
|
connection = (HttpURLConnection) realUrl.openConnection();
|
||||||
|
|
||||||
|
// connection setting
|
||||||
|
connection.setRequestMethod("POST");
|
||||||
|
connection.setDoOutput(true);
|
||||||
|
connection.setDoInput(true);
|
||||||
|
connection.setUseCaches(false);
|
||||||
|
connection.setReadTimeout(timeout * 1000);
|
||||||
|
connection.setConnectTimeout(3 * 1000);
|
||||||
|
connection.setRequestProperty("connection", "Keep-Alive");
|
||||||
|
connection.setRequestProperty("Content-Type", "application/json;charset=UTF-8");
|
||||||
|
connection.setRequestProperty("Accept-Charset", "application/json;charset=UTF-8");
|
||||||
|
|
||||||
|
if(accessToken!=null && accessToken.trim().length()>0){
|
||||||
|
connection.setRequestProperty(XXL_RPC_ACCESS_TOKEN, accessToken);
|
||||||
|
}
|
||||||
|
|
||||||
|
// do connection
|
||||||
|
connection.connect();
|
||||||
|
|
||||||
|
// write requestBody
|
||||||
|
String requestBody = BasicJson.toJson(requestObj);
|
||||||
|
|
||||||
|
DataOutputStream dataOutputStream = new DataOutputStream(connection.getOutputStream());
|
||||||
|
dataOutputStream.writeBytes(requestBody);
|
||||||
|
dataOutputStream.flush();
|
||||||
|
dataOutputStream.close();
|
||||||
|
|
||||||
|
/*byte[] requestBodyBytes = requestBody.getBytes("UTF-8");
|
||||||
|
connection.setRequestProperty("Content-Length", String.valueOf(requestBodyBytes.length));
|
||||||
|
OutputStream outwritestream = connection.getOutputStream();
|
||||||
|
outwritestream.write(requestBodyBytes);
|
||||||
|
outwritestream.flush();
|
||||||
|
outwritestream.close();*/
|
||||||
|
|
||||||
|
// valid StatusCode
|
||||||
|
int statusCode = connection.getResponseCode();
|
||||||
|
if (statusCode != 200) {
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "xxl-rpc remoting fail, StatusCode("+ statusCode +") invalid. for url : " + url);
|
||||||
|
}
|
||||||
|
|
||||||
|
// result
|
||||||
|
bufferedReader = new BufferedReader(new InputStreamReader(connection.getInputStream()));
|
||||||
|
StringBuilder result = new StringBuilder();
|
||||||
|
String line;
|
||||||
|
while ((line = bufferedReader.readLine()) != null) {
|
||||||
|
result.append(line);
|
||||||
|
}
|
||||||
|
String resultJson = result.toString();
|
||||||
|
|
||||||
|
// parse returnT
|
||||||
|
try {
|
||||||
|
Map<String, Object> resultMap = BasicJson.parseMap(resultJson);
|
||||||
|
|
||||||
|
ReturnT<String> returnT = new ReturnT<String>();
|
||||||
|
if (resultMap==null) {
|
||||||
|
returnT.setCode(ReturnT.FAIL_CODE);
|
||||||
|
returnT.setMsg("AdminBizClient Remoting call fail.");
|
||||||
|
} else {
|
||||||
|
returnT.setCode(Integer.valueOf(String.valueOf(resultMap.get("code"))));
|
||||||
|
returnT.setMsg(String.valueOf(resultMap.get("msg")));
|
||||||
|
returnT.setContent(String.valueOf(resultMap.get("content")));
|
||||||
|
}
|
||||||
|
return returnT;
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "xxl-rpc remoting response content invalid("+ resultJson +"), for url : " + url);
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (Exception e) {
|
||||||
|
logger.error(e.getMessage(), e);
|
||||||
|
return new ReturnT<String>(ReturnT.FAIL_CODE, "xxl-rpc remoting error("+ e.getMessage() +"), for url : " + url);
|
||||||
|
} finally {
|
||||||
|
try {
|
||||||
|
if (bufferedReader != null) {
|
||||||
|
bufferedReader.close();
|
||||||
|
}
|
||||||
|
if (connection != null) {
|
||||||
|
connection.disconnect();
|
||||||
|
}
|
||||||
|
} catch (Exception e2) {
|
||||||
|
logger.error(e2.getMessage(), e2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in new issue