package com.se.simu.service.Impl;
|
|
import cn.hutool.crypto.SecureUtil;
|
import cn.hutool.crypto.asymmetric.AsymmetricAlgorithm;
|
import cn.hutool.crypto.asymmetric.AsymmetricCrypto;
|
import cn.hutool.crypto.asymmetric.KeyType;
|
import cn.hutool.crypto.asymmetric.RSA;
|
import com.alibaba.fastjson.JSONObject;
|
import com.se.simu.constant.CacheConstants;
|
import com.se.simu.domain.EntityTypeInfo;
|
import com.se.simu.domain.LoginParams;
|
import com.se.simu.helper.CaffeineHelper;
|
import com.se.simu.service.ProjectRelatedService;
|
import com.se.simu.utils.CustomWebClient;
|
import com.se.simu.utils.EntityLibraryUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
import reactor.core.publisher.Mono;
|
|
import java.security.KeyPair;
|
import java.security.PrivateKey;
|
import java.security.PublicKey;
|
import java.util.HashMap;
|
import java.util.Objects;
|
import java.util.concurrent.CompletableFuture;
|
import java.util.concurrent.ExecutionException;
|
import java.util.concurrent.atomic.AtomicReference;
|
|
@Slf4j
|
@Service
|
public class ProjectRelatedServiceImpl implements ProjectRelatedService {
|
|
|
// 公钥地址
|
@Value(value = "${app-server.publicKeyUrl}")
|
private String publicKeyUrl;
|
// devops 登录地址
|
@Value(value = "${app-server.loginUrl}")
|
private String loginUrl;
|
@Value(value = "${app-server.getDbUrl}")
|
private String getDbUrl;
|
|
// 查询地址
|
@Value(value = "${app-server.queryUrl}")
|
private String queryUrl;
|
|
|
/**
|
* 获取公钥
|
*
|
* @return {@link Object}
|
*/
|
@Override
|
public Object getPublicKey() {
|
HashMap<String, String> headers = new HashMap<>();
|
headers.put("Content-Type", "application/json");
|
CompletableFuture<String> postResponse = CustomWebClient.postAsFuture(publicKeyUrl, "", headers, String.class);
|
// 异步处理响应
|
postResponse.thenAccept(response -> {
|
log.info("response: {}", response);
|
if (response.contains("code")) {
|
try {
|
JSONObject postResponseJson = JSONObject.parseObject(response);
|
int statusCode = postResponseJson.getIntValue("code");
|
log.info("statusCode = " + statusCode);
|
String data = postResponseJson.getString("data");
|
// 缓存data,并设置1小时有效期
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "rsa_data_set");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "rsa_data_set", data);
|
log.info("publicKey = " + data);
|
} catch (Exception e) {
|
log.info("Failed to parse JSON: " + e.getMessage());
|
}
|
} else {
|
log.info("No 'code' field in the response: " + response);
|
}
|
});
|
return JSONObject.parseObject(postResponse.join());
|
}
|
|
/**
|
* 登录实体
|
*
|
* @param loginParams 登录参数
|
* @return {@link String}
|
*/
|
@Override
|
public Object loginEntity(LoginParams loginParams) {
|
// 判断redis中缓存是否存在（过期）
|
boolean isExists = Objects.isNull(CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token"));
|
if (!isExists) {
|
return JSONObject.parseObject((String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_response"));
|
} else {
|
// 清除所有缓存
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_response");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "EntityPublicKey");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "rsa_data_set");
|
}
|
// 获取私钥和公钥，长度必须是16、24或32
|
String publicKey = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "rsa_data_set");
|
// 假设从Redis中获取到用户名,判断
|
if (!StringUtils.isNotBlank(publicKey) && Objects.isNull(publicKey)) {
|
// 缓存中没有用户名,则进行登录
|
getPublicKey();
|
log.info("调用了登录获取用户名方法 $= ");
|
}
|
// 防止主线程提前结束
|
try {
|
// 等待异步请求完成
|
Thread.sleep(500);
|
} catch (InterruptedException e) {
|
e.printStackTrace();
|
}
|
publicKey = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "rsa_data_set");
|
log.info("publicKey = " + publicKey);
|
// 公钥加密
|
AsymmetricCrypto rsa = SecureUtil.rsa(null, publicKey);
|
String encrypt = rsa.encryptBase64(loginParams.getPassword(), KeyType.PublicKey);
|
// 设置加密后的密码
|
loginParams.setPassword(encrypt);
|
// 将loginParams实体转成JSON字符串
|
// String json = JSON.toJSONString(loginParams);
|
// log.info("json = " + json);
|
// 发送登录请求
|
Mono<String> postResponse = CustomWebClient.postAsMono(loginUrl, loginParams, String.class);
|
// 用于保存响应数据
|
AtomicReference<String> responseData = new AtomicReference<>();
|
postResponse.subscribe(response -> {
|
// 将响应数据保存到变量中
|
responseData.set(response);
|
String code = JSONObject.parseObject(response).getString("code");
|
// 如果code是200则登录成功
|
if (code.equals("200")) {
|
// 登录成功后,获取data
|
String data = JSONObject.parseObject(response).getString("data");
|
String token = JSONObject.parseObject(data).getString("token");
|
log.info("token = " + token);
|
// 设置data到Redis中
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_response");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "entity_db_response", response);
|
// 设置data到Redis中
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "entity_db_token", token);
|
} else {
|
// 登录失败
|
log.error("登录失败 $= " + response);
|
}
|
});
|
// 防止主线程提前结束
|
try {
|
postResponse.toFuture().get(); // 阻塞等待订阅操作完成
|
// 等待异步请求完成
|
Thread.sleep(1000);
|
} catch (InterruptedException | ExecutionException e) {
|
e.printStackTrace();
|
}
|
// 返回token
|
return JSONObject.parseObject(responseData.get());
|
}
|
|
/**
|
* 自定义密钥生成加密数据
|
*
|
* @param data 数据
|
* @param publicKey 公钥
|
* @return {@link String}
|
*/
|
private static String customKeysGenerateEncryptedData(String data, String publicKey) {
|
AsymmetricCrypto rsa = SecureUtil.rsa(null, publicKey);
|
String encrypt = rsa.encryptBase64(data, KeyType.PublicKey);
|
System.out.println("encrypt = " + encrypt);
|
return encrypt;
|
}
|
|
/**
|
* 自定义 RSAgenerate 加密数据0
|
*/
|
private static void customRSAGenerateEncryptedData0() {
|
RSA rsa = new RSA();
|
// 获取公钥和私钥
|
System.out.println(rsa.getPublicKey());
|
System.out.println(rsa.getPrivateKeyBase64());
|
System.out.println(rsa.getPrivateKey());
|
System.out.println(rsa.getPrivateKeyBase64());
|
// 私钥加密,公钥解密
|
System.out.println(new String(rsa.encrypt("testaa", KeyType.PrivateKey)));
|
System.out.println(new String(rsa.decrypt(rsa.encrypt("testaa", KeyType.PrivateKey), KeyType.PublicKey)));
|
// 公钥加密,私钥解密
|
System.out.println(new String(rsa.encrypt("testaa", KeyType.PublicKey)));
|
System.out.println(new String(rsa.decrypt(rsa.encrypt("testaa", KeyType.PublicKey), KeyType.PrivateKey)));
|
}
|
|
/**
|
* 自定义 RSAgenerate 加密数据1
|
*/
|
private static void customRSAGenerateEncryptedData1() {
|
KeyPair keyPair = SecureUtil.generateKeyPair(AsymmetricAlgorithm.RSA.getValue());
|
PrivateKey privateKey = keyPair.getPrivate();
|
PublicKey publicKey = keyPair.getPublic();
|
System.out.println(publicKey);
|
System.out.println(privateKey);
|
System.out.println("----------");
|
|
RSA rsa = new RSA(privateKey, publicKey);
|
// 私钥加密,公钥解密
|
System.out.println(new String(rsa.encrypt("testaa", KeyType.PrivateKey)));
|
System.out.println(new String(rsa.decrypt(rsa.encrypt("testaa", KeyType.PrivateKey), KeyType.PublicKey)));
|
// 公钥加密,私钥解密
|
System.out.println(new String(rsa.encrypt("testaa", KeyType.PublicKey)));
|
System.out.println(new String(rsa.decrypt(rsa.encrypt("testaa", KeyType.PublicKey), KeyType.PrivateKey)));
|
|
}
|
|
|
/**
|
* 获取访问实体库的token
|
*
|
* @return {@link Object}
|
*/
|
@Override
|
public Object getEntityPublicKey() {
|
// 判断redis中缓存是否存在（过期）
|
boolean isExists = Objects.isNull(CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "EntityPublicKey"));
|
if (!isExists) {
|
return JSONObject.parseObject((String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "EntityPublicKey"));
|
} else {
|
// 清除所有缓存
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_response");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "EntityPublicKey");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "rsa_data_set");
|
}
|
HashMap<String, String> headers = new HashMap<>();
|
headers.put("Content-Type", "application/json");
|
CompletableFuture<String> postResponse = CustomWebClient.postAsFuture(queryUrl, "", headers, String.class);
|
// 异步处理响应
|
postResponse.thenAccept(response -> {
|
log.info("response: {}", response);
|
if (response.contains("code")) {
|
try {
|
JSONObject postResponseJson = JSONObject.parseObject(response);
|
int statusCode = postResponseJson.getIntValue("code");
|
log.info("statusCode = " + statusCode);
|
String data = postResponseJson.getString("data");
|
// 缓存data,并设置1小时有效期
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "EntityPublicKey");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "EntityPublicKey", response);
|
log.info("EntityPublicKey = " + data);
|
} catch (Exception e) {
|
log.info("Failed to parse JSON: " + e.getMessage());
|
}
|
} else {
|
log.info("No 'code' field in the response: " + response);
|
}
|
});
|
return JSONObject.parseObject(postResponse.join());
|
}
|
|
|
/**
|
* 获取访问实体库的token
|
*
|
* @return {@link Object}
|
*/
|
@Override
|
public Object getDbLits() {
|
HashMap<String, String> headers = new HashMap<>();
|
headers.put("Content-Type", "application/json");
|
String token = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
log.info("token = " + token);
|
// 添加form参数
|
HashMap<String, Object> params = new HashMap<>();
|
params.put("token", token);
|
|
CompletableFuture<String> postResponse = CustomWebClient.postAsFuture(getDbUrl, params, headers, String.class);
|
// 异步处理响应
|
postResponse.thenAccept(response -> {
|
log.info("response: {}", response);
|
if (response.contains("code")) {
|
try {
|
JSONObject postResponseJson = JSONObject.parseObject(response);
|
int statusCode = postResponseJson.getIntValue("code");
|
log.info("statusCode = " + statusCode);
|
String data = postResponseJson.getString("data");
|
// 缓存data,并设置1小时有效期
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "EntityDbNameList");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "EntityDbNameList", response);
|
log.info("EntityDbNameList = " + data);
|
} catch (Exception e) {
|
log.info("Failed to parse JSON: " + e.getMessage());
|
}
|
} else {
|
log.info("No 'code' field in the response: " + response);
|
}
|
});
|
return JSONObject.parseObject(postResponse.join());
|
}
|
|
@Override
|
public Object getEntityTypeInfo(EntityTypeInfo entityTypeInfo) {
|
HashMap<String, String> headers = new HashMap<>();
|
headers.put("Content-Type", "application/json");
|
String token;
|
// 判断redis中缓存是否存在（过期）
|
boolean isExists = Objects.isNull(CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token"));
|
if (!isExists) {
|
token = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
} else {
|
try {
|
token = EntityLibraryUtils.login();
|
CaffeineHelper.remove(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
CaffeineHelper.put(CacheConstants.USER_CACHE_KEY + "entity_db_token", token);
|
}catch (Exception e){
|
return "登录失败！";
|
}
|
}
|
log.info("token = " + token);
|
// 添加form参数
|
HashMap<String, Object> params = new HashMap<>();
|
params.put("token", token);
|
params.put("start", Objects.nonNull(entityTypeInfo.getStart()) ? entityTypeInfo.getStart() : 1);
|
params.put("containCount", true);
|
params.put("count", Objects.nonNull(entityTypeInfo.getCount()) ? entityTypeInfo.getCount() : 20);
|
params.put("dbid", Objects.nonNull(entityTypeInfo.getDbid()) ? entityTypeInfo.getDbid() : "85257774fdb64e5f99f6778696cad02a");
|
params.put("layerid", Objects.nonNull(entityTypeInfo.getLayerid()) ? entityTypeInfo.getLayerid() : "8208c5be-adc1-4e7b-b952-37362e0bef32");
|
params.put("like", "");
|
params.put("querytype", "entity");
|
|
CompletableFuture<String> postResponse = CustomWebClient.postAsFuture(queryUrl, params, headers, String.class);
|
// 异步处理响应
|
postResponse.thenAccept(response -> {
|
log.info("response: {}", response);
|
if (response.contains("code")) {
|
try {
|
JSONObject postResponseJson = JSONObject.parseObject(response);
|
int statusCode = postResponseJson.getIntValue("code");
|
log.info("statusCode = " + statusCode);
|
String data = postResponseJson.getString("data");
|
log.info("getEntityTypeInfo = " + data);
|
} catch (Exception e) {
|
log.info("Failed to parse JSON: " + e.getMessage());
|
}
|
} else {
|
log.info("No 'code' field in the response: " + response);
|
}
|
});
|
return JSONObject.parseObject(postResponse.join());
|
}
|
|
|
@Override
|
public Object getEntityPipeInfo(EntityTypeInfo entityTypeInfo) {
|
HashMap<String, String> headers = new HashMap<>();
|
headers.put("Content-Type", "application/json");
|
String token;
|
// 判断redis中缓存是否存在（过期）
|
boolean isExists = Objects.isNull(CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token"));
|
if (!isExists) {
|
token = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
} else {
|
// 获取token
|
LoginParams loginParams = new LoginParams();
|
loginParams.setUserid("admin");
|
loginParams.setPassword("admin");
|
loginEntity(loginParams);
|
token = (String) CaffeineHelper.get(CacheConstants.USER_CACHE_KEY + "entity_db_token");
|
}
|
log.info("token = " + token);
|
// 添加form参数
|
HashMap<String, Object> params = new HashMap<>();
|
// e7e2af87096c45e0a14c0a4855cb0b90
|
params.put("token", token);
|
params.put("start", Objects.nonNull(entityTypeInfo.getStart()) ? entityTypeInfo.getStart() : 1);
|
params.put("containCount", true);
|
params.put("count", Objects.nonNull(entityTypeInfo.getCount()) ? entityTypeInfo.getCount() : 20);
|
params.put("dbid", Objects.nonNull(entityTypeInfo.getDbid()) ? entityTypeInfo.getDbid() : "85257774fdb64e5f99f6778696cad02a");
|
params.put("layerid", Objects.nonNull(entityTypeInfo.getLayerid()) ? entityTypeInfo.getLayerid() : "1e677d48-8dff-4975-b9a0-c16500193629");
|
params.put("like", "");
|
params.put("querytype", "entity");
|
|
CompletableFuture<String> postResponse = CustomWebClient.postAsFuture(queryUrl, params, headers, String.class);
|
// 异步处理响应
|
postResponse.thenAccept(response -> {
|
log.info("response: {}", response);
|
if (response.contains("code")) {
|
try {
|
JSONObject postResponseJson = JSONObject.parseObject(response);
|
int statusCode = postResponseJson.getIntValue("code");
|
log.info("statusCode = " + statusCode);
|
String data = postResponseJson.getString("data");
|
log.info("getEntityTypeInfo = " + data);
|
} catch (Exception e) {
|
log.info("Failed to parse JSON: " + e.getMessage());
|
}
|
} else {
|
log.info("No 'code' field in the response: " + response);
|
}
|
});
|
return JSONObject.parseObject(postResponse.join());
|
}
|
|
|
}
|
