fix: 复合主键表 shop_players 的 update 改用原生 SQL

ent 的 Update().Where() 对复合主键表会报 "sql/sqlgraph: invalid composite id for update table",这是 ent 框架已知的限制(field.ID 复合主键仅对 edge schema 完整支持)。同表的 Create/Delete/Query 操作不受影响,仅 Update 需要绕过。在 ServiceContext 中暴露底层 *sql.DB 以供 ExecContext 执行原生 SQL。
This commit is contained in:
zetaloop
2026-04-23 16:59:38 +08:00
parent b10d97117e
commit fbfdedb3ed
2 changed files with 15 additions and 12 deletions
@@ -3,8 +3,6 @@ package logic
import ( import (
"context" "context"
"errors" "errors"
"juwan-backend/app/shop/rpc/internal/models/predicate"
"juwan-backend/app/shop/rpc/internal/models/shopplayers"
"time" "time"
"juwan-backend/app/shop/rpc/internal/svc" "juwan-backend/app/shop/rpc/internal/svc"
@@ -32,25 +30,28 @@ func (l *UpdateShopPlayersLogic) UpdateShopPlayers(in *pb.UpdateShopPlayersReq)
return nil, errors.New("invalid shop_id or player_id") return nil, errors.New("invalid shop_id or player_id")
} }
preds := []predicate.ShopPlayers{ // ent 的 Update().Where() 不支持复合主键表(shop_players 的主键是 (shop_id, player_id)),
shopplayers.ShopIDEQ(in.ShopId), // 会报 "sql/sqlgraph: invalid composite id for update table"。
shopplayers.PlayerIDEQ(in.PlayerId), // Create/Delete/Query 不受影响,仅 Update 需要走原生 SQL。
} query := `UPDATE shop_players SET is_primary = $1, left_at = $2 WHERE shop_id = $3 AND player_id = $4`
updater := l.svcCtx.ShopModelRW.ShopPlayers.Update().
Where(shopplayers.And(preds...)).
SetIsPrimary(in.IsPrimary)
var leftAt *time.Time
if in.LeftAt > 0 { if in.LeftAt > 0 {
t := time.Unix(in.LeftAt, 0) t := time.Unix(in.LeftAt, 0)
updater = updater.SetLeftAt(t) leftAt = &t
} }
affected, err := updater.Save(l.ctx) result, err := l.svcCtx.DBRW.ExecContext(l.ctx, query, in.IsPrimary, leftAt, in.ShopId, in.PlayerId)
if err != nil { if err != nil {
logx.Errorf("update shop players failed, %s", err.Error()) logx.Errorf("update shop players failed, %s", err.Error())
return nil, errors.New("update shop players failed") return nil, errors.New("update shop players failed")
} }
affected, err := result.RowsAffected()
if err != nil {
logx.Errorf("get rows affected failed, %s", err.Error())
return nil, errors.New("update shop players failed")
}
if affected == 0 { if affected == 0 {
return nil, errors.New("shop player not found") return nil, errors.New("shop player not found")
} }
@@ -27,6 +27,7 @@ type ServiceContext struct {
UsersRpc usercenter.Usercenter UsersRpc usercenter.Usercenter
ShopModelRW *models.Client ShopModelRW *models.Client
ShopModelRO *models.Client ShopModelRO *models.Client
DBRW *stdsql.DB
} }
func NewServiceContext(c config.Config) *ServiceContext { func NewServiceContext(c config.Config) *ServiceContext {
@@ -68,5 +69,6 @@ func NewServiceContext(c config.Config) *ServiceContext {
UsersRpc: usercenter.NewUsercenter(zrpc.MustNewClient(c.UsersRpcConf)), UsersRpc: usercenter.NewUsercenter(zrpc.MustNewClient(c.UsersRpcConf)),
ShopModelRO: models.NewClient(roModelOpts...), ShopModelRO: models.NewClient(roModelOpts...),
ShopModelRW: models.NewClient(rwModelOpts...), ShopModelRW: models.NewClient(rwModelOpts...),
DBRW: rawRW,
} }
} }