import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.scheduling.annotation.EnableScheduling;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
@EnableScheduling
@Component
public class FTPLimitLoginToOneDevice {
private static final int MAX_ALLOWED_DEVICES = 1;
@Resource
private RedisTemplate<String, Object> redisTemplate;
public boolean checkLogin(String ftpId) {
Long count = (Long) redisTemplate.opsForHash().increment(ftpId, "count", 1);
if (count > MAX_ALLOWED_DEVICES) {
Long currentCount = (Long) redisTemplate.opsForHash().get(ftpId, "count");
if (currentCount!= null) {
redisTemplate.opsForHash().put(ftpId, "count", currentCount - 1);
}
return false;
}
return true;
}
public void releaseLogin(String ftpId) {
Long currentCount = (Long) redisTemplate.opsForHash().get(ftpId, "count");
if (currentCount!= null && currentCount > 0) {
redisTemplate.opsForHash().put(ftpId, "count", currentCount - 1);
}
}
@Scheduled(fixedDelay = 10000)
public void checkReleasedConnections() {
for (String key : redisTemplate.keys("*")) {
Long count = (Long) redisTemplate.opsForHash().get(key, "count");
if (count <= 0) {
redisTemplate.delete(key);
}
}
}
public static void main(String[] args) {
FTPLimitLoginToOneDevice limitLogin = new FTPLimitLoginToOneDevice();
String ftpId = "user1";
boolean canLogin = limitLogin.checkLogin(ftpId);
if (canLogin) {
System.out.println("登录成功");
} else {
System.out.println("登录失败,已有设备登录");
}
}
}