admin/ko-time
1
0
Fork 0
mirror of https://gitee.com/huoyo/ko-time.git synced 2025-12-06 08:48:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

add the redis support for saving data

Browse Source
This commit is contained in:
huoyo 2022-08-30 00:33:15 +08:00
parent e941d97020
commit a1450e6761
7 changed files with 245 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pom.xml
View File

@ -29,7 +29,8 @@
<spring-web.version>5.2.11.RELEASE</spring-web.version>
<spring-webmvc.version>5.3.6</spring-webmvc.version>
<spring-boot.version>2.3.1.RELEASE</spring-boot.version>
<spring-redis.version>2.6.4</spring-redis.version>
<spring-redis.version>2.6.2</spring-redis.version>
<fastjson.version>2.0.9</fastjson.version>
</properties>
<dependencies>
<dependency>
@ -69,16 +70,16 @@
<scope>provided</scope>
</dependency>
<dependency>
<groupId>org.javassist</groupId>
<artifactId>javassist</artifactId>
<version>3.26.0-GA</version>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-redis</artifactId>
<version>${spring-redis.version}</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>${fastjson.version}</version>
</dependency>
<!-- <dependency>-->
<!-- <groupId>org.springframework.boot</groupId>-->
<!-- <artifactId>spring-boot-starter-data-redis</artifactId>-->
<!-- <version>${spring-redis.version}</version>-->
<!-- <scope>provided</scope>-->
<!-- </dependency>-->
</dependencies>
<developers>
<developer>

10
src/main/java/cn/langpy/kotime/config/DefaultConfig.java
View File

@ -22,12 +22,22 @@ public class DefaultConfig {
private Boolean paramAnalyse;
private String saver;
private String dataSource;
private String redisTemplate;
private Integer threadNum;
private String contextPath;
private Boolean authEnable;
private String userName;
private String password;
public String getRedisTemplate() {
return redisTemplate;
}
public void setRedisTemplate(String redisTemplate) {
this.redisTemplate = redisTemplate;
}
public String getLanguage() {
return language;
}

69
src/main/java/cn/langpy/kotime/config/LoadConfig.java
View File

@ -15,13 +15,17 @@ import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.sql.DataSource;
import java.util.Map;
import java.util.logging.Logger;
import java.util.stream.Collectors;
/**
* zhangchang
@ -62,7 +66,6 @@ public class LoadConfig {
public void initConfig() {
DefaultConfig config = new DefaultConfig();
config.setLogEnable(defaultConfig.getLogEnable() == null ? logEnable : defaultConfig.getLogEnable());
// config.setPointcut(defaultConfig.getPointcut() == null ? pointcut : defaultConfig.getPointcut());
config.setLogLanguage(defaultConfig.getLogLanguage() == null ? logLanguage : defaultConfig.getLogLanguage());
config.setThreshold(defaultConfig.getThreshold() == null ? timeThreshold : defaultConfig.getThreshold());
config.setExceptionEnable(defaultConfig.getExceptionEnable() == null ? exceptionEnable : defaultConfig.getExceptionEnable());
@ -73,20 +76,8 @@ public class LoadConfig {
config.setThreadNum(defaultConfig.getThreadNum() == null ? 2 : defaultConfig.getThreadNum());
config.setAuthEnable(defaultConfig.getAuthEnable() == null ? false : defaultConfig.getAuthEnable());
config.setParamAnalyse(defaultConfig.getParamAnalyse() == null ? true : defaultConfig.getParamAnalyse());
try {
DataSource dataSource = applicationContext.getBean(DataSource.class);
Context.setDataSource(dataSource);
}catch (NoUniqueBeanDefinitionException e){
if (StringUtils.isEmpty(config.getDataSource())) {
log.warning("kotime=>No unique bean of type 'DataSource' available,you can define it by `ko-time.data-source=xxx`");
}else {
DataSource dataSource = applicationContext.getBean(config.getDataSource(),DataSource.class);
Context.setDataSource(dataSource);
}
}catch (NoSuchBeanDefinitionException e){
log.warning("kotime=>No qualifying bean of type 'DataSource' available,but you can ignore it if your KoTime saver is `ko-time.saver=memory`");
}
configDataSource(config);
configRedisTemplate(config);
Context.setConfig(config);
String[] names = applicationContext.getBeanNamesForType(GraphService.class);
for (String name : names) {
@ -99,8 +90,8 @@ public class LoadConfig {
}
}
}
if (null==Context.getSaver()) {
throw new DataBaseException("error `ko-time.saver="+config.getSaver()+"`, and you can only choose an option in {memory,database} for `ko-time.saver=`!");
if (null == Context.getSaver()) {
throw new DataBaseException("error `ko-time.saver=" + config.getSaver() + "`, and you can only choose an option in {memory,database,redis} for `ko-time.saver=`!");
}
log.info("kotime=>loading config");
@ -112,6 +103,48 @@ public class LoadConfig {
initMethodHandlers();
}
public void configDataSource(DefaultConfig config) {
if (!"database".equals(config.getSaver())) {
return;
}
try {
DataSource dataSource = applicationContext.getBean(DataSource.class);
Context.setDataSource(dataSource);
} catch (NoUniqueBeanDefinitionException e) {
if (StringUtils.isEmpty(config.getDataSource())) {
log.warning("kotime=>No unique bean of type 'DataSource' available,you can define it by `ko-time.data-source=xxx`");
} else {
DataSource dataSource = applicationContext.getBean(config.getDataSource(), DataSource.class);
Context.setDataSource(dataSource);
}
} catch (NoSuchBeanDefinitionException e) {
log.warning("kotime=>No qualifying bean of type 'DataSource' available,but you can ignore it if your KoTime saver is `ko-time.saver=memory`");
}
}
public void configRedisTemplate(DefaultConfig config) {
if (!"redis".equals(config.getSaver())) {
return;
}
try {
StringRedisTemplate redisTemplate = applicationContext.getBean(StringRedisTemplate.class);
Context.setStringRedisTemplate(redisTemplate);
} catch (NoUniqueBeanDefinitionException e) {
if (!StringUtils.isEmpty(config.getRedisTemplate())) {
StringRedisTemplate redisTemplate = applicationContext.getBean(config.getRedisTemplate(), StringRedisTemplate.class);
Context.setStringRedisTemplate(redisTemplate);
} else {
Map<String, StringRedisTemplate> beansOfType = applicationContext.getBeansOfType(StringRedisTemplate.class);
log.warning("kotime=>No unique bean of type 'StringRedisTemplate' available,you can define it by `ko-time.redis-template=xxx`,and you can choose a name in "+beansOfType.keySet().stream().collect(Collectors.toList()));
log.warning("kotime=>Now the firsr was be set.");
Context.setStringRedisTemplate(beansOfType.values().stream().collect(Collectors.toList()).get(0));
}
} catch (NoSuchBeanDefinitionException e) {
log.warning("kotime=>No qualifying bean of type 'StringRedisTemplate' available,but you can ignore it if your KoTime saver is `ko-time.saver=memory`");
}
}
public void initMethodHandlers() {
String[] names = applicationContext.getBeanNamesForType(InvokedHandler.class);
for (String name : names) {
@ -126,7 +159,7 @@ public class LoadConfig {
}
}
for (int i = 0; i < Context.getConfig().getThreadNum(); i++) {
new Thread(()->InvokedQueue.onInveked()).start();
new Thread(() -> InvokedQueue.onInveked()).start();
}
}

28
src/main/java/cn/langpy/kotime/config/SaveResourceConfig.java Normal file
View File

@ -0,0 +1,28 @@
package cn.langpy.kotime.config;
import cn.langpy.kotime.util.Context;
import cn.langpy.kotime.util.KoUtil;
import org.springframework.boot.CommandLineRunner;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import javax.sql.DataSource;
@Component
public class SaveResourceConfig implements CommandLineRunner{
@Override
public void run(String... args) throws Exception {
DataSource dataSource = KoUtil.getDataSource();
if (null!=dataSource) {
Context.setDataSource(dataSource);
}
StringRedisTemplate redisTemplate = KoUtil.getStringRedisTemplate();
if (null!=redisTemplate) {
Context.setStringRedisTemplate(redisTemplate);
}
KoUtil.clearCaches();
}
}

169
src/main/java/cn/langpy/kotime/data/RedisBase.java
View File

@ -5,6 +5,13 @@ import cn.langpy.kotime.service.GraphService;
import cn.langpy.kotime.util.Common;
import cn.langpy.kotime.util.Context;
import cn.langpy.kotime.util.MethodType;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import org.springframework.context.annotation.Lazy;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.GenericJackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
@ -18,25 +25,21 @@ import static java.util.stream.Collectors.toList;
/**
* zhangchang
*/
//@Component("redis")
@Lazy
@Component("redis")
public class RedisBase implements GraphService {
private volatile static Map<String, MethodNode> methodNodes;
private static final String methodPre = "KO_METHODS:";
private static final String exceptionPre = "KO_EXCEPTIONS:";
private static final String methodRelationPre = "KO_METHODRES:";
private static final String exceptionRelationPre = "KO_EXCEPTIONRES:";
private static final String paramValueMetricMapPre = "KO_PARAM:";
private volatile static Map<String, ExceptionNode> exceptions;
private static StringRedisTemplate redisTemplate;
private volatile static Map<String, MethodRelation> methodRelations;
public RedisBase() {
redisTemplate = Context.getStringRedisTemplate();
private volatile static Map<String, ExceptionRelation> exceptionRelations;
private volatile static Map<String, Map<String, ParamMetric>> paramValueMetricMap;
static {
methodNodes = new HashMap<>();
exceptions = new HashMap<>();
methodRelations = new HashMap<>();
exceptionRelations = new HashMap<>();
paramValueMetricMap = new HashMap<String, Map<String, ParamMetric>>();
}
@Override
@ -44,24 +47,26 @@ public class RedisBase implements GraphService {
if (null == methodNode) {
return;
}
if (!methodNodes.containsKey(methodNode.getId())) {
methodNodes.put(methodNode.getId(), methodNode);
String key = methodPre + methodNode.getId();
if (!redisTemplate.hasKey(key)) {
insert(key, methodNode);
} else {
if (methodNode.getMethodType() == MethodType.Controller && !StringUtils.isEmpty(methodNode.getRouteName())) {
MethodNode controller = methodNodes.get(methodNode.getId());
MethodNode controller = query(key, MethodNode.class);
controller.setRouteName(methodNode.getRouteName());
methodNodes.put(methodNode.getId(), controller);
insert(key, controller);
}
}
}
public void addParamAnalyse(String methodId, Parameter[] names, Object[] values, double v) {
String paramsKey = Common.getPramsStr(names,values);
if (paramValueMetricMap.containsKey(methodId)) {
Map<String, ParamMetric> paramMetricMap = paramValueMetricMap.get(methodId);
String paramsKey = Common.getPramsStr(names, values);
String key = paramValueMetricMapPre + methodId;
if (redisTemplate.hasKey(key)) {
Map<String, JSONObject> paramMetricMap = query(key, Map.class);
if (paramMetricMap.containsKey(paramsKey)) {
ParamMetric paramMetric = paramMetricMap.get(paramsKey);
ParamMetric paramMetric = paramMetricMap.get(paramsKey).toJavaObject(ParamMetric.class);
BigDecimal bg = BigDecimal.valueOf((paramMetric.getAvgRunTime() + v) / 2.0);
double avg = bg.setScale(2, BigDecimal.ROUND_HALF_UP).doubleValue();
paramMetric.setAvgRunTime(avg);
@ -71,35 +76,29 @@ public class RedisBase implements GraphService {
if (v < paramMetric.getMinRunTime()) {
paramMetric.setMinRunTime(v);
}
} else {
ParamMetric paramMetric = new ParamMetric();
paramMetric.setMaxRunTime(v);
paramMetric.setAvgRunTime(v);
paramMetric.setMaxRunTime(v);
paramMetricMap.put(paramsKey, paramMetric);
insert(key, paramMetricMap);
}
} else {
ParamMetric paramMetric = new ParamMetric();
paramMetric.setMaxRunTime(v);
paramMetric.setAvgRunTime(v);
paramMetric.setMaxRunTime(v);
Map<String, ParamMetric> paramMetricMap = new HashMap<>();
paramMetricMap.put(paramsKey, paramMetric);
paramValueMetricMap.put(methodId, paramMetricMap);
insert(key, paramMetricMap);
}
}
@Override
public MethodRelation addMethodRelation(MethodNode sourceMethodNode, MethodNode targetMethodNode) {
if (null == sourceMethodNode || null == targetMethodNode ) {
if (null == sourceMethodNode || null == targetMethodNode) {
return null;
}
if (sourceMethodNode.getId().equals(targetMethodNode.getId())) {
return null;
}
if (methodRelations.containsKey(targetMethodNode.getId()+sourceMethodNode.getId())) {
if (redisTemplate.hasKey(methodRelationPre + targetMethodNode.getId() + sourceMethodNode.getId())) {
return null;
}
MethodRelation methodRelation = new MethodRelation();
@ -109,9 +108,10 @@ public class RedisBase implements GraphService {
methodRelation.setAvgRunTime(targetMethodNode.getValue());
methodRelation.setMaxRunTime(targetMethodNode.getValue());
methodRelation.setMinRunTime(targetMethodNode.getValue());
MethodRelation old = methodRelations.get(methodRelation.getId());
String key = methodRelationPre + methodRelation.getId();
MethodRelation old = query(key, MethodRelation.class);
if (null == old) {
methodRelations.put(methodRelation.getId(), methodRelation);
insert(key, methodRelation);
return methodRelation;
} else {
BigDecimal bg = BigDecimal.valueOf((methodRelation.getAvgRunTime() + old.getAvgRunTime()) / 2.0);
@ -119,6 +119,7 @@ public class RedisBase implements GraphService {
old.setAvgRunTime(avg);
old.setMaxRunTime(methodRelation.getMaxRunTime() > old.getMaxRunTime() ? methodRelation.getMaxRunTime() : old.getMaxRunTime());
old.setMinRunTime(methodRelation.getMinRunTime() < old.getMinRunTime() ? methodRelation.getMinRunTime() : old.getMinRunTime());
insert(key, old);
return old;
}
}
@ -130,9 +131,10 @@ public class RedisBase implements GraphService {
exceptionRelation.setSourceId(sourceMethodNode.getId());
exceptionRelation.setTargetId(exceptionNode.getId());
exceptionRelation.setLocation(exceptionNode.getValue());
ExceptionRelation old = exceptionRelations.get(exceptionRelation.getId());
String key = exceptionRelationPre + exceptionRelation.getId();
ExceptionRelation old = query(key, ExceptionRelation.class);
if (null == old) {
exceptionRelations.put(exceptionRelation.getId(), exceptionRelation);
insert(key, exceptionRelation);
return exceptionRelation;
} else {
return old;
@ -141,18 +143,20 @@ public class RedisBase implements GraphService {
@Override
public void addExceptionNode(ExceptionNode exceptionNode) {
if (!exceptions.containsKey(exceptionNode.getId())) {
exceptions.put(exceptionNode.getId(), exceptionNode);
String key = exceptionPre + exceptionNode.getId();
if (!redisTemplate.hasKey(key)) {
insert(key, exceptionNode);
}
}
@Override
public List<ExceptionInfo> getExceptions(String methodId) {
List<ExceptionInfo> exceptionInfos = new ArrayList<>();
List<ExceptionRelation> relations = exceptionRelations.values().stream().filter(exceptionRelation -> exceptionRelation.getSourceId().equals(methodId)).collect(toList());
List<ExceptionRelation> searchs = searchList(exceptionRelationPre, ExceptionRelation.class);
List<ExceptionRelation> relations = searchs.stream().filter(exceptionRelation -> exceptionRelation.getSourceId().equals(methodId)).collect(toList());
for (ExceptionRelation relation : relations) {
String exceptionId = relation.getTargetId();
ExceptionNode exceptionNode = exceptions.get(exceptionId);
ExceptionNode exceptionNode = query(exceptionPre + exceptionId, ExceptionNode.class);
ExceptionInfo exceptionInfo = new ExceptionInfo();
exceptionInfo.setId(exceptionNode.getId());
exceptionInfo.setName(exceptionNode.getName());
@ -168,16 +172,21 @@ public class RedisBase implements GraphService {
@Override
public List<ExceptionNode> getExceptions() {
return exceptions.values().stream().distinct().collect(toList());
List<ExceptionNode> searchs = searchList(exceptionPre, ExceptionNode.class);
return searchs.stream().distinct().collect(toList());
}
@Override
public List<MethodInfo> getControllers() {
List<MethodInfo> methodInfos = new ArrayList<>();
for (MethodNode methodNode : methodNodes.values()) {
List<MethodNode> smethodNodes = searchList(methodPre, MethodNode.class);
for (MethodNode methodNode : smethodNodes) {
if (MethodType.Controller == methodNode.getMethodType()) {
String id = methodNode.getId();
Optional<MethodRelation> relations = methodRelations.values().stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
List<MethodRelation> smethodRelations = searchList(methodRelationPre, MethodRelation.class);
Optional<MethodRelation> relations = smethodRelations.stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
MethodRelation relation = null;
if (relations.isPresent()) {
relation = relations.get();
@ -205,17 +214,24 @@ public class RedisBase implements GraphService {
@Override
public Map<String, ParamMetric> getMethodParamGraph(String methodId) {
Map<String, ParamMetric> paramMetricMap = paramValueMetricMap.get(methodId);
return paramMetricMap;
String key = paramValueMetricMapPre + methodId;
Map<String, ParamMetric> res = new HashMap<>();
Map<String, JSONObject> paramMetric = query(key, Map.class);
for (String name : paramMetric.keySet()) {
res.put(name, paramMetric.get(name).toJavaObject(ParamMetric.class));
}
return res;
}
@Override
public List<MethodInfo> searchMethods(String question) {
List<MethodInfo> methodInfos = new ArrayList<>();
for (MethodNode methodNode : methodNodes.values()) {
List<MethodNode> smethodNodes = searchList(methodPre, MethodNode.class);
List<MethodRelation> methodRelationList = searchList(methodRelationPre, MethodRelation.class);
for (MethodNode methodNode : smethodNodes) {
if (methodNode.getName().toLowerCase().contains(question.toLowerCase())) {
String id = methodNode.getId();
Optional<MethodRelation> relations = methodRelations.values().stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
Optional<MethodRelation> relations = methodRelationList.stream().filter(methodRelation -> methodRelation.getTargetId().equals(id)).findFirst();
MethodRelation relation = null;
if (relations.isPresent()) {
relation = relations.get();
@ -244,7 +260,8 @@ public class RedisBase implements GraphService {
@Override
public List<String> getCondidates(String question) {
List<String> methodInfos = new ArrayList<>();
for (MethodNode methodNode : methodNodes.values()) {
List<MethodNode> smethodNodes = searchList(methodPre, MethodNode.class);
for (MethodNode methodNode : smethodNodes) {
if (methodNode.getName().toLowerCase().contains(question.toLowerCase())) {
if (!methodInfos.contains(methodNode.getName())) {
methodInfos.add(methodNode.getName());
@ -260,11 +277,12 @@ public class RedisBase implements GraphService {
@Override
public List<ExceptionInfo> getExceptionInfos(String exceptionId) {
List<ExceptionInfo> exceptionInfos = new ArrayList<>();
for (ExceptionRelation relation : exceptionRelations.values()) {
List<ExceptionRelation> sexceptionRelations = searchList(exceptionRelationPre, ExceptionRelation.class);
for (ExceptionRelation relation : sexceptionRelations) {
if (relation.getTargetId().equals(exceptionId)) {
String sourceMethodId = relation.getSourceId();
MethodNode methodNode = methodNodes.get(sourceMethodId);
ExceptionNode exceptionNode = exceptions.get(exceptionId);
MethodNode methodNode = query(methodPre + sourceMethodId, MethodNode.class);
ExceptionNode exceptionNode = query(exceptionPre + exceptionId, ExceptionNode.class);
ExceptionInfo exceptionInfo = new ExceptionInfo();
exceptionInfo.setId(exceptionNode.getId());
@ -285,10 +303,11 @@ public class RedisBase implements GraphService {
@Override
public List<MethodInfo> getChildren(String methodId) {
List<MethodInfo> methodInfos = new ArrayList<>();
for (MethodRelation methodRelation : methodRelations.values()) {
List<MethodRelation> methodRelationList = searchList(methodRelationPre, MethodRelation.class);
for (MethodRelation methodRelation : methodRelationList) {
if (methodRelation.getSourceId().equals(methodId)) {
String targetMethodId = methodRelation.getTargetId();
MethodNode methodNode = methodNodes.get(targetMethodId);
MethodNode methodNode = query(methodPre + targetMethodId, MethodNode.class);
MethodInfo methodInfo = new MethodInfo();
methodInfo.setId(methodNode.getId());
methodInfo.setName(methodNode.getName());
@ -338,14 +357,16 @@ public class RedisBase implements GraphService {
@Override
public MethodInfo getTree(String methodId) {
MethodInfo rootInfo = new MethodInfo();
MethodNode methodNode = methodNodes.get(methodId);
MethodNode methodNode = query(methodPre + methodId, MethodNode.class);
rootInfo.setId(methodNode.getId());
rootInfo.setName(methodNode.getName());
rootInfo.setClassName(methodNode.getClassName());
rootInfo.setMethodName(methodNode.getMethodName());
rootInfo.setMethodType(methodNode.getMethodType());
rootInfo.setRouteName(methodNode.getRouteName());
MethodRelation methodRelation = methodRelations.values().stream().filter(relation -> relation.getTargetId().equals(methodId)).findFirst().get();
List<MethodRelation> methodRelationList = searchList(methodRelationPre, MethodRelation.class);
MethodRelation methodRelation = methodRelationList.stream().filter(relation -> relation.getTargetId().equals(methodId)).findFirst().get();
rootInfo.setValue(methodRelation.getAvgRunTime());
rootInfo.setAvgRunTime(methodRelation.getAvgRunTime());
rootInfo.setMaxRunTime(methodRelation.getMaxRunTime());
@ -374,4 +395,42 @@ public class RedisBase implements GraphService {
}
private void insert(String key, Object o) {
redisTemplate.opsForValue().set(key, toJson(o));
}
private <T> T query(String key, Class<T> c) {
Object o = redisTemplate.opsForValue().get(key);
if (o == null) {
return null;
}
return toObject(o.toString(), c);
}
private <T> List<T> searchList(String pre, Class<T> c) {
List<T> res = new ArrayList<>();
Set<String> keys = redisTemplate.keys(pre + "*");
if (keys == null || keys.size() == 0) {
return res;
}
List<String> objects = redisTemplate.opsForValue().multiGet(keys);
if (objects == null || objects.size() == 0) {
return res;
}
for (Object object : objects) {
res.add(toObject(object.toString(), c));
}
return res;
}
private String toJson(Object o) {
return JSON.toJSONString(o);
}
private <T> T toObject(String str, Class<T> c) {
return JSONObject.parseObject(str, c);
}
}

11
src/main/java/cn/langpy/kotime/util/Context.java
View File

@ -3,6 +3,7 @@ package cn.langpy.kotime.util;
import cn.langpy.kotime.config.DefaultConfig;
import cn.langpy.kotime.handler.InvokedHandler;
import cn.langpy.kotime.service.GraphService;
import org.springframework.data.redis.core.StringRedisTemplate;
import javax.sql.DataSource;
import java.util.ArrayList;
@ -16,6 +17,7 @@ public class Context {
private static DefaultConfig config;
private static List<InvokedHandler> invokedHandlers;
private static DataSource dataSource;
private static StringRedisTemplate stringRedisTemplate;
private static GraphService saver;
static {
@ -26,6 +28,14 @@ public class Context {
invokedHandlers = new ArrayList<>();
}
public static StringRedisTemplate getStringRedisTemplate() {
return stringRedisTemplate;
}
public static void setStringRedisTemplate(StringRedisTemplate stringRedisTemplate) {
Context.stringRedisTemplate = stringRedisTemplate;
}
public static void setConfig(DefaultConfig koTimeConfig) {
config = koTimeConfig;
}
@ -50,6 +60,7 @@ public class Context {
Context.dataSource = dataSource;
}
public static GraphService getSaver() {
return saver;
}

30
src/main/java/cn/langpy/kotime/util/KoUtil.java
View File

@ -1,5 +1,8 @@
package cn.langpy.kotime.util;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import javax.sql.DataSource;
import java.util.*;
@ -8,6 +11,8 @@ public class KoUtil {
private final static List<Integer> choices = randomSecretIndexs();
private static Map<String,Object> caches = new HashMap<>();
/**
* nothing to introduce for this, that everyone knows!
*/
@ -39,24 +44,29 @@ public class KoUtil {
* set a Datasource for saving of kotime data
* note: this Datasource will not affect project's datasource
*/
public void setDataSource(DataSource dataSource) {
//I will write it, and do not rush me!
public static void setDataSource(DataSource dataSource) {
caches.put("dataSource",dataSource);
}
public static DataSource getDataSource() {
return (DataSource)caches.get("dataSource");
}
/**
* set a RedisTemplate for saving of kotime data
* note: you can choose one between setRedisTemplate and setJedisPool to save data
*/
public void setRedisTemplate() {
//I will write it, and do not rush me!
public static void setStringRedisTemplate(RedisTemplate redisTemplate) {
caches.put("redisTemplate",redisTemplate);
}
/**
* set a RedisTemplate for saving of kotime data
* note: you can choose one between setRedisTemplate and setJedisPool to save data
*/
public void setJedisPool() {
//I will write it, and do not rush me!
public static StringRedisTemplate getStringRedisTemplate() {
return (StringRedisTemplate)caches.get("redisTemplate");
}
public static void clearCaches() {
caches.clear();
}