本地清库脚本修改

master_dev
grimm 2024-05-07 17:52:51 +08:00
parent e0b8ebf56c
commit 532cb834e5
1 changed files with 22 additions and 10 deletions

View File

@ -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<String> keys = template.keys(properties.get("server.id") + "*");
template.delete(keys);
System.out.println("redis清库完成");
}
}