init push

backend/repo/cache/geo.go

@@ -0,0 +1,91 @@
+package cache
+
+import (
+	"context"
+	"fmt"
+	"strconv"
+	"time"
+
+	"github.com/chaitin/panda-wiki/domain"
+	"github.com/chaitin/panda-wiki/log"
+	"github.com/chaitin/panda-wiki/store/cache"
+	"github.com/chaitin/panda-wiki/store/pg"
+	"github.com/chaitin/panda-wiki/utils"
+)
+
+type GeoRepo struct {
+	cache  *cache.Cache
+	db     *pg.DB
+	logger *log.Logger
+}
+
+func NewGeoCache(cache *cache.Cache, db *pg.DB, logger *log.Logger) *GeoRepo {
+	return &GeoRepo{
+		cache:  cache,
+		db:     db,
+		logger: logger.WithModule("repo.cache.geo"),
+	}
+}
+
+func (r *GeoRepo) SetGeo(ctx context.Context, kbID, field string) error {
+	now := time.Now()
+	key := fmt.Sprintf("geo:%s:%s", kbID, now.Format("2006-01-02-15"))
+
+	// First try to increment the field
+	result := r.cache.HIncrBy(ctx, key, field, 1)
+	if result.Err() != nil {
+		return result.Err()
+	}
+
+	// If this is the first increment (value = 1), set expire
+	if result.Val() == 1 {
+		return r.cache.Expire(ctx, key, 25*time.Hour).Err()
+	}
+
+	return nil
+}
+
+func (r *GeoRepo) GetLast24HourGeo(ctx context.Context, kbID string) (map[string]int64, error) {
+	counts := make(map[string]int64)
+	now := time.Now()
+
+	// Get data for the last 24 hours
+	for i := 0; i < 24; i++ {
+		targetTime := now.Add(-time.Duration(i) * time.Hour)
+		key := fmt.Sprintf("geo:%s:%s", kbID, targetTime.Format("2006-01-02-15"))
+
+		values, err := r.cache.HGetAll(ctx, key).Result()
+		if err != nil {
+			return nil, fmt.Errorf("get geo count failed: %w", err)
+		}
+
+		for field, value := range values {
+			valueInt, err := strconv.ParseInt(value, 10, 64)
+			if err != nil {
+				return nil, fmt.Errorf("parse geo count failed: %w", err)
+			}
+			counts[field] += valueInt
+		}
+	}
+	return counts, nil
+}
+
+func (r *GeoRepo) GetGeoByHour(ctx context.Context, kbID string, startHour int64) (map[string]int64, error) {
+	counts := make(map[string]int64)
+
+	geoCounts := make([]domain.MapStrInt64, 0)
+	if err := r.db.WithContext(ctx).Model(&domain.StatPageHour{}).
+		Select("geo_count").
+		Where("kb_id = ?", kbID).
+		Where("hour >= ? and hour < ?", utils.GetTimeHourOffset(-startHour), utils.GetTimeHourOffset(-24)).
+		Pluck("geo_count", &geoCounts).Error; err != nil {
+		return nil, err
+	}
+	for i := range geoCounts {
+		for k, v := range geoCounts[i] {
+			counts[k] += v
+		}
+	}
+
+	return counts, nil
+}

backend/repo/cache/kb.go

@@ -0,0 +1,55 @@
+package cache
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+
+	"github.com/chaitin/panda-wiki/domain"
+	"github.com/chaitin/panda-wiki/store/cache"
+	"github.com/redis/go-redis/v9"
+)
+
+type KBRepo struct {
+	cache *cache.Cache
+}
+
+func NewKBRepo(cache *cache.Cache) *KBRepo {
+	return &KBRepo{cache: cache}
+}
+
+func (r *KBRepo) GetKB(ctx context.Context, kbID string) (*domain.KnowledgeBase, error) {
+	kbStr, err := r.cache.Get(ctx, kbID).Result()
+	if err != nil {
+		if errors.Is(err, redis.Nil) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	if kbStr == "" {
+		return nil, nil
+	}
+
+	var kb domain.KnowledgeBase
+	err = json.Unmarshal([]byte(kbStr), &kb)
+	if err != nil {
+		return nil, err
+	}
+	return &kb, nil
+}
+
+func (r *KBRepo) SetKB(ctx context.Context, kbID string, kb *domain.KnowledgeBase) error {
+	kbStr, err := json.Marshal(kb)
+	if err != nil {
+		return err
+	}
+	return r.cache.Set(ctx, kbID, kbStr, 0).Err()
+}
+
+func (r *KBRepo) DeleteKB(ctx context.Context, kbID string) error {
+	return r.cache.Del(ctx, kbID).Err()
+}
+
+func (r *KBRepo) ClearSession(ctx context.Context) error {
+	return r.cache.DeleteKeysWithPrefix(ctx, "session_")
+}

backend/repo/cache/provider.go

@@ -0,0 +1,13 @@
+package cache
+
+import (
+	"github.com/google/wire"
+
+	"github.com/chaitin/panda-wiki/store/cache"
+)
+
+var ProviderSet = wire.NewSet(
+	cache.NewCache,
+	NewKBRepo,
+	NewGeoCache,
+)