첫 커밋: 로컬 프로젝트 업로드
This commit is contained in:
103
baron-sso/backend/internal/pagination/cursor.go
Normal file
103
baron-sso/backend/internal/pagination/cursor.go
Normal file
@@ -0,0 +1,103 @@
|
||||
package pagination
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type Cursor struct {
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
ID string `json:"id"`
|
||||
}
|
||||
|
||||
func Encode(timestamp time.Time, id string) string {
|
||||
if timestamp.IsZero() || strings.TrimSpace(id) == "" {
|
||||
return ""
|
||||
}
|
||||
payload, err := json.Marshal(Cursor{
|
||||
Timestamp: timestamp.UTC(),
|
||||
ID: id,
|
||||
})
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
return base64.RawURLEncoding.EncodeToString(payload)
|
||||
}
|
||||
|
||||
func Decode(raw string) (*Cursor, error) {
|
||||
raw = strings.TrimSpace(raw)
|
||||
if raw == "" {
|
||||
return nil, nil
|
||||
}
|
||||
decoded, err := base64.RawURLEncoding.DecodeString(raw)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var cursor Cursor
|
||||
if err := json.Unmarshal(decoded, &cursor); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if cursor.Timestamp.IsZero() || strings.TrimSpace(cursor.ID) == "" {
|
||||
return nil, errors.New("invalid cursor")
|
||||
}
|
||||
return &cursor, nil
|
||||
}
|
||||
|
||||
func ComesAfter(timestamp time.Time, id string, cursor *Cursor) bool {
|
||||
if cursor == nil {
|
||||
return true
|
||||
}
|
||||
if timestamp.Before(cursor.Timestamp) {
|
||||
return true
|
||||
}
|
||||
return timestamp.Equal(cursor.Timestamp) && id < cursor.ID
|
||||
}
|
||||
|
||||
func SortByKeyDesc[T any](items []T, key func(T) (time.Time, string)) {
|
||||
sort.SliceStable(items, func(i, j int) bool {
|
||||
leftTime, leftID := key(items[i])
|
||||
rightTime, rightID := key(items[j])
|
||||
if !leftTime.Equal(rightTime) {
|
||||
return leftTime.After(rightTime)
|
||||
}
|
||||
return leftID > rightID
|
||||
})
|
||||
}
|
||||
|
||||
func PageByCursor[T any](items []T, limit int, cursorRaw string, key func(T) (time.Time, string)) ([]T, string, error) {
|
||||
cursor, err := Decode(cursorRaw)
|
||||
if err != nil {
|
||||
return nil, "", err
|
||||
}
|
||||
filtered := make([]T, 0, len(items))
|
||||
for _, item := range items {
|
||||
timestamp, id := key(item)
|
||||
if ComesAfter(timestamp, id, cursor) {
|
||||
filtered = append(filtered, item)
|
||||
}
|
||||
}
|
||||
if len(filtered) <= limit {
|
||||
return filtered, "", nil
|
||||
}
|
||||
page := filtered[:limit]
|
||||
lastTimestamp, lastID := key(page[len(page)-1])
|
||||
return page, Encode(lastTimestamp, lastID), nil
|
||||
}
|
||||
|
||||
func ApplyCreatedAtIDCursor(db *gorm.DB, cursor *Cursor, createdAtColumn string, idColumn string) *gorm.DB {
|
||||
if cursor == nil {
|
||||
return db
|
||||
}
|
||||
return db.Where(
|
||||
createdAtColumn+" < ? OR ("+createdAtColumn+" = ? AND "+idColumn+" < ?)",
|
||||
cursor.Timestamp,
|
||||
cursor.Timestamp,
|
||||
cursor.ID,
|
||||
)
|
||||
}
|
||||
142
baron-sso/backend/internal/pagination/cursor_test.go
Normal file
142
baron-sso/backend/internal/pagination/cursor_test.go
Normal file
@@ -0,0 +1,142 @@
|
||||
package pagination
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type testItem struct {
|
||||
id string
|
||||
createdAt time.Time
|
||||
}
|
||||
|
||||
func TestPageByCursorReturnsNextCursorAndNextPage(t *testing.T) {
|
||||
now := time.Date(2026, 5, 13, 8, 0, 0, 0, time.UTC)
|
||||
items := []testItem{
|
||||
{id: "c", createdAt: now},
|
||||
{id: "b", createdAt: now.Add(-time.Minute)},
|
||||
{id: "a", createdAt: now.Add(-2 * time.Minute)},
|
||||
}
|
||||
key := func(item testItem) (time.Time, string) {
|
||||
return item.createdAt, item.id
|
||||
}
|
||||
|
||||
firstPage, nextCursor, err := PageByCursor(items, 2, "", key)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, firstPage, 2)
|
||||
require.NotEmpty(t, nextCursor)
|
||||
|
||||
secondPage, nextCursor, err := PageByCursor(items, 2, nextCursor, key)
|
||||
require.NoError(t, err)
|
||||
require.Len(t, secondPage, 1)
|
||||
require.Empty(t, nextCursor)
|
||||
require.Equal(t, "a", secondPage[0].id)
|
||||
}
|
||||
|
||||
func TestSortByKeyDescUsesIDAsTieBreaker(t *testing.T) {
|
||||
now := time.Date(2026, 5, 13, 8, 0, 0, 0, time.UTC)
|
||||
items := []testItem{
|
||||
{id: "a", createdAt: now},
|
||||
{id: "c", createdAt: now},
|
||||
{id: "b", createdAt: now},
|
||||
}
|
||||
|
||||
SortByKeyDesc(items, func(item testItem) (time.Time, string) {
|
||||
return item.createdAt, item.id
|
||||
})
|
||||
|
||||
require.Equal(t, []string{"c", "b", "a"}, []string{
|
||||
items[0].id,
|
||||
items[1].id,
|
||||
items[2].id,
|
||||
})
|
||||
}
|
||||
|
||||
func TestEncodeReturnsEmptyForInvalidInput(t *testing.T) {
|
||||
require.Empty(t, Encode(time.Time{}, "id"))
|
||||
require.Empty(t, Encode(time.Now(), " "))
|
||||
}
|
||||
|
||||
func TestDecodeRejectsInvalidCursor(t *testing.T) {
|
||||
t.Run("blank cursor is nil", func(t *testing.T) {
|
||||
cursor, err := Decode(" ")
|
||||
require.NoError(t, err)
|
||||
require.Nil(t, cursor)
|
||||
})
|
||||
|
||||
t.Run("invalid base64 returns error", func(t *testing.T) {
|
||||
cursor, err := Decode("not base64")
|
||||
require.Error(t, err)
|
||||
require.Nil(t, cursor)
|
||||
})
|
||||
|
||||
t.Run("invalid json returns error", func(t *testing.T) {
|
||||
raw := base64.RawURLEncoding.EncodeToString([]byte("{invalid"))
|
||||
cursor, err := Decode(raw)
|
||||
require.Error(t, err)
|
||||
require.Nil(t, cursor)
|
||||
})
|
||||
|
||||
t.Run("missing timestamp returns invalid cursor", func(t *testing.T) {
|
||||
raw := base64.RawURLEncoding.EncodeToString([]byte(`{"id":"abc"}`))
|
||||
cursor, err := Decode(raw)
|
||||
require.EqualError(t, err, "invalid cursor")
|
||||
require.Nil(t, cursor)
|
||||
})
|
||||
|
||||
t.Run("missing id returns invalid cursor", func(t *testing.T) {
|
||||
raw := base64.RawURLEncoding.EncodeToString([]byte(`{"timestamp":"2026-05-29T00:00:00Z"}`))
|
||||
cursor, err := Decode(raw)
|
||||
require.EqualError(t, err, "invalid cursor")
|
||||
require.Nil(t, cursor)
|
||||
})
|
||||
}
|
||||
|
||||
func TestComesAfter(t *testing.T) {
|
||||
now := time.Date(2026, 5, 29, 8, 0, 0, 0, time.UTC)
|
||||
cursor := &Cursor{Timestamp: now, ID: "m"}
|
||||
|
||||
require.True(t, ComesAfter(now, "id", nil))
|
||||
require.True(t, ComesAfter(now.Add(-time.Second), "z", cursor))
|
||||
require.True(t, ComesAfter(now, "a", cursor))
|
||||
require.False(t, ComesAfter(now.Add(time.Second), "a", cursor))
|
||||
require.False(t, ComesAfter(now, "z", cursor))
|
||||
}
|
||||
|
||||
func TestPageByCursorReturnsDecodeError(t *testing.T) {
|
||||
items := []testItem{{id: "a", createdAt: time.Now()}}
|
||||
|
||||
page, nextCursor, err := PageByCursor(items, 10, "not base64", func(item testItem) (time.Time, string) {
|
||||
return item.createdAt, item.id
|
||||
})
|
||||
|
||||
require.Error(t, err)
|
||||
require.Nil(t, page)
|
||||
require.Empty(t, nextCursor)
|
||||
}
|
||||
|
||||
func TestApplyCreatedAtIDCursor(t *testing.T) {
|
||||
db, err := gorm.Open(postgres.New(postgres.Config{
|
||||
DSN: "host=localhost user=test dbname=test sslmode=disable",
|
||||
}), &gorm.Config{
|
||||
DryRun: true,
|
||||
DisableAutomaticPing: true,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
require.Same(t, db, ApplyCreatedAtIDCursor(db, nil, "created_at", "id"))
|
||||
|
||||
cursor := &Cursor{
|
||||
Timestamp: time.Date(2026, 5, 29, 8, 0, 0, 0, time.UTC),
|
||||
ID: "cursor-id",
|
||||
}
|
||||
query := ApplyCreatedAtIDCursor(db.Model(&struct{}{}), cursor, "created_at", "id").Find(&[]struct{}{})
|
||||
|
||||
require.Contains(t, query.Statement.SQL.String(), "created_at < $1 OR (created_at = $2 AND id < $3)")
|
||||
require.Equal(t, []any{cursor.Timestamp, cursor.Timestamp, cursor.ID}, query.Statement.Vars)
|
||||
}
|
||||
Reference in New Issue
Block a user