本地清库脚本修改
parent
e0b8ebf56c
commit
532cb834e5
|
@ -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清库完成");
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue