add: 打手初始化接口

This commit is contained in:
zetaloop
2026-04-04 03:35:36 +08:00
parent 85ce6a45c5
commit 32d58ef462
12 changed files with 224 additions and 21 deletions
@@ -33,18 +33,28 @@ func (l *AddPlayersLogic) AddPlayers(in *pb.AddPlayersReq) (*pb.AddPlayersResp,
logx.Errorf("addPlayerServices err:%v", err)
return nil, errors.New("create player service id failed")
}
_, err = l.svcCtx.PlayerModelRW.Players.Create().
creator := l.svcCtx.PlayerModelRW.Players.Create().
SetID(idResp.Id).
SetUserID(in.UserId).
SetStatus(in.Status).
SetRating(decimal.NewFromFloat(in.Rating)).
SetTotalOrders(int(in.TotalOrders)).
SetCompletedOrders(int(in.CompletedOrders)).
SetShopID(in.ShopId).
SetTags(in.Tags).
SetGender(in.Gender != 0).
SetGames(in.Games).
Save(l.ctx)
SetGender(in.Gender != 0)
if in.TotalOrders != 0 {
creator.SetTotalOrders(int(in.TotalOrders))
}
if in.CompletedOrders != 0 {
creator.SetCompletedOrders(int(in.CompletedOrders))
}
if in.ShopId != 0 {
creator.SetShopID(in.ShopId)
}
if len(in.GetTags()) > 0 {
creator.SetTags(in.GetTags())
}
if len(in.GetGames()) > 0 {
creator.SetGames(in.GetGames())
}
_, err = creator.Save(l.ctx)
if err != nil {
logx.Errorf("addPlayerServices err:%v", err)
return nil, errors.New("add player service failed")
@@ -3,12 +3,15 @@ package logic
import (
"context"
"errors"
"juwan-backend/app/player/rpc/internal/models"
"juwan-backend/app/player/rpc/internal/models/players"
"juwan-backend/app/player/rpc/internal/svc"
"juwan-backend/app/player/rpc/pb"
"github.com/zeromicro/go-zero/core/logx"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
type GetPlayerByUserIdLogic struct {
@@ -32,6 +35,9 @@ func (l *GetPlayerByUserIdLogic) GetPlayerByUserId(in *pb.SearchPlayersReq) (*pb
player, err := l.svcCtx.PlayerModelRO.Players.Query().Where(players.UserIDEQ(in.GetUserId())).First(l.ctx)
if err != nil {
if models.IsNotFound(err) {
return nil, status.Error(codes.NotFound, "player not found")
}
logx.WithContext(l.ctx).Errorf("GetPlayerByUserIdLogic err: %v", err)
return nil, errors.New("get player by user id failed")
}
@@ -40,6 +46,7 @@ func (l *GetPlayerByUserIdLogic) GetPlayerByUserId(in *pb.SearchPlayersReq) (*pb
Id: player.ID,
UserId: player.UserID,
Status: player.Status,
Gender: player.Gender,
Rating: player.Rating.InexactFloat64(),
TotalOrders: int64(player.TotalOrders),
CompletedOrders: int64(player.CompletedOrders),
+5 -7
View File
@@ -3,9 +3,9 @@
package models
import (
"encoding/json"
"fmt"
"juwan-backend/app/player/rpc/internal/models/players"
"juwan-backend/pkg/types"
"strings"
"time"
@@ -51,7 +51,7 @@ func (*Players) scanValues(columns []string) ([]any, error) {
for i := range columns {
switch columns[i] {
case players.FieldTags:
values[i] = new([]byte)
values[i] = new(types.TextArray)
case players.FieldRating:
values[i] = new(decimal.Decimal)
case players.FieldGames:
@@ -129,12 +129,10 @@ func (_m *Players) assignValues(columns []string, values []any) error {
*_m.ShopID = value.Int64
}
case players.FieldTags:
if value, ok := values[i].(*[]byte); !ok {
if value, ok := values[i].(*types.TextArray); !ok {
return fmt.Errorf("unexpected type %T for field tags", values[i])
} else if value != nil && len(*value) > 0 {
if err := json.Unmarshal(*value, &_m.Tags); err != nil {
return fmt.Errorf("unmarshal field tags: %w", err)
}
} else if value != nil {
_m.Tags = append([]string{}, value.Elements...)
}
case players.FieldGames:
if value, ok := values[i].(*pq.Int64Array); !ok {
@@ -7,6 +7,7 @@ import (
"errors"
"fmt"
"juwan-backend/app/player/rpc/internal/models/players"
"juwan-backend/pkg/types"
"time"
"entgo.io/ent/dialect/sql/sqlgraph"
@@ -314,7 +315,8 @@ func (_c *PlayersCreate) createSpec() (*Players, *sqlgraph.CreateSpec) {
_node.ShopID = &value
}
if value, ok := _c.mutation.Tags(); ok {
_spec.SetField(players.FieldTags, field.TypeJSON, value)
tags := types.TextArray{Elements: append([]string{}, value...), Valid: true}
_spec.SetField(players.FieldTags, field.TypeOther, tags)
_node.Tags = value
}
if value, ok := _c.mutation.Games(); ok {
@@ -8,6 +8,7 @@ import (
"fmt"
"juwan-backend/app/player/rpc/internal/models/players"
"juwan-backend/app/player/rpc/internal/models/predicate"
"juwan-backend/pkg/types"
"time"
"entgo.io/ent/dialect/sql"
@@ -326,7 +327,8 @@ func (_u *PlayersUpdate) sqlSave(ctx context.Context) (_node int, err error) {
_spec.ClearField(players.FieldShopID, field.TypeInt64)
}
if value, ok := _u.mutation.Tags(); ok {
_spec.SetField(players.FieldTags, field.TypeJSON, value)
tags := types.TextArray{Elements: append([]string{}, value...), Valid: true}
_spec.SetField(players.FieldTags, field.TypeOther, tags)
}
if value, ok := _u.mutation.AppendedTags(); ok {
_spec.AddModifier(func(u *sql.UpdateBuilder) {
@@ -334,7 +336,7 @@ func (_u *PlayersUpdate) sqlSave(ctx context.Context) (_node int, err error) {
})
}
if _u.mutation.TagsCleared() {
_spec.ClearField(players.FieldTags, field.TypeJSON)
_spec.ClearField(players.FieldTags, field.TypeOther)
}
if value, ok := _u.mutation.Games(); ok {
_spec.SetField(players.FieldGames, field.TypeOther, value)
@@ -690,7 +692,8 @@ func (_u *PlayersUpdateOne) sqlSave(ctx context.Context) (_node *Players, err er
_spec.ClearField(players.FieldShopID, field.TypeInt64)
}
if value, ok := _u.mutation.Tags(); ok {
_spec.SetField(players.FieldTags, field.TypeJSON, value)
tags := types.TextArray{Elements: append([]string{}, value...), Valid: true}
_spec.SetField(players.FieldTags, field.TypeOther, tags)
}
if value, ok := _u.mutation.AppendedTags(); ok {
_spec.AddModifier(func(u *sql.UpdateBuilder) {
@@ -698,7 +701,7 @@ func (_u *PlayersUpdateOne) sqlSave(ctx context.Context) (_node *Players, err er
})
}
if _u.mutation.TagsCleared() {
_spec.ClearField(players.FieldTags, field.TypeJSON)
_spec.ClearField(players.FieldTags, field.TypeOther)
}
if value, ok := _u.mutation.Games(); ok {
_spec.SetField(players.FieldGames, field.TypeOther, value)