Test4

config/config.go

@@ -56,6 +56,7 @@ type Settings struct {
 	DevBotid               string   `yaml:"develop_bot_id"`
 	SandBoxMode            bool     `yaml:"sandbox_mode"`
 	Title                  string   `yaml:"title"`
+	HashID                 bool     `yaml:"hash_id"`
 }
 
 // LoadConfig 从文件中加载配置并初始化单例配置
@@ -389,6 +390,18 @@ func GetLotusValue() bool {
 	return instance.Settings.Lotus
 }
 
+// 获取HashID
+func GetHashIDValue() bool {
+	mu.Lock()
+	defer mu.Unlock()
+
+	if instance == nil {
+		mylog.Println("Warning: instance is nil when trying to get hashid value.")
+		return false
+	}
+	return instance.Settings.HashID
+}
+
 // 获取RemoveAt的值
 func GetRemoveAt() bool {
 	mu.Lock()

idmap/service.go

@@ -1,14 +1,18 @@
 package idmap
 
 import (
+	"crypto/md5"
 	"encoding/binary"
+	"encoding/hex"
 	"encoding/json"
 	"errors"
 	"fmt"
 	"io"
 	"log"
 	"net/http"
 	"net/url"
+	"strconv"
+	"strings"
 
 	"github.com/boltdb/bolt"
 	"github.com/hoshinonyaruko/gensokyo/config"
@@ -42,6 +46,38 @@ func InitializeDB() {
 func CloseDB() {
 	db.Close()
 }
+func generateRowID(id string, length int) (int64, error) {
+	// 计算MD5哈希值
+	hasher := md5.New()
+	hasher.Write([]byte(id))
+	hash := hex.EncodeToString(hasher.Sum(nil))
+
+	// 只保留数字
+	var digitsBuilder strings.Builder
+	for _, r := range hash {
+		if r >= '0' && r <= '9' {
+			digitsBuilder.WriteRune(r)
+		}
+	}
+	digits := digitsBuilder.String()
+
+	// 取出需要长度的数字
+	var rowIDStr string
+	if len(digits) >= length {
+		rowIDStr = digits[:length]
+	} else {
+		// 如果数字不足指定长度，返回错误
+		return 0, fmt.Errorf("not enough digits in MD5 hash")
+	}
+
+	// 将数字字符串转换为int64
+	rowID, err := strconv.ParseInt(rowIDStr, 10, 64)
+	if err != nil {
+		return 0, err
+	}
+
+	return rowID, nil
+}
 
 // 根据a储存b
 func StoreID(id string) (int64, error) {
@@ -56,19 +92,46 @@ func StoreID(id string) (int64, error) {
 			newRow = int64(binary.BigEndian.Uint64(existingRowBytes))
 			return nil
 		}
-
-		// 如果ID不存在，则为它分配一个新的行号
-		currentRowBytes := b.Get([]byte(CounterKey))
-		if currentRowBytes == nil {
-			newRow = 1
+		//写入虚拟值
+		if !config.GetHashIDValue() {
+			// 如果ID不存在，则为它分配一个新的行号 数字递增
+			currentRowBytes := b.Get([]byte(CounterKey))
+			if currentRowBytes == nil {
+				newRow = 1
+			} else {
+				currentRow := binary.BigEndian.Uint64(currentRowBytes)
+				newRow = int64(currentRow) + 1
+			}
 		} else {
-			currentRow := binary.BigEndian.Uint64(currentRowBytes)
-			newRow = int64(currentRow) + 1
+			// 生成新的行号
+			var err error
+			newRow, err = generateRowID(id, 9)
+			if err != nil {
+				return err
+			}
+			// 检查新生成的行号是否重复
+			rowKey := fmt.Sprintf("row-%d", newRow)
+			if b.Get([]byte(rowKey)) != nil {
+				// 如果行号重复，使用10位数字生成行号
+				newRow, err = generateRowID(id, 10)
+				if err != nil {
+					return err
+				}
+				rowKey = fmt.Sprintf("row-%d", newRow)
+				// 再次检查重复性，如果还是重复，则返回错误
+				if b.Get([]byte(rowKey)) != nil {
+					return fmt.Errorf("unable to find a unique row ID")
+				}
+			}
 		}
 
 		rowBytes := make([]byte, 8)
 		binary.BigEndian.PutUint64(rowBytes, uint64(newRow))
-		b.Put([]byte(CounterKey), rowBytes)
+		//写入递增值
+		if !config.GetHashIDValue() {
+			b.Put([]byte(CounterKey), rowBytes)
+		}
+		//真实对应虚拟 用来直接判断是否存在,并快速返回
 		b.Put([]byte(id), rowBytes)
 
 		reverseKey := fmt.Sprintf("row-%d", newRow)

mylog/mylog.go

@@ -84,14 +84,6 @@ func WsHandlerWithDependencies(c *gin.Context) {
 	// 输出新的 WebSocket 客户端连接信息
 	fmt.Println("新的webui用户已连接!")
 
-	// Start a goroutine for heartbeats
-	go func() {
-		for {
-			time.Sleep(5 * time.Second) // send a heartbeat every 30 seconds
-			client.conn.WriteMessage(websocket.PingMessage, nil)
-		}
-	}()
-
 	go client.writePump()
 	go client.readPump()
 

template/config_template.go

@@ -55,6 +55,7 @@ settings:
   develop_bot_id : "1234"           #开发者环境需自行获取botid 填入 用户请不要设置这两行...开发者调试用
   sandbox_mode : false              #默认false 如果你只希望沙箱频道使用,请改为true
   title : "Gensokyo © 2023 - Hoshinonyaruko"              #程序的标题 如果多个机器人 可根据标题区分
+  hash_id : false                   #使用hash来进行idmaps转换,可以让user_id不是123开始的递增值
 `
 const Logo = `
 '                                                                                                      

template/config_template.yml

@@ -46,4 +46,5 @@ settings:
   develop_access_token_dir : ""     #开发者测试环境access_token自定义获取地址 默认留空 请留空忽略
   develop_bot_id : "1234"           #开发者环境需自行获取botid 填入 用户请不要设置这两行...开发者调试用
   sandbox_mode : false              #默认false 如果你只希望沙箱频道使用,请改为true
-  title : "Gensokyo © 2023 - Hoshinonyaruko"              #程序的标题 如果多个机器人 可根据标题区分
+  title : "Gensokyo © 2023 - Hoshinonyaruko"              #程序的标题 如果多个机器人 可根据标题区分
+  hash_id : false                   #使用hash来进行idmaps转换,可以让user_id不是123开始的递增值