1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- package agent
- import (
- "context"
- "errors"
- "wechat-api/ent/agent"
- "wechat-api/internal/svc"
- "wechat-api/internal/types"
- "wechat-api/internal/utils/dberrorhandler"
- "github.com/suyuan32/simple-admin-common/msg/errormsg"
- "github.com/zeromicro/go-zero/core/logx"
- )
- type UpdateAgentLogic struct {
- ctx context.Context
- svcCtx *svc.ServiceContext
- logx.Logger
- }
- func NewUpdateAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateAgentLogic {
- return &UpdateAgentLogic{
- ctx: ctx,
- svcCtx: svcCtx,
- Logger: logx.WithContext(ctx),
- }
- }
- func (l *UpdateAgentLogic) UpdateAgent(req *types.AgentInfo) (*types.BaseMsgResp, error) {
- item, err := l.svcCtx.DB.Agent.Query().Where(agent.Name(*req.Name)).Where(agent.IDNEQ(*req.Id)).First(l.ctx)
- if err != nil {
- l.Error("DB查询失败")
- return nil, err
- }
- if item != nil && item.ID > 0 {
- return nil, errors.New("已有相同名字的角色")
- }
- err = l.svcCtx.DB.Agent.UpdateOneID(*req.Id).
- SetNotNilName(req.Name).
- SetNotNilRole(req.Role).
- SetNotNilStatus(req.Status).
- SetNotNilBackground(req.Background).
- SetNotNilExamples(req.Examples).
- Exec(l.ctx)
- if err != nil {
- return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
- }
- return &types.BaseMsgResp{Msg: errormsg.UpdateSuccess}, nil
- }
|