refactor: V3 版本重构已基本完成

api/go/utils/openai.go

@@ -0,0 +1,17 @@
+package utils
+
+import (
+	"fmt"
+	"github.com/pkoukk/tiktoken-go"
+)
+
+func CalcTokens(text string, model string) (int, error) {
+	encoding := tiktoken.MODEL_TO_ENCODING[model]
+	tke, err := tiktoken.GetEncoding(encoding)
+	if err != nil {
+		return 0, fmt.Errorf("getEncoding: %v", err)
+	}
+
+	token := tke.Encode(text, nil, nil)
+	return len(token), nil
+}

api/go/utils/param/param.go

@@ -0,0 +1,57 @@
+package param
+
+import (
+	"github.com/gin-gonic/gin"
+	"strconv"
+	"strings"
+)
+
+func GetTrim(c *gin.Context, key string) string {
+	return strings.TrimSpace(c.Query(key))
+}
+
+func GetInt(c *gin.Context, key string, defaultValue int) int {
+	return intValue(c.Query(key), defaultValue)
+}
+
+func PostInt(c *gin.Context, key string, defaultValue int) int {
+	return intValue(c.PostForm(key), defaultValue)
+}
+
+func intValue(str string, defaultValue int) int {
+	value, err := strconv.Atoi(str)
+	if err != nil {
+		return defaultValue
+	}
+	return value
+}
+
+func GetFloat(c *gin.Context, key string) float64 {
+	return floatValue(c.Query(key))
+}
+func PostFloat(c *gin.Context, key string) float64 {
+	return floatValue(c.PostForm(key))
+}
+
+func floatValue(str string) float64 {
+	value, err := strconv.ParseFloat(str, 64)
+	if err != nil {
+		return 0
+	}
+	return value
+}
+
+func GetBool(c *gin.Context, key string) bool {
+	return boolValue(c.Query(key))
+}
+func PostBool(c *gin.Context, key string) bool {
+	return boolValue(c.PostForm(key))
+}
+
+func boolValue(str string) bool {
+	value, err := strconv.ParseBool(str)
+	if err != nil {
+		return false
+	}
+	return value
+}

api/go/utils/resp/response.go

@@ -0,0 +1,28 @@
+package resp
+
+import (
+	"chatplus/core/types"
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+func SUCCESS(c *gin.Context, values ...interface{}) {
+	if values != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Data: values[0]})
+	} else {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Success})
+	}
+
+}
+
+func ERROR(c *gin.Context, messages ...string) {
+	if messages != nil {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: messages[0]})
+	} else {
+		c.JSON(http.StatusOK, types.BizVo{Code: types.Failed})
+	}
+}
+
+func NotAuth(c *gin.Context) {
+	c.JSON(http.StatusOK, types.BizVo{Code: types.NotAuthorized, Message: "Not Authorized"})
+}

api/go/utils/string.go

@@ -0,0 +1,155 @@
+package utils
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
+	"golang.org/x/crypto/sha3"
+	"math/rand"
+	"reflect"
+	"strings"
+	"time"
+)
+
+// RandString generate rand string with specified length
+func RandString(length int) string {
+	str := "0123456789abcdefghijklmnopqrstuvwxyz"
+	data := []byte(str)
+	var result []byte
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	for i := 0; i < length; i++ {
+		result = append(result, data[r.Intn(len(data))])
+	}
+	return string(result)
+}
+
+func RandomNumber(bit int) int {
+	rand.Seed(time.Now().UnixNano())
+	min := 1 // min value
+	max := 1 //max value
+	for i := 0; i < bit; i++ {
+		min = min * 10
+		max = max * 10
+	}
+	max = max * 10
+	return rand.Intn(max-min+1) + min
+}
+
+func ContainsStr(slice []string, item string) bool {
+	for _, e := range slice {
+		if e == item {
+			return true
+		}
+	}
+	return false
+}
+
+// Stamp2str 时间戳转字符串
+func Stamp2str(timestamp int64) string {
+	if timestamp == 0 {
+		return ""
+	}
+	return time.Unix(timestamp, 0).Format("2006-01-02 15:04:05")
+}
+
+// Str2stamp 字符串转时间戳
+func Str2stamp(str string) int64 {
+	layout := "2006-01-02 15:04:05"
+	t, err := time.Parse(layout, str)
+	if err != nil {
+		return 0
+	}
+	return t.Unix()
+}
+
+func GenPassword(pass string, salt string) string {
+	data := []byte(pass + salt)
+	hash := sha3.Sum256(data)
+	return fmt.Sprintf("%x", hash)
+}
+
+func JsonEncode(value interface{}) string {
+	bytes, err := json.Marshal(value)
+	if err != nil {
+		return ""
+	}
+	return string(bytes)
+}
+
+func JsonDecode(src string, dest interface{}) error {
+	return json.Unmarshal([]byte(src), dest)
+}
+
+// CopyObject 拷贝对象
+func CopyObject(src interface{}, dst interface{}) error {
+
+	srcType := reflect.TypeOf(src)
+	srcValue := reflect.ValueOf(src)
+	dstValue := reflect.ValueOf(dst).Elem()
+	reflect.TypeOf(dst)
+	for i := 0; i < srcType.NumField(); i++ {
+		field := srcType.Field(i)
+		value := dstValue.FieldByName(field.Name)
+		if !value.IsValid() {
+			continue
+		}
+		// 数据类型相同，直接赋值
+		v := srcValue.FieldByName(field.Name)
+		if value.Type() == field.Type {
+			value.Set(v)
+		} else {
+			// src data type is  string，dst data type is slice, map, struct
+			// use json decode the data
+			if field.Type.Kind() == reflect.String && (value.Type().Kind() == reflect.Struct ||
+				value.Type().Kind() == reflect.Map ||
+				value.Type().Kind() == reflect.Slice) {
+				pType := reflect.New(value.Type())
+				v2 := pType.Interface()
+				err := json.Unmarshal([]byte(v.String()), &v2)
+				if err == nil {
+					value.Set(reflect.ValueOf(v2).Elem())
+				}
+				// string to map, struct, slice
+			} else if (field.Type.Kind() == reflect.Struct ||
+				field.Type.Kind() == reflect.Map ||
+				field.Type.Kind() == reflect.Slice) && value.Type().Kind() == reflect.String {
+				ba, err := json.Marshal(v.Interface())
+				if err == nil {
+					value.Set(reflect.ValueOf(string(ba)))
+				}
+			} else { // 简单数据类型的强制类型转换
+				switch value.Kind() {
+				case reflect.Int:
+				case reflect.Int8:
+				case reflect.Int16:
+				case reflect.Int32:
+				case reflect.Int64:
+					value.SetInt(v.Int())
+					break
+				case reflect.Float32:
+				case reflect.Float64:
+					value.SetFloat(v.Float())
+					break
+				case reflect.Bool:
+					value.SetBool(v.Bool())
+					break
+				}
+			}
+
+		}
+	}
+
+	return nil
+}
+
+func Ip2Region(searcher *xdb.Searcher, ip string) string {
+	str, err := searcher.SearchByStr(ip)
+	if err != nil {
+		return ""
+	}
+	arr := strings.Split(str, "|")
+	if len(arr) < 3 {
+		return arr[0]
+	}
+	return fmt.Sprintf("%s-%s-%s", arr[0], arr[2], arr[3])
+}

api/go/utils/user.go

@@ -0,0 +1,37 @@
+package utils
+
+import (
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"errors"
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+func SetLoginUser(c *gin.Context, userId uint) error {
+	session := sessions.Default(c)
+	session.Set(types.SessionUserId, userId)
+	return session.Save()
+}
+
+func GetLoginUser(c *gin.Context, db *gorm.DB) (model.User, error) {
+	value, exists := c.Get(types.LoginUserCache)
+	if exists {
+		return value.(model.User), nil
+	}
+
+	session := sessions.Default(c)
+	userId := session.Get(types.SessionUserId)
+	if userId == nil {
+		return model.User{}, errors.New("user not login")
+	}
+
+	var user model.User
+	res := db.First(&user, userId)
+	// 更新缓存
+	if res.Error == nil {
+		c.Set(types.LoginUserCache, user)
+	}
+	return user, res.Error
+}