56 lines
2.1 KiB
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清库完成");
|
|
}
|
|
}
|