NCBACK-27 #12

Merged
backdev merged 6 commits from NCBACK-27 into dev 2026-01-20 16:20:38 +08:00
3 changed files with 9 additions and 8 deletions
Showing only changes of commit 2b7a8b97e1 - Show all commits
@@ -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);
}
} }
} }