Files
quinn-bbc/src/test/java/com/quinn/test/RedisTest.java

55 lines
1.7 KiB
Java

package com.quinn.test;
import com.quinn.QuinnApplication;
import com.quinn.common.QuinnConstant;
import com.quinn.pojo.Source;
import com.quinn.service.SourceService;
import com.quinn.utils.RedisUtils;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.util.CollectionUtils;
import org.springframework.util.StringUtils;
import javax.annotation.Resource;
import java.util.List;
@RunWith(SpringRunner.class)
@SpringBootTest(classes={QuinnApplication.class})
public class RedisTest {
@Resource
SourceService sourceService;
@Resource
RedisUtils redisUtils;
@Test
public void updateViewTask(){
List<String> keys = redisUtils.scan(QuinnConstant.SOURCE_KEY + QuinnConstant.REDIS_PATTEN);
if (!CollectionUtils.isEmpty(keys)){
keys.forEach(x->{
Source source = sourceService.getById(getIdFromKey(x,QuinnConstant.SOURCE_KEY));
if (source != null && !StringUtils.isEmpty(source.getSourceName())){
source.setDownRecord(Integer.parseInt(redisUtils.get(x)));
sourceService.updateById(source);
redisUtils.del(x);
}
});
}
}
@Test
public void setTestKey(){
redisUtils.set(QuinnConstant.SOURCE_KEY+1,"123");
redisUtils.set(QuinnConstant.SOURCE_KEY+2,"456");
redisUtils.set(QuinnConstant.SOURCE_KEY+3,"789");
}
private String getIdFromKey(String key,String keyType){
return key.substring(keyType.length());
}
}