修改
This commit is contained in:
dimengzhe
71
src/main/java/com/yxt/ss/gateway/api/ClientService.java
Normal file
71
src/main/java/com/yxt/ss/gateway/api/ClientService.java
Normal file
@@ -0,0 +1,71 @@
|
||||
package com.yxt.ss.gateway.api;
|
||||
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.io.UnsupportedEncodingException;
|
||||
import java.net.URLEncoder;
|
||||
import java.security.MessageDigest;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* @description:
|
||||
* @author: dimengzhe
|
||||
* @date: 2024/12/10
|
||||
**/
|
||||
@Service
|
||||
public class ClientService {
|
||||
|
||||
/**
|
||||
* 生成请求签名
|
||||
*
|
||||
* @param parameters 请求参数
|
||||
* @param secret 密钥
|
||||
* @return 签名
|
||||
*/
|
||||
public String generateSignature(Map<String, String> parameters, String secret) throws UnsupportedEncodingException, NoSuchAlgorithmException {
|
||||
|
||||
// 2. 拼接参数字符串
|
||||
String content = joinParameters(parameters);
|
||||
|
||||
// 3. 将密钥加在参数字符串的前后
|
||||
content = secret + content + secret;
|
||||
|
||||
// 4. 计算签名 (MD5)
|
||||
return md5(content);
|
||||
}
|
||||
|
||||
/**
|
||||
* 拼接参数字符串
|
||||
*
|
||||
* @param tree 排序后的参数
|
||||
* @return 拼接后的参数字符串
|
||||
*/
|
||||
public String joinParameters(Map<String, String> tree) throws UnsupportedEncodingException {
|
||||
StringBuilder builder = new StringBuilder();
|
||||
for (Map.Entry<String, String> entry : tree.entrySet()) {
|
||||
if (builder.length() > 0) {
|
||||
builder.append("&");
|
||||
}
|
||||
builder.append(entry.getKey()).append("=");
|
||||
builder.append(URLEncoder.encode(entry.getValue(), "UTF-8"));
|
||||
}
|
||||
return builder.toString();
|
||||
}
|
||||
|
||||
/**
|
||||
* 计算 MD5
|
||||
*
|
||||
* @param content 要计算 MD5 的字符串
|
||||
* @return MD5 值
|
||||
*/
|
||||
public String md5(String content) throws NoSuchAlgorithmException {
|
||||
MessageDigest md = MessageDigest.getInstance("MD5");
|
||||
byte[] bytes = md.digest(content.getBytes());
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (byte b : bytes) {
|
||||
sb.append(String.format("%02x", b));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
}
|
||||
128
src/main/java/com/yxt/ss/gateway/api/rest/ClientRest.java
Normal file
128
src/main/java/com/yxt/ss/gateway/api/rest/ClientRest.java
Normal file
@@ -0,0 +1,128 @@
|
||||
package com.yxt.ss.gateway.api.rest;
|
||||
|
||||
import com.fasterxml.jackson.databind.JsonNode;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.yxt.ss.gateway.api.ClientService;
|
||||
import com.yxt.ss.gateway.api.utils.ResultBean;
|
||||
import com.yxt.ss.gateway.api.utils.SignatureQuery;
|
||||
import com.yxt.ss.gateway.api.utils.SignatureUtil;
|
||||
import okhttp3.*;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.UnsupportedEncodingException;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
import java.util.Map;
|
||||
import java.util.TreeMap;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* @description:
|
||||
* @author: dimengzhe
|
||||
* @date: 2024/12/10
|
||||
**/
|
||||
@RestController
|
||||
@RequestMapping("/client")
|
||||
public class ClientRest {
|
||||
|
||||
@Autowired
|
||||
private ClientService clientService;
|
||||
|
||||
|
||||
//appkey
|
||||
static final String APPKEY = "appKey4";
|
||||
//secret
|
||||
static final String SECRET = "secret";
|
||||
|
||||
//开发端，生成签名并调用服务器端验证签名、appKey等值。
|
||||
@PostMapping("/getSign")
|
||||
ResultBean getSign(SignatureQuery query) {
|
||||
ResultBean<String> rb = ResultBean.fireFail();
|
||||
try {
|
||||
Map<String, String> formData = query.getParameters();
|
||||
//使用treeMap排序
|
||||
Map<String, String> tree = new TreeMap<>(formData);
|
||||
tree.put("_app", APPKEY);
|
||||
tree.put("_t", String.valueOf(System.currentTimeMillis() / 1000));
|
||||
tree.put("_s", "");
|
||||
// 生成签名
|
||||
String sign = clientService.generateSignature(tree, SECRET);
|
||||
//添加签名值map
|
||||
tree.put("_sign", sign);
|
||||
//发起请求
|
||||
ResultBean resultBean = client(tree);
|
||||
if (!resultBean.getSuccess()) {
|
||||
return rb.setMsg(resultBean.getMsg());
|
||||
}
|
||||
//通过验证继续调用接口
|
||||
|
||||
|
||||
return rb.success();
|
||||
} catch (UnsupportedEncodingException e) {
|
||||
return rb.setMsg("Unsupported encoding: " + e.getMessage());
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
return rb.setMsg("Algorithm not found: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
//发起请求验证签名等
|
||||
public ResultBean client(Map<String, String> data) {
|
||||
ResultBean rb = ResultBean.fireFail();
|
||||
OkHttpClient client = new OkHttpClient.Builder()
|
||||
.connectTimeout(10, TimeUnit.SECONDS)
|
||||
.writeTimeout(10, TimeUnit.SECONDS)
|
||||
.readTimeout(30, TimeUnit.SECONDS)
|
||||
.build();
|
||||
|
||||
try {
|
||||
// 构建URL
|
||||
String endPoint = "http://127.0.0.1:9999";
|
||||
String path = "/signature/validate";
|
||||
|
||||
// 创建FormData
|
||||
FormBody.Builder formBuilder = new FormBody.Builder();
|
||||
for (Map.Entry<String, String> entry : data.entrySet()) {
|
||||
formBuilder.add(entry.getKey(), entry.getValue());
|
||||
}
|
||||
RequestBody formBody = formBuilder.build();
|
||||
|
||||
// 构建POST请求
|
||||
String url = endPoint + path;
|
||||
System.out.println("Request URL: " + url);
|
||||
System.out.println("Request Data: " + data);
|
||||
|
||||
Request request = new Request.Builder()
|
||||
.url(url)
|
||||
.post(formBody)
|
||||
.build();
|
||||
|
||||
// 发送请求
|
||||
try (Response response = client.newCall(request).execute()) {
|
||||
String responseBody = response.body().string();
|
||||
// 使用 Jackson 解析 JSON 响应
|
||||
ObjectMapper objectMapper = new ObjectMapper();
|
||||
JsonNode jsonNode = objectMapper.readTree(responseBody);
|
||||
String success = jsonNode.path("success").asText();
|
||||
String msg = jsonNode.path("msg").asText();
|
||||
if ("false".equals(success)) {
|
||||
return rb.setMsg(msg);
|
||||
}
|
||||
if (response.isSuccessful()) {
|
||||
System.out.println("Response: " + response.body().string());
|
||||
} else {
|
||||
System.err.println("Request failed: " + response.message());
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
} catch (IOException e) {
|
||||
System.err.println("Network error: " + e.getMessage());
|
||||
} catch (Exception e) {
|
||||
System.err.println("Unexpected error: " + e.getMessage());
|
||||
}
|
||||
return rb.success();
|
||||
}
|
||||
}
|
||||
@@ -1,27 +1,18 @@
|
||||
package com.yxt.ss.gateway.api.rest;
|
||||
|
||||
import com.fasterxml.jackson.databind.JsonNode;
|
||||
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||
import com.yxt.ss.gateway.api.utils.*;
|
||||
import okhttp3.*;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.util.StringUtils;
|
||||
import org.springframework.web.bind.annotation.PostMapping;
|
||||
import org.springframework.web.bind.annotation.RequestMapping;
|
||||
import org.springframework.web.bind.annotation.RequestParam;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.UnsupportedEncodingException;
|
||||
import java.net.URLEncoder;
|
||||
import java.security.MessageDigest;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
import java.time.Instant;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
import java.util.TreeMap;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
/**
|
||||
* @description:
|
||||
@@ -45,36 +36,6 @@ public class Signature {
|
||||
//secret
|
||||
static final String SECRET = "secret";
|
||||
|
||||
//获取参数
|
||||
@PostMapping("/getSign")
|
||||
ResultBean getSign(SignatureQuery query) {
|
||||
ResultBean<String> rb = ResultBean.fireFail();
|
||||
try {
|
||||
Map<String, String> formData = query.getParameters();
|
||||
Map<String, String> tree = new TreeMap<>(formData);
|
||||
tree.put("_app", APPKEY);
|
||||
tree.put("_t", String.valueOf(System.currentTimeMillis() / 1000));
|
||||
tree.put("_s", "");
|
||||
// 生成签名
|
||||
String sign = SignatureUtil.generateSignature(tree, SECRET);
|
||||
//添加签名值map
|
||||
tree.put("_sign", sign);
|
||||
//发起请求
|
||||
ResultBean resultBean = client(tree);
|
||||
if (!resultBean.getSuccess()) {
|
||||
return rb.setMsg(resultBean.getMsg());
|
||||
}
|
||||
//通过验证继续调用接口
|
||||
|
||||
|
||||
return rb.success();
|
||||
} catch (UnsupportedEncodingException e) {
|
||||
return rb.setMsg("Unsupported encoding: " + e.getMessage());
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
return rb.setMsg("Algorithm not found: " + e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
//验证
|
||||
@PostMapping("/validate")
|
||||
ResultBean validate(Map<String, String> data) {
|
||||
@@ -138,60 +99,5 @@ public class Signature {
|
||||
|
||||
}
|
||||
|
||||
public ResultBean client(Map<String, String> data) {
|
||||
ResultBean rb = ResultBean.fireFail();
|
||||
OkHttpClient client = new OkHttpClient.Builder()
|
||||
.connectTimeout(10, TimeUnit.SECONDS)
|
||||
.writeTimeout(10, TimeUnit.SECONDS)
|
||||
.readTimeout(30, TimeUnit.SECONDS)
|
||||
.build();
|
||||
|
||||
try {
|
||||
// 构建URL
|
||||
String endPoint = "http://127.0.0.1:9999";
|
||||
String path = "/signature/validate";
|
||||
|
||||
// 创建FormData
|
||||
FormBody.Builder formBuilder = new FormBody.Builder();
|
||||
for (Map.Entry<String, String> entry : data.entrySet()) {
|
||||
formBuilder.add(entry.getKey(), entry.getValue());
|
||||
}
|
||||
RequestBody formBody = formBuilder.build();
|
||||
|
||||
// 构建POST请求
|
||||
String url = endPoint + path;
|
||||
System.out.println("Request URL: " + url);
|
||||
System.out.println("Request Data: " + data);
|
||||
|
||||
Request request = new Request.Builder()
|
||||
.url(url)
|
||||
.post(formBody)
|
||||
.build();
|
||||
|
||||
// 发送请求
|
||||
try (Response response = client.newCall(request).execute()) {
|
||||
String responseBody = response.body().string();
|
||||
// 使用 Jackson 解析 JSON 响应
|
||||
ObjectMapper objectMapper = new ObjectMapper();
|
||||
JsonNode jsonNode = objectMapper.readTree(responseBody);
|
||||
String success = jsonNode.path("success").asText();
|
||||
String msg = jsonNode.path("msg").asText();
|
||||
if ("false".equals(success)) {
|
||||
return rb.setMsg(msg);
|
||||
}
|
||||
if (response.isSuccessful()) {
|
||||
System.out.println("Response: " + response.body().string());
|
||||
} else {
|
||||
System.err.println("Request failed: " + response.message());
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
} catch (IOException e) {
|
||||
System.err.println("Network error: " + e.getMessage());
|
||||
} catch (Exception e) {
|
||||
System.err.println("Unexpected error: " + e.getMessage());
|
||||
}
|
||||
return rb.success();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user