Tuas W06

internal/store/session_store.go

@@ -2,6 +2,7 @@ package store
 
 import (
 	"context"
+	"errors"
 	"sync"
 	"time"
 
@@ -9,6 +10,9 @@ import (
 	"github.com/jackc/pgx/v5"
 )
 
+// ErrSessionNotFound dikembalikan saat token sesi tidak ditemukan (atau sudah dihapus).
+var ErrSessionNotFound = errors.New("session not found")
+
 type SessionStore struct {
 	conn *pgx.Conn
 	mu   *sync.Mutex
@@ -45,16 +49,28 @@ func (s *SessionStore) Get(ctx context.Context, token string) (*models.Session,
 	query := `SELECT token, user_id, created_at FROM sessions WHERE token = $1`
 	err := s.conn.QueryRow(ctx, query, token).Scan(&sess.Token, &sess.UserID, &sess.CreatedAt)
 	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			return nil, ErrSessionNotFound
+		}
 		return nil, err
 	}
 	return &sess, nil
 }
 
+// Delete menghapus token sesi. Jika token tidak ditemukan (sudah dihapus atau
+// tidak pernah ada), ia mengembalikan ErrSessionNotFound tanpa panic — memanggil
+// Delete dua kali dengan token yang sama aman dan tidak menghapus token lain.
 func (s *SessionStore) Delete(ctx context.Context, token string) error {
 	s.mu.Lock()
 	defer s.mu.Unlock()
 
 	query := `DELETE FROM sessions WHERE token = $1`
-	_, err := s.conn.Exec(ctx, query, token)
-	return err
+	tag, err := s.conn.Exec(ctx, query, token)
+	if err != nil {
+		return err
+	}
+	if tag.RowsAffected() == 0 {
+		return ErrSessionNotFound
+	}
+	return nil
 }