miduo_server/serverlogic/src/main/java/com/ljsd/DbClear.java

56 lines
2.1 KiB
Java

package com.ljsd;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientURI;
import org.springframework.data.redis.connection.jedis.JedisConnectionFactory;
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());
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());
System.out.println("the mongo url=" + property);
System.out.println("mongo清库完成");
}
public static void redisDbClear(Properties properties){
JedisConnectionFactory connectionFactory = new JedisConnectionFactory();
//设置host
String hostName = properties.getProperty("spring.redis.host");
connectionFactory.setHostName(hostName);
//设置端口
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();
//实例化
StringRedisTemplate template = new StringRedisTemplate(connectionFactory);
Set<String> keys = template.keys(properties.get("server.id") + "*");
template.delete(keys);
System.out.println("redis清库完成");
}
}