feat: the power log page is ready

api/core/types/chat.go

@@ -56,6 +56,7 @@ type ChatSession struct {
 type ChatModel struct {
 	Id          uint     `json:"id"`
 	Platform    Platform `json:"platform"`
+	Name        string   `json:"name"`
 	Value       string   `json:"value"`
 	Power       int      `json:"power"`
 	MaxTokens   int      `json:"max_tokens"`  // 最大响应长度
@@ -86,6 +87,21 @@ const (
 	PowerReward   = PowerType(5) // 众筹
 )
 
+func (t PowerType) String() string {
+	switch t {
+	case PowerRecharge:
+		return "充值"
+	case PowerConsume:
+		return "消费"
+	case PowerRefund:
+		return "退款"
+	case PowerReward:
+		return "众筹"
+
+	}
+	return "其他"
+}
+
 type PowerMark int
 
 const (

api/handler/chat_role_handler.go

@@ -22,6 +22,7 @@ func NewChatRoleHandler(app *core.AppServer, db *gorm.DB) *ChatRoleHandler {
 
 // List 获取用户聊天应用列表
 func (h *ChatRoleHandler) List(c *gin.Context) {
+	all := h.GetBool(c, "all")
 	userId := h.GetLoginUserId(c)
 	var roles []model.ChatRole
 	res := h.DB.Where("enable", true).Order("sort_num ASC").Find(&roles)
@@ -31,7 +32,7 @@ func (h *ChatRoleHandler) List(c *gin.Context) {
 	}
 
 	// 获取所有角色
-	if userId == 0 {
+	if userId == 0 || all {
 		// 转成 vo
 		var roleVos = make([]vo.ChatRole, 0)
 		for _, r := range roles {

api/handler/chatimpl/chat_handler.go

@@ -105,6 +105,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 	session.ChatId = chatId
 	session.Model = types.ChatModel{
 		Id:          chatModel.Id,
+		Name:        chatModel.Name,
 		Value:       chatModel.Value,
 		Power:       chatModel.Power,
 		MaxTokens:   chatModel.MaxTokens,
@@ -535,7 +536,7 @@ func (h *ChatHandler) subUserPower(userVo vo.User, session *types.ChatSession, p
 			Mark:      types.PowerSub,
 			Balance:   userVo.Power - power,
 			Model:     session.Model.Value,
-			Remark:    fmt.Sprintf("提问长度：%d，回复长度：%d", promptTokens, replyTokens),
+			Remark:    fmt.Sprintf("模型名称：%s, 提问长度：%d，回复长度：%d", session.Model.Name, promptTokens, replyTokens),
 			CreatedAt: time.Now(),
 		})
 	}

api/handler/power_log_handler.go

@@ -0,0 +1,65 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/store/vo"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type PowerLogHandler struct {
+	BaseHandler
+}
+
+func NewPowerLogHandler(app *core.AppServer, db *gorm.DB) *PowerLogHandler {
+	return &PowerLogHandler{BaseHandler: BaseHandler{App: app, DB: db}}
+}
+
+func (h *PowerLogHandler) List(c *gin.Context) {
+	var data struct {
+		Model    string   `json:"model"`
+		Date     []string `json:"date"`
+		Page     int      `json:"page"`
+		PageSize int      `json:"page_size"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	session := h.DB.Session(&gorm.Session{})
+	if data.Model != "" {
+		session = session.Where("model", data.Model)
+	}
+	if len(data.Date) == 2 {
+		start := data.Date[0] + " 00:00:00"
+		end := data.Date[1] + " 00:00:00"
+		session = session.Where("created_at >= ? AND created_at <= ?", start, end)
+	}
+
+	var total int64
+	session.Model(&model.PowerLog{}).Count(&total)
+	var items []model.PowerLog
+	var list = make([]vo.PowerLog, 0)
+	offset := (data.Page - 1) * data.PageSize
+	res := session.Debug().Order("id DESC").Offset(offset).Limit(data.PageSize).Find(&items)
+	if res.Error == nil {
+		for _, item := range items {
+			var log vo.PowerLog
+			err := utils.CopyObject(item, &log)
+			if err != nil {
+				continue
+			}
+			log.Id = item.Id
+			log.CreatedAt = item.CreatedAt.Unix()
+			log.TypeStr = item.Type.String()
+			list = append(list, log)
+		}
+	}
+	resp.SUCCESS(c, vo.NewPage(total, data.Page, data.PageSize, list))
+}

api/handler/user_handler.go

@@ -108,6 +108,20 @@ func (h *UserHandler) Register(c *gin.Context) {
 		h.DB.Model(&model.InviteCode{}).Where("code = ?", data.InviteCode).UpdateColumn("reg_num", gorm.Expr("reg_num + ?", 1))
 		if h.App.SysConfig.InvitePower > 0 {
 			h.DB.Model(&model.User{}).Where("id = ?", inviteCode.UserId).UpdateColumn("power", gorm.Expr("power + ?", h.App.SysConfig.InvitePower))
+			// 记录邀请算力充值日志
+			var inviter model.User
+			h.DB.Where("id", inviteCode.UserId).First(&inviter)
+			h.DB.Create(&model.PowerLog{
+				UserId:    inviter.Id,
+				Username:  inviter.Username,
+				Type:      types.PowerInvite,
+				Amount:    h.App.SysConfig.InvitePower,
+				Balance:   inviter.Power,
+				Mark:      types.PowerAdd,
+				Model:     "",
+				Remark:    fmt.Sprintf("邀请用户注册奖励，金额：%d，邀请码：%s，新用户：%s", h.App.SysConfig.InvitePower, inviteCode.Code, user.Username),
+				CreatedAt: time.Now(),
+			})
 		}
 
 		// 添加邀请记录

api/main.go

@@ -126,6 +126,7 @@ func main() {
 		fx.Provide(handler.NewOrderHandler),
 		fx.Provide(handler.NewProductHandler),
 		fx.Provide(handler.NewConfigHandler),
+		fx.Provide(handler.NewPowerLogHandler),
 
 		fx.Provide(admin.NewConfigHandler),
 		fx.Provide(admin.NewAdminHandler),
@@ -401,8 +402,9 @@ func main() {
 			group.GET("remove", h.RemoveChat)
 			group.GET("message/remove", h.RemoveMessage)
 		}),
-		fx.Provide(handler.NewTestHandler),
-		fx.Invoke(func(s *core.AppServer, h *handler.TestHandler) {
+		fx.Invoke(func(s *core.AppServer, h *handler.PowerLogHandler) {
+			group := s.Engine.Group("/api/powerLog/")
+			group.POST("list", h.List)
 		}),
 		fx.Invoke(func(s *core.AppServer, db *gorm.DB) {
 			err := s.Run(db)

api/store/vo/power_log.go

@@ -6,9 +6,10 @@ type PowerLog struct {
 	Id        uint            `json:"id"`
 	UserId    uint            `json:"user_id"`
 	Username  string          `json:"username"`
-	Type      types.PowerType `json:"name"`
+	Type      types.PowerType `json:"type"`
+	TypeStr   string          `json:"type_str"`
 	Amount    int             `json:"amount"`
-	Mark      types.PowerMark `json:"fund_type"`
+	Mark      types.PowerMark `json:"mark"`
 	Balance   int             `json:"balance"`
 	Model     string          `json:"model"`
 	Remark    string          `json:"remark"`