diff --git a/serverlogic/src/main/java/com/ljsd/DbClear.java b/serverlogic/src/main/java/com/ljsd/DbClear.java index 48a399f3b..b818cb876 100644 --- a/serverlogic/src/main/java/com/ljsd/DbClear.java +++ b/serverlogic/src/main/java/com/ljsd/DbClear.java @@ -1,7 +1,5 @@ package com.ljsd; - - import com.mongodb.MongoClient; import com.mongodb.MongoClientURI; import org.springframework.data.redis.connection.jedis.JedisConnectionFactory; @@ -10,34 +8,48 @@ import org.springframework.data.redis.core.StringRedisTemplate; import java.io.*; import java.net.URL; import java.util.Properties; +import java.util.Set; +/** + * 本地清库脚本 + */ public class DbClear { public static void main(String[] args) throws IOException { Properties properties = new Properties(); URL resource = DbClear.class.getClassLoader().getResource( "application.properties"); File file = new File(resource.getFile()); - //BufferedInputStream bufferedInputStream = new BufferedInputStream(DbClear.class.getResourceAsStream("application.properties")); System.out.println(file.getAbsolutePath()); properties.load(new FileReader(file)); + mongoDbClear(properties); + redisDbClear(properties); + } + + public static void mongoDbClear(Properties properties){ String property = properties.getProperty("spring.data.mongodb.uri"); MongoClientURI mongoClientURI = new MongoClientURI(property); MongoClient mongoClient = new MongoClient(mongoClientURI); mongoClient.dropDatabase(mongoClientURI.getDatabase()); - StringRedisTemplate redisTemplate = getRedisTemplate(); - redisTemplate.delete(redisTemplate.keys(properties.get("server.id")+"*")); System.out.println("the mongo url=" + property); + System.out.println("mongo清库完成"); } - public static StringRedisTemplate getRedisTemplate(){ + public static void redisDbClear(Properties properties){ JedisConnectionFactory connectionFactory = new JedisConnectionFactory(); //设置host - connectionFactory.setHostName("60.1.1.32"); + String hostName = properties.getProperty("spring.redis.host"); + connectionFactory.setHostName(hostName); //设置端口 - connectionFactory.setPort(6379); - connectionFactory.setPassword("ysj666"); + String port = properties.getProperty("spring.redis.port"); + connectionFactory.setPort(Integer.parseInt(port)); + // 设置密码 + String password = properties.getProperty("spring.redis.password"); + connectionFactory.setPassword(password); //初始化connectionFactory connectionFactory.afterPropertiesSet(); //实例化 - return new StringRedisTemplate(connectionFactory); + StringRedisTemplate template = new StringRedisTemplate(connectionFactory); + Set keys = template.keys(properties.get("server.id") + "*"); + template.delete(keys); + System.out.println("redis清库完成"); } }