WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

2026-09-22 16:19:06 +08:00
commit f7fc992f57
3929 changed files with 1006607 additions and 0 deletions
@@ -0,0 +1,37 @@
+package stream
+
+import (
+	"os"
+	"strconv"
+	"time"
+
+	"github.com/Tencent/WeKnora/internal/types/interfaces"
+)
+
+// 流管理器类型
+const (
+	TypeMemory = "memory"
+	TypeRedis  = "redis"
+)
+
+// NewStreamManager 创建流管理器
+func NewStreamManager() (interfaces.StreamManager, error) {
+	switch os.Getenv("STREAM_MANAGER_TYPE") {
+	case TypeRedis:
+		db, err := strconv.Atoi(os.Getenv("REDIS_DB"))
+		if err != nil {
+			db = 0
+		}
+		ttl := time.Hour // 默认1小时
+		return NewRedisStreamManager(
+			os.Getenv("REDIS_ADDR"),
+			os.Getenv("REDIS_USERNAME"),
+			os.Getenv("REDIS_PASSWORD"),
+			db,
+			os.Getenv("REDIS_PREFIX"),
+			ttl,
+		)
+	default:
+		return NewMemoryStreamManager(), nil
+	}
+}