diff --git a/README.md b/README.md
index aa3e0b2..ba3f6da 100644
--- a/README.md
+++ b/README.md
@@ -34,6 +34,8 @@
## 界面截图
+
+



diff --git a/common/image.go b/common/image.go
new file mode 100644
index 0000000..d80cbb7
--- /dev/null
+++ b/common/image.go
@@ -0,0 +1,64 @@
+package common
+
+import (
+ "bytes"
+ "encoding/base64"
+ "errors"
+ "fmt"
+ "github.com/chai2010/webp"
+ "image"
+ "io"
+ "net/http"
+ "strings"
+)
+
+func DecodeBase64ImageData(base64String string) (image.Config, error) {
+ // 去除base64数据的URL前缀(如果有)
+ if idx := strings.Index(base64String, ","); idx != -1 {
+ base64String = base64String[idx+1:]
+ }
+
+ // 将base64字符串解码为字节切片
+ decodedData, err := base64.StdEncoding.DecodeString(base64String)
+ if err != nil {
+ fmt.Println("Error: Failed to decode base64 string")
+ return image.Config{}, err
+ }
+
+ // 创建一个bytes.Buffer用于存储解码后的数据
+ reader := bytes.NewReader(decodedData)
+ config, err := getImageConfig(reader)
+ return config, err
+}
+
+func DecodeUrlImageData(imageUrl string) (image.Config, error) {
+ response, err := http.Get(imageUrl)
+ if err != nil {
+ SysLog(fmt.Sprintf("fail to get image from url: %s", err.Error()))
+ return image.Config{}, err
+ }
+
+ // 限制读取的字节数,防止下载整个图片
+ limitReader := io.LimitReader(response.Body, 8192)
+ config, err := getImageConfig(limitReader)
+ response.Body.Close()
+ return config, err
+}
+
+func getImageConfig(reader io.Reader) (image.Config, error) {
+ // 读取图片的头部信息来获取图片尺寸
+ config, _, err := image.DecodeConfig(reader)
+ if err != nil {
+ err = errors.New(fmt.Sprintf("fail to decode image config(gif, jpg, png): %s", err.Error()))
+ SysLog(err.Error())
+ config, err = webp.DecodeConfig(reader)
+ if err != nil {
+ err = errors.New(fmt.Sprintf("fail to decode image config(webp): %s", err.Error()))
+ SysLog(err.Error())
+ }
+ }
+ if err != nil {
+ return image.Config{}, err
+ }
+ return config, nil
+}
diff --git a/common/pprof.go b/common/pprof.go
new file mode 100644
index 0000000..4bec30f
--- /dev/null
+++ b/common/pprof.go
@@ -0,0 +1,44 @@
+package common
+
+import (
+ "fmt"
+ "github.com/shirou/gopsutil/cpu"
+ "os"
+ "runtime/pprof"
+ "time"
+)
+
+// Monitor 定时监控cpu使用率,超过阈值输出pprof文件
+func Monitor() {
+ for {
+ percent, err := cpu.Percent(time.Second, false)
+ if err != nil {
+ panic(err)
+ }
+ if percent[0] > 80 {
+ fmt.Println("cpu usage too high")
+ // write pprof file
+ if _, err := os.Stat("./pprof"); os.IsNotExist(err) {
+ err := os.Mkdir("./pprof", os.ModePerm)
+ if err != nil {
+ SysLog("创建pprof文件夹失败 " + err.Error())
+ continue
+ }
+ }
+ f, err := os.Create("./pprof/" + fmt.Sprintf("cpu-%s.pprof", time.Now().Format("20060102150405")))
+ if err != nil {
+ SysLog("创建pprof文件失败 " + err.Error())
+ continue
+ }
+ err = pprof.StartCPUProfile(f)
+ if err != nil {
+ SysLog("启动pprof失败 " + err.Error())
+ continue
+ }
+ time.Sleep(10 * time.Second) // profile for 30 seconds
+ pprof.StopCPUProfile()
+ f.Close()
+ }
+ time.Sleep(30 * time.Second)
+ }
+}
diff --git a/controller/midjourney.go b/controller/midjourney.go
index d9f4f89..a5ef8e4 100644
--- a/controller/midjourney.go
+++ b/controller/midjourney.go
@@ -19,12 +19,12 @@ import (
func UpdateMidjourneyTask() {
//revocer
imageModel := "midjourney"
+ defer func() {
+ if err := recover(); err != nil {
+ log.Printf("UpdateMidjourneyTask panic: %v", err)
+ }
+ }()
for {
- defer func() {
- if err := recover(); err != nil {
- log.Printf("UpdateMidjourneyTask panic: %v", err)
- }
- }()
time.Sleep(time.Duration(15) * time.Second)
tasks := model.GetAllUnFinishTasks()
if len(tasks) != 0 {
@@ -55,7 +55,6 @@ func UpdateMidjourneyTask() {
// 设置超时时间
timeout := time.Second * 5
ctx, cancel := context.WithTimeout(context.Background(), timeout)
- defer cancel()
// 使用带有超时的 context 创建新的请求
req = req.WithContext(ctx)
@@ -68,8 +67,8 @@ func UpdateMidjourneyTask() {
log.Printf("UpdateMidjourneyTask error: %v", err)
continue
}
- defer resp.Body.Close()
responseBody, err := io.ReadAll(resp.Body)
+ resp.Body.Close()
log.Printf("responseBody: %s", string(responseBody))
var responseItem Midjourney
// err = json.NewDecoder(resp.Body).Decode(&responseItem)
@@ -83,12 +82,12 @@ func UpdateMidjourneyTask() {
if err1 == nil && err2 == nil {
jsonData, err3 := json.Marshal(responseWithoutStatus)
if err3 != nil {
- log.Fatalf("UpdateMidjourneyTask error1: %v", err3)
+ log.Printf("UpdateMidjourneyTask error1: %v", err3)
continue
}
err4 := json.Unmarshal(jsonData, &responseStatus)
if err4 != nil {
- log.Fatalf("UpdateMidjourneyTask error2: %v", err4)
+ log.Printf("UpdateMidjourneyTask error2: %v", err4)
continue
}
responseItem.Status = strconv.Itoa(responseStatus.Status)
@@ -138,6 +137,7 @@ func UpdateMidjourneyTask() {
log.Printf("UpdateMidjourneyTask error5: %v", err)
}
log.Printf("UpdateMidjourneyTask success: %v", task)
+ cancel()
}
}
}
diff --git a/controller/relay-image.go b/controller/relay-image.go
index 8c16ec1..a1abd03 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -147,7 +147,7 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
var textResponse ImageResponse
defer func(ctx context.Context) {
if consumeQuota {
- err := model.PostConsumeTokenQuota(tokenId, userId, quota, 0, true)
+ err := model.PostConsumeTokenQuota(tokenId, userQuota, quota, 0, true)
if err != nil {
common.SysError("error consuming token remain quota: " + err.Error())
}
diff --git a/controller/relay-utils.go b/controller/relay-utils.go
index 6b1322b..ce9a3bf 100644
--- a/controller/relay-utils.go
+++ b/controller/relay-utils.go
@@ -4,7 +4,6 @@ import (
"encoding/json"
"errors"
"fmt"
- "github.com/chai2010/webp"
"github.com/gin-gonic/gin"
"github.com/pkoukk/tiktoken-go"
"image"
@@ -75,29 +74,21 @@ func getImageToken(imageUrl MessageImageUrl) (int, error) {
if imageUrl.Detail == "low" {
return 85, nil
}
-
- response, err := http.Get(imageUrl.Url)
+ var config image.Config
+ var err error
+ if strings.HasPrefix(imageUrl.Url, "http") {
+ common.SysLog(fmt.Sprintf("downloading image: %s", imageUrl.Url))
+ config, err = common.DecodeUrlImageData(imageUrl.Url)
+ } else {
+ common.SysLog(fmt.Sprintf("decoding image"))
+ config, err = common.DecodeBase64ImageData(imageUrl.Url)
+ }
if err != nil {
- fmt.Println("Error: Failed to get the URL")
return 0, err
}
- // 限制读取的字节数,防止下载整个图片
- limitReader := io.LimitReader(response.Body, 8192)
-
- response.Body.Close()
-
- // 读取图片的头部信息来获取图片尺寸
- config, _, err := image.DecodeConfig(limitReader)
- if err != nil {
- common.SysLog(fmt.Sprintf("fail to decode image config(gif, jpg, png): %s", err.Error()))
- config, err = webp.DecodeConfig(limitReader)
- if err != nil {
- common.SysLog(fmt.Sprintf("fail to decode image config(webp): %s", err.Error()))
- }
- }
if config.Width == 0 || config.Height == 0 {
- return 0, errors.New(fmt.Sprintf("fail to decode image config: %s", err.Error()))
+ return 0, errors.New(fmt.Sprintf("fail to decode image config: %s", imageUrl.Url))
}
if config.Width < 512 && config.Height < 512 {
if imageUrl.Detail == "auto" || imageUrl.Detail == "" {
diff --git a/controller/user.go b/controller/user.go
index 8fd10b8..f04f9ca 100644
--- a/controller/user.go
+++ b/controller/user.go
@@ -79,6 +79,7 @@ func setupLogin(user *model.User, c *gin.Context) {
DisplayName: user.DisplayName,
Role: user.Role,
Status: user.Status,
+ Group: user.Group,
}
c.JSON(http.StatusOK, gin.H{
"message": "",
@@ -284,6 +285,42 @@ func GenerateAccessToken(c *gin.Context) {
return
}
+type TransferAffQuotaRequest struct {
+ Quota int `json:"quota" binding:"required"`
+}
+
+func TransferAffQuota(c *gin.Context) {
+ id := c.GetInt("id")
+ user, err := model.GetUserById(id, true)
+ if err != nil {
+ c.JSON(http.StatusOK, gin.H{
+ "success": false,
+ "message": err.Error(),
+ })
+ return
+ }
+ tran := TransferAffQuotaRequest{}
+ if err := c.ShouldBindJSON(&tran); err != nil {
+ c.JSON(http.StatusOK, gin.H{
+ "success": false,
+ "message": err.Error(),
+ })
+ return
+ }
+ err = user.TransferAffQuotaToQuota(tran.Quota)
+ if err != nil {
+ c.JSON(http.StatusOK, gin.H{
+ "success": false,
+ "message": "划转失败 " + err.Error(),
+ })
+ return
+ }
+ c.JSON(http.StatusOK, gin.H{
+ "success": true,
+ "message": "划转成功",
+ })
+}
+
func GetAffCode(c *gin.Context) {
id := c.GetInt("id")
user, err := model.GetUserById(id, true)
@@ -330,6 +367,28 @@ func GetSelf(c *gin.Context) {
return
}
+func GetUserModels(c *gin.Context) {
+ id, err := strconv.Atoi(c.Param("id"))
+ if err != nil {
+ id = c.GetInt("id")
+ }
+ user, err := model.GetUserById(id, true)
+ if err != nil {
+ c.JSON(http.StatusOK, gin.H{
+ "success": false,
+ "message": err.Error(),
+ })
+ return
+ }
+ models := model.GetGroupModels(user.Group)
+ c.JSON(http.StatusOK, gin.H{
+ "success": true,
+ "message": "",
+ "data": models,
+ })
+ return
+}
+
func UpdateUser(c *gin.Context) {
var updatedUser model.User
err := json.NewDecoder(c.Request.Body).Decode(&updatedUser)
diff --git a/go.mod b/go.mod
index 3a75341..d7ac8c1 100644
--- a/go.mod
+++ b/go.mod
@@ -17,6 +17,7 @@ require (
github.com/gorilla/websocket v1.5.0
github.com/pkoukk/tiktoken-go v0.1.1
github.com/samber/lo v1.38.1
+ github.com/shirou/gopsutil v3.21.11+incompatible
github.com/star-horizon/go-epay v0.0.0-20230204124159-fa2e2293fdc2
golang.org/x/crypto v0.14.0
gorm.io/driver/mysql v1.4.3
@@ -33,6 +34,7 @@ require (
github.com/dlclark/regexp2 v1.8.1 // indirect
github.com/gabriel-vasile/mimetype v1.4.2 // indirect
github.com/gin-contrib/sse v0.1.0 // indirect
+ github.com/go-ole/go-ole v1.2.6 // indirect
github.com/go-playground/locales v0.14.1 // indirect
github.com/go-playground/universal-translator v0.18.1 // indirect
github.com/go-sql-driver/mysql v1.6.0 // indirect
@@ -54,8 +56,11 @@ require (
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/pelletier/go-toml/v2 v2.0.8 // indirect
+ github.com/tklauser/go-sysconf v0.3.12 // indirect
+ github.com/tklauser/numcpus v0.6.1 // indirect
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
github.com/ugorji/go/codec v1.2.11 // indirect
+ github.com/yusufpapurcu/wmi v1.2.3 // indirect
golang.org/x/arch v0.3.0 // indirect
golang.org/x/exp v0.0.0-20220303212507-bbda1eaf7a17 // indirect
golang.org/x/net v0.17.0 // indirect
diff --git a/go.sum b/go.sum
index 6e7f963..7e4cfa7 100644
--- a/go.sum
+++ b/go.sum
@@ -33,6 +33,8 @@ github.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwv
github.com/gin-gonic/gin v1.8.1/go.mod h1:ji8BvRH1azfM+SYow9zQ6SZMvR8qOMZHmsCuWR9tTTk=
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
+github.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=
+github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
@@ -131,6 +133,8 @@ github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUA
github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
github.com/samber/lo v1.38.1 h1:j2XEAqXKb09Am4ebOg31SpvzUTTs6EN3VfgeLUhPdXM=
github.com/samber/lo v1.38.1/go.mod h1:+m/ZKRl6ClXCE2Lgf3MsQlWfh4bn1bz6CXEOxnEXnEA=
+github.com/shirou/gopsutil v3.21.11+incompatible h1:+1+c1VGhc88SSonWP6foOcLhvnKlUeu/erjjvaPEYiI=
+github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
github.com/star-horizon/go-epay v0.0.0-20230204124159-fa2e2293fdc2 h1:avbt5a8F/zbYwFzTugrqWOBJe/K1cJj6+xpr+x1oVAI=
github.com/star-horizon/go-epay v0.0.0-20230204124159-fa2e2293fdc2/go.mod h1:SiffGCWGGMVwujne2dUQbJ5zUVD1V1Yj0hDuTfqFNEo=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
@@ -146,6 +150,10 @@ github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/tklauser/go-sysconf v0.3.12 h1:0QaGUFOdQaIVdPgfITYzaTegZvdCjmYO52cSFAEVmqU=
+github.com/tklauser/go-sysconf v0.3.12/go.mod h1:Ho14jnntGE1fpdOqQEEaiKRpvIavV0hSfmBq8nJbHYI=
+github.com/tklauser/numcpus v0.6.1 h1:ng9scYS7az0Bk4OZLvrNXNSAO2Pxr1XXRAPyjhIx+Fk=
+github.com/tklauser/numcpus v0.6.1/go.mod h1:1XfjsgE2zo8GVw7POkMbHENHzVg3GzmoZ9fESEdAacY=
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
@@ -154,6 +162,8 @@ github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLY
github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY=
github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+github.com/yusufpapurcu/wmi v1.2.3 h1:E1ctvB7uKFMOJw3fdOW32DwGE9I7t++CRUEMKvFoFiw=
+github.com/yusufpapurcu/wmi v1.2.3/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
@@ -165,6 +175,7 @@ golang.org/x/exp v0.0.0-20220303212507-bbda1eaf7a17/go.mod h1:lgLbSvA5ygNOMpwM/9
golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
+golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
@@ -172,6 +183,8 @@ golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBc
golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.13.0 h1:Af8nKPmuFypiUBjVoU9V20FiaFXOcuZI21p0ycVYYGE=
golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
diff --git a/main.go b/main.go
index de0b2a0..d4a0f4b 100644
--- a/main.go
+++ b/main.go
@@ -83,6 +83,12 @@ func main() {
common.SysLog("batch update enabled with interval " + strconv.Itoa(common.BatchUpdateInterval) + "s")
model.InitBatchUpdater()
}
+
+ if os.Getenv("ENABLE_PPROF") == "true" {
+ go common.Monitor()
+ common.SysLog("pprof enabled")
+ }
+
controller.InitTokenEncoders()
// Initialize HTTP server
diff --git a/model/ability.go b/model/ability.go
index 3da83be..2e8766a 100644
--- a/model/ability.go
+++ b/model/ability.go
@@ -13,6 +13,17 @@ type Ability struct {
Priority *int64 `json:"priority" gorm:"bigint;default:0;index"`
}
+func GetGroupModels(group string) []string {
+ var abilities []Ability
+ //去重
+ DB.Where("`group` = ?", group).Distinct("model").Find(&abilities)
+ models := make([]string, 0, len(abilities))
+ for _, ability := range abilities {
+ models = append(models, ability.Model)
+ }
+ return models
+}
+
func GetRandomSatisfiedChannel(group string, model string) (*Channel, error) {
ability := Ability{}
groupCol := "`group`"
diff --git a/model/token.go b/model/token.go
index ee6b4a9..66b8369 100644
--- a/model/token.go
+++ b/model/token.go
@@ -220,28 +220,30 @@ func PostConsumeTokenQuota(tokenId int, userQuota int, quota int, preConsumedQuo
}
if sendEmail {
- quotaTooLow := userQuota >= common.QuotaRemindThreshold && userQuota-(quota+preConsumedQuota) < common.QuotaRemindThreshold
- noMoreQuota := userQuota-(quota+preConsumedQuota) <= 0
- if quotaTooLow || noMoreQuota {
- go func() {
- email, err := GetUserEmail(token.UserId)
- if err != nil {
- common.SysError("failed to fetch user email: " + err.Error())
- }
- prompt := "您的额度即将用尽"
- if noMoreQuota {
- prompt = "您的额度已用尽"
- }
- if email != "" {
- topUpLink := fmt.Sprintf("%s/topup", common.ServerAddress)
- err = common.SendEmail(prompt, email,
- fmt.Sprintf("%s,当前剩余额度为 %d,为了不影响您的使用,请及时充值。
充值链接:%s", prompt, userQuota, topUpLink, topUpLink))
+ if (quota + preConsumedQuota) != 0 {
+ quotaTooLow := userQuota >= common.QuotaRemindThreshold && userQuota-(quota+preConsumedQuota) < common.QuotaRemindThreshold
+ noMoreQuota := userQuota-(quota+preConsumedQuota) <= 0
+ if quotaTooLow || noMoreQuota {
+ go func() {
+ email, err := GetUserEmail(token.UserId)
if err != nil {
- common.SysError("failed to send email" + err.Error())
+ common.SysError("failed to fetch user email: " + err.Error())
}
- common.SysLog("user quota is low, consumed quota: " + strconv.Itoa(quota) + ", user quota: " + strconv.Itoa(userQuota))
- }
- }()
+ prompt := "您的额度即将用尽"
+ if noMoreQuota {
+ prompt = "您的额度已用尽"
+ }
+ if email != "" {
+ topUpLink := fmt.Sprintf("%s/topup", common.ServerAddress)
+ err = common.SendEmail(prompt, email,
+ fmt.Sprintf("%s,当前剩余额度为 %d,为了不影响您的使用,请及时充值。
充值链接:%s", prompt, userQuota, topUpLink, topUpLink))
+ if err != nil {
+ common.SysError("failed to send email" + err.Error())
+ }
+ common.SysLog("user quota is low, consumed quota: " + strconv.Itoa(quota) + ", user quota: " + strconv.Itoa(userQuota))
+ }
+ }()
+ }
}
}
diff --git a/model/user.go b/model/user.go
index 31eae3b..422d268 100644
--- a/model/user.go
+++ b/model/user.go
@@ -27,6 +27,9 @@ type User struct {
RequestCount int `json:"request_count" gorm:"type:int;default:0;"` // request number
Group string `json:"group" gorm:"type:varchar(32);default:'default'"`
AffCode string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"`
+ AffCount int `json:"aff_count" gorm:"type:int;default:0;column:aff_count"`
+ AffQuota int `json:"aff_quota" gorm:"type:int;default:0;column:aff_quota"` // 邀请剩余额度
+ AffHistoryQuota int `json:"aff_history_quota" gorm:"type:int;default:0;column:aff_history"` // 邀请历史额度
InviterId int `json:"inviter_id" gorm:"type:int;column:inviter_id;index"`
}
@@ -77,6 +80,54 @@ func DeleteUserById(id int) (err error) {
return user.Delete()
}
+func inviteUser(inviterId int) (err error) {
+ user, err := GetUserById(inviterId, true)
+ if err != nil {
+ return err
+ }
+ user.AffCount++
+ user.AffQuota += common.QuotaForInviter
+ user.AffHistoryQuota += common.QuotaForInviter
+ return DB.Save(user).Error
+}
+
+func (user *User) TransferAffQuotaToQuota(quota int) error {
+ // 检查quota是否小于最小额度
+ if float64(quota) < common.QuotaPerUnit {
+ return fmt.Errorf("转移额度最小为%s!", common.LogQuota(int(common.QuotaPerUnit)))
+ }
+
+ // 开始数据库事务
+ tx := DB.Begin()
+ if tx.Error != nil {
+ return tx.Error
+ }
+ defer tx.Rollback() // 确保在函数退出时事务能回滚
+
+ // 加锁查询用户以确保数据一致性
+ err := tx.Set("gorm:query_option", "FOR UPDATE").First(&user, user.Id).Error
+ if err != nil {
+ return err
+ }
+
+ // 再次检查用户的AffQuota是否足够
+ if user.AffQuota < quota {
+ return errors.New("邀请额度不足!")
+ }
+
+ // 更新用户额度
+ user.AffQuota -= quota
+ user.Quota += quota
+
+ // 保存用户状态
+ if err := tx.Save(user).Error; err != nil {
+ return err
+ }
+
+ // 提交事务
+ return tx.Commit().Error
+}
+
func (user *User) Insert(inviterId int) error {
var err error
if user.Password != "" {
@@ -101,8 +152,9 @@ func (user *User) Insert(inviterId int) error {
RecordLog(user.Id, LogTypeSystem, fmt.Sprintf("使用邀请码赠送 %s", common.LogQuota(common.QuotaForInvitee)))
}
if common.QuotaForInviter > 0 {
- _ = IncreaseUserQuota(inviterId, common.QuotaForInviter)
+ //_ = IncreaseUserQuota(inviterId, common.QuotaForInviter)
RecordLog(inviterId, LogTypeSystem, fmt.Sprintf("邀请用户赠送 %s", common.LogQuota(common.QuotaForInviter)))
+ _ = inviteUser(inviterId)
}
}
return nil
diff --git a/router/api-router.go b/router/api-router.go
index 17758b0..038b750 100644
--- a/router/api-router.go
+++ b/router/api-router.go
@@ -39,6 +39,7 @@ func SetApiRouter(router *gin.Engine) {
selfRoute.Use(middleware.UserAuth())
{
selfRoute.GET("/self", controller.GetSelf)
+ selfRoute.GET("/models", controller.GetUserModels)
selfRoute.PUT("/self", controller.UpdateSelf)
selfRoute.DELETE("/self", controller.DeleteSelf)
selfRoute.GET("/token", controller.GenerateAccessToken)
@@ -46,6 +47,7 @@ func SetApiRouter(router *gin.Engine) {
selfRoute.POST("/topup", controller.TopUp)
selfRoute.POST("/pay", controller.RequestEpay)
selfRoute.POST("/amount", controller.RequestAmount)
+ selfRoute.POST("/aff_transfer", controller.TransferAffQuota)
}
adminRoute := userRoute.Group("/")
diff --git a/web/src/components/LogsTable.js b/web/src/components/LogsTable.js
index e2aa8aa..8556493 100644
--- a/web/src/components/LogsTable.js
+++ b/web/src/components/LogsTable.js
@@ -106,7 +106,7 @@ const LogsTable = () => {
return (
record.type === 0 || record.type === 2 ?
+ 微信扫码关注公众号,输入「验证码」获取验证码(三分钟内有效) +
+- 微信扫码关注公众号,输入「验证码」获取验证码(三分钟内有效) -
-