Skip to content

Commit

Permalink
Java：MultiDataSource 新增 Redis 缓存
Browse files Browse the repository at this point in the history
  • Loading branch information
@TommyLemon
TommyLemon committed Jan 5, 2023
1 parent 6208b75 commit 060a10e
Show file tree
Hide file tree
Showing 3 changed files with 99 additions and 13 deletions.
19 changes: 19 additions & 0 deletions APIJSON-Java-Server/APIJSONBoot-MultiDataSource/pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
<java.version>1.8</java.version>
<version>2.5.13</version>
</properties>

<dependencies>
Expand Down Expand Up @@ -142,6 +143,24 @@
<version>1.2.9</version>
</dependency>

<dependency>
<groupId>redis.clients</groupId>
<artifactId>jedis</artifactId>
<version>3.9.0</version>
</dependency>

<dependency>
<groupId>org.springframework.data</groupId>
<artifactId>spring-data-redis</artifactId>
<version>2.4.0</version>
<exclusions>
<exclusion>
<groupId>redis.clients</groupId>
<artifactId>jedis</artifactId>
</exclusion>
</exclusions>
</dependency>

</dependencies>

<build>
Expand Down
92 changes: 79 additions & 13 deletions ...N-Java-Server/APIJSONBoot-MultiDataSource/src/main/java/apijson/demo/DemoSQLExecutor.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,21 +14,35 @@

package apijson.demo;

import apijson.JSON;
import apijson.RequestMethod;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson.support.spring.FastJsonRedisSerializer;
import com.vesoft.nebula.jdbc.impl.NebulaDriver;
import com.zaxxer.hikari.HikariDataSource;

import java.io.Serializable;
import java.sql.Connection;
import java.sql.DriverManager;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.TimeUnit;

import javax.sql.DataSource;

import apijson.Log;
import apijson.boot.DemoApplication;
import apijson.framework.APIJSONSQLExecutor;
import apijson.orm.SQLConfig;
import org.springframework.data.redis.connection.RedisStandaloneConfiguration;
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.serializer.GenericToStringSerializer;
import org.springframework.data.redis.serializer.StringRedisSerializer;

import static apijson.framework.APIJSONConstant.PRIVACY_;
import static apijson.framework.APIJSONConstant.USER_;


/**SQL 执行器，支持连接池及多数据源
Expand All @@ -38,19 +52,71 @@
public class DemoSQLExecutor extends APIJSONSQLExecutor {
public static final String TAG = "DemoSQLExecutor";

// Redis 缓存 <<<<<<<<<<<<<<<<<<<<<<<
public static final RedisTemplate<String, String> REDIS_TEMPLATE;
static {
REDIS_TEMPLATE = new RedisTemplate<>();
REDIS_TEMPLATE.setConnectionFactory(new JedisConnectionFactory(new RedisStandaloneConfiguration("127.0.0.1", 6379)));
REDIS_TEMPLATE.setKeySerializer(new StringRedisSerializer());
REDIS_TEMPLATE.setHashValueSerializer(new GenericToStringSerializer<>(Serializable.class));
REDIS_TEMPLATE.setValueSerializer(new GenericToStringSerializer<>(Serializable.class));
// REDIS_TEMPLATE.setValueSerializer(new FastJsonRedisSerializer<List<JSONObject>>(List.class));
REDIS_TEMPLATE.afterPropertiesSet();
}

// 可重写以下方法，支持 Redis 等单机全局缓存或分布式缓存
// @Override
// public List<JSONObject> getCache(String sql, int type) {
// return super.getCache(sql, type);
// }
// @Override
// public synchronized void putCache(String sql, List<JSONObject> list, int type) {
// super.putCache(sql, list, type);
// }
// @Override
// public synchronized void removeCache(String sql, int type) {
// super.removeCache(sql, type);
// }
@Override
public List<JSONObject> getCache(String sql, SQLConfig config) {
List<JSONObject> list = super.getCache(sql, config);
if (list == null) {
list = JSON.parseArray(REDIS_TEMPLATE.opsForValue().get(sql), JSONObject.class);
}
return list;
}
@Override
public synchronized void putCache(String sql, List<JSONObject> list, SQLConfig config) {
super.putCache(sql, list, config);
if (config != null && config.isMain()) {
if (config.isExplain() || RequestMethod.isHeadMethod(config.getMethod(), true)) {
REDIS_TEMPLATE.opsForValue().set(sql, JSON.toJSONString(list), 10*60, TimeUnit.SECONDS);
} else {
String table = config.getTable();
REDIS_TEMPLATE.opsForValue().set(sql, JSON.toJSONString(list), USER_.equals(table) || PRIVACY_.equals(table) ? 10*60 : 60, TimeUnit.SECONDS);
}
}
}
@Override
public synchronized void removeCache(String sql, SQLConfig config) {
super.removeCache(sql, config);
if (config.getMethod() == RequestMethod.DELETE) { // 避免缓存击穿
REDIS_TEMPLATE.expire(sql, 60, TimeUnit.SECONDS);
} else {
REDIS_TEMPLATE.delete(sql);
}
}

@Override
public JSONObject execute(SQLConfig config, boolean unknownType) throws Exception {
JSONObject result = super.execute(config, unknownType);
RequestMethod method = config.getMethod();
if (method == RequestMethod.POST) { // 没必要，直接查就行了
// Object id = result.get(config.getIdKey());
// Object idIn = result.get(config.getIdKey() + "[]");
// SQLConfig cacheConfig = APIJSONRouterApplication.DEFAULT_APIJSON_CREATOR.createSQLConfig();
// cacheConfig.setMethod(RequestMethod.GET);
//
}
else if (method == RequestMethod.PUT || method == RequestMethod.DELETE) { // RequestMethod.isQueryMethod(method) == false) {
config.setMethod(RequestMethod.GET);
boolean isPrepared = config.isPrepared();
removeCache(config.getSQL(false), config);
config.setPrepared(isPrepared);
config.setMethod(method);
}
return result;
}

// Redis 缓存 >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>

// 适配连接池，如果这里能拿到连接池的有效 Connection，则 SQLConfig 不需要配置 dbVersion, dbUri, dbAccount, dbPassword
@Override
Expand Down
1 change: 1 addition & 0 deletions APIJSON-Java-Server/APIJSONBoot-MultiDataSource/src/main/resources/static/js/main.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3592,6 +3592,7 @@
'[]': {
'count': this.testCaseCount || 100, //200 条测试直接卡死 0,
'page': this.testCasePage || 0,
'join': '@/TestRecord,@/Script:pre,@/Script:post',
'Document': {
'@order': 'version-,date-',
'userId': this.User.id,
Expand Down

2 comments on commit 060a10e

@TommyLemon
Copy link
Member Author

@TommyLemon TommyLemon commented on 060a10e Jan 5, 2023 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这段可以封装到 APIJSONORM 中的父类方法
060a10e#diff-8dfe5875f3d4605d5cbe632993b210e70ebe5111367746fe1bc6449338841087R109-R115
image

@TommyLemon
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

已完成

Please sign in to comment.