Add scheduler to clean old sessions

config/config.go

@@ -22,6 +22,7 @@ const (
 	DefaultKeyFile                 = ""
 	DefaultCertDomain              = ""
 	DefaultCertCache               = "/tmp/cert_cache"
+	DefaultSessionCleanupFrequency = 24
 )
 
 // Config manages configuration parameters.

main.go

@@ -47,13 +47,15 @@ func run(cfg *config.Config, store *storage.Storage) {
 	pool := scheduler.NewWorkerPool(feedHandler, cfg.GetInt("WORKER_POOL_SIZE", config.DefaultWorkerPoolSize))
 	server := server.NewServer(cfg, store, pool, feedHandler)
 
-	scheduler.NewScheduler(
+	scheduler.NewFeedScheduler(
 		store,
 		pool,
 		cfg.GetInt("POLLING_FREQUENCY", config.DefaultPollingFrequency),
 		cfg.GetInt("BATCH_SIZE", config.DefaultBatchSize),
 	)
 
+	scheduler.NewSessionScheduler(store, config.DefaultSessionCleanupFrequency)
+
 	<-stop
 	logger.Info("Shutting down the server...")
 	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)

scheduler/scheduler.go

@@ -11,18 +11,30 @@ import (
 	"github.com/miniflux/miniflux/storage"
 )
 
-// NewScheduler starts a new scheduler that push jobs to a pool of workers.
-func NewScheduler(store *storage.Storage, workerPool *WorkerPool, frequency, batchSize int) {
+// NewFeedScheduler starts a new scheduler that push jobs to a pool of workers.
+func NewFeedScheduler(store *storage.Storage, workerPool *WorkerPool, frequency, batchSize int) {
 	go func() {
 		c := time.Tick(time.Duration(frequency) * time.Minute)
-		for now := range c {
+		for _ = range c {
 			jobs, err := store.NewBatch(batchSize)
 			if err != nil {
-				logger.Error("[Scheduler] %v", err)
+				logger.Error("[FeedScheduler] %v", err)
 			} else {
-				logger.Debug("[Scheduler:%v] => Pushing %d jobs", now, len(jobs))
+				logger.Debug("[FeedScheduler] Pushing %d jobs", len(jobs))
 				workerPool.Push(jobs)
 			}
 		}
 	}()
 }
+
+// NewSessionScheduler starts a new scheduler that clean old sessions.
+func NewSessionScheduler(store *storage.Storage, frequency int) {
+	go func() {
+		c := time.Tick(time.Duration(frequency) * time.Hour)
+		for _ = range c {
+			nbSessions := store.CleanOldSessions()
+			nbUserSessions := store.CleanOldUserSessions()
+			logger.Debug("[SessionScheduler] cleaned %d sessions and %d user sessions", nbSessions, nbUserSessions)
+		}
+	}()
+}

storage/session.go

@@ -75,3 +75,17 @@ func (s *Storage) FlushAllSessions() (err error) {
 
 	return nil
 }
+
+// CleanOldSessions removes sessions older than 30 days.
+func (s *Storage) CleanOldSessions() int64 {
+	query := `DELETE FROM sessions
+		WHERE id IN (SELECT id FROM sessions WHERE created_at < now() - interval '30 days')`
+
+	result, err := s.db.Exec(query)
+	if err != nil {
+		return 0
+	}
+
+	n, _ := result.RowsAffected()
+	return n
+}

storage/user_session.go

@@ -127,3 +127,17 @@ func (s *Storage) RemoveUserSessionByID(userID, sessionID int64) error {
 
 	return nil
 }
+
+// CleanOldUserSessions removes user sessions older than 30 days.
+func (s *Storage) CleanOldUserSessions() int64 {
+	query := `DELETE FROM user_sessions
+		WHERE id IN (SELECT id FROM user_sessions WHERE created_at < now() - interval '30 days')`
+
+	result, err := s.db.Exec(query)
+	if err != nil {
+		return 0
+	}
+
+	n, _ := result.RowsAffected()
+	return n
+}