first commit

internal/pkg/session/session.go

@@ -0,0 +1,72 @@
+// Package session 提供会话存储能力。
+// 当前默认实现为进程内内存存储（无 Redis 环境），后续可替换为 Redis 实现。
+package session
+
+import (
+	"context"
+	"sync"
+	"time"
+
+	"portal-mini-go/internal/pkg/errors"
+)
+
+// Store 会话存储接口，便于后续切换为 Redis 等实现。
+type Store interface {
+	Set(ctx context.Context, key string, userID uint64, ttl time.Duration) error
+	Get(ctx context.Context, key string) (uint64, error)
+	Delete(ctx context.Context, key string) error
+}
+
+// MemoryStore 进程内会话存储，适用于无 Redis 的本地/测试环境。
+// 注意：多实例部署时各进程不共享，仅供单实例使用。
+type MemoryStore struct {
+	mu   sync.RWMutex
+	data map[string]memoryEntry
+	now  func() time.Time
+}
+
+type memoryEntry struct {
+	userID   uint64
+	expireAt time.Time
+}
+
+// NewMemoryStore 创建内存会话存储。
+func NewMemoryStore() *MemoryStore {
+	return &MemoryStore{
+		data: make(map[string]memoryEntry),
+		now:  time.Now,
+	}
+}
+
+// Set 写入会话，key 映射到 userID，ttl 后过期。
+func (m *MemoryStore) Set(_ context.Context, key string, userID uint64, ttl time.Duration) error {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.data[key] = memoryEntry{userID: userID, expireAt: m.now().Add(ttl)}
+	return nil
+}
+
+// Get 读取会话对应的 userID；不存在或已过期返回 ErrUnauthorized。
+func (m *MemoryStore) Get(_ context.Context, key string) (uint64, error) {
+	m.mu.RLock()
+	e, ok := m.data[key]
+	m.mu.RUnlock()
+	if !ok {
+		return 0, errors.ErrUnauthorized
+	}
+	if m.now().After(e.expireAt) {
+		m.mu.Lock()
+		delete(m.data, key)
+		m.mu.Unlock()
+		return 0, errors.ErrUnauthorized
+	}
+	return e.userID, nil
+}
+
+// Delete 删除会话。
+func (m *MemoryStore) Delete(_ context.Context, key string) error {
+	m.mu.Lock()
+	delete(m.data, key)
+	m.mu.Unlock()
+	return nil
+}