@@ -5,11 +5,12 @@ import org.springframework.stereotype.Repository;
|
|||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.AuthToken;
|
||||||
|
|
||||||
import java.time.LocalDateTime;
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.List;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
|
|
||||||
@Repository
|
@Repository
|
||||||
public interface AuthTokenRepository extends JpaRepository<AuthToken, Long> {
|
public interface AuthTokenRepository extends JpaRepository<AuthToken, Long> {
|
||||||
Optional<AuthToken> findByLastUsedAtAndExpiresAtBefore(LocalDateTime lastUsedAt, LocalDateTime expiresAtBefore);
|
List<AuthToken> findByExpiresAtBefore(LocalDateTime expiresAtBefore);
|
||||||
Optional<AuthToken> findByLastUsedAtBefore(LocalDateTime lastUsedAt);
|
Optional<AuthToken> findByLastUsedAtBefore(LocalDateTime lastUsedAt);
|
||||||
Optional<AuthToken> findByToken(String token);
|
Optional<AuthToken> findByToken(String token);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -92,6 +92,7 @@ public class AuthService {
|
|||||||
AuthToken authToken = authTokenRepository.findByToken(token)
|
AuthToken authToken = authTokenRepository.findByToken(token)
|
||||||
.orElseThrow(() -> new NotFoundAuthToken("Token not found"));
|
.orElseThrow(() -> new NotFoundAuthToken("Token not found"));
|
||||||
authToken.setLastUsedAt(LocalDateTime.now());
|
authToken.setLastUsedAt(LocalDateTime.now());
|
||||||
|
authToken.setExpiresAt(LocalDateTime.now().plusHours(1));
|
||||||
|
|
||||||
authTokenRepository.save(authToken);
|
authTokenRepository.save(authToken);
|
||||||
|
|
||||||
|
|||||||
@@ -25,14 +25,13 @@ public class CleanupTokenSessionsService {
|
|||||||
|
|
||||||
@Transactional
|
@Transactional
|
||||||
@Scheduled(fixedDelay = 30000)
|
@Scheduled(fixedDelay = 30000)
|
||||||
public void cleanupExpiredFiles() {
|
public void cleanupExpiredTokens() {
|
||||||
LocalDateTime hourBefore = LocalDateTime.now().minusHours(authTokenLife);
|
LocalDateTime now = LocalDateTime.now();
|
||||||
|
|
||||||
List<AuthToken> tokensForDelete = authTokenRepository.findByLastUsedAtBefore(hourBefore).stream().toList();
|
List<AuthToken> expiredTokens = authTokenRepository.findByExpiresAtBefore(now);
|
||||||
List<AuthToken> tokensDeleteWithNullUpdate = authTokenRepository.findByLastUsedAtAndExpiresAtBefore(
|
|
||||||
null, hourBefore).stream().toList();
|
|
||||||
|
|
||||||
authTokenRepository.deleteAll(tokensForDelete);
|
if (!expiredTokens.isEmpty()) {
|
||||||
authTokenRepository.deleteAll(tokensDeleteWithNullUpdate);
|
authTokenRepository.deleteAll(expiredTokens);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user