Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/repository/server_timing_redis.go

@@ -0,0 +1,39 @@
+package repository
+
+import (
+	"context"
+	"time"
+
+	"github.com/Wei-Shaw/sub2api/internal/pkg/servertiming"
+	"github.com/redis/go-redis/v9"
+)
+
+type serverTimingRedisHook struct{}
+
+func (serverTimingRedisHook) DialHook(next redis.DialHook) redis.DialHook {
+	return next
+}
+
+func (serverTimingRedisHook) ProcessHook(next redis.ProcessHook) redis.ProcessHook {
+	return func(ctx context.Context, cmd redis.Cmder) error {
+		if !servertiming.Active(ctx) {
+			return next(ctx, cmd)
+		}
+		startedAt := time.Now()
+		err := next(ctx, cmd)
+		servertiming.Record(ctx, servertiming.MetricRedis, startedAt, time.Now(), 1)
+		return err
+	}
+}
+
+func (serverTimingRedisHook) ProcessPipelineHook(next redis.ProcessPipelineHook) redis.ProcessPipelineHook {
+	return func(ctx context.Context, cmds []redis.Cmder) error {
+		if !servertiming.Active(ctx) {
+			return next(ctx, cmds)
+		}
+		startedAt := time.Now()
+		err := next(ctx, cmds)
+		servertiming.Record(ctx, servertiming.MetricRedis, startedAt, time.Now(), len(cmds))
+		return err
+	}
+}