create_agent_logic.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. package agent
  2. import (
  3. "context"
  4. "errors"
  5. "wechat-api/ent/agent"
  6. "wechat-api/internal/svc"
  7. "wechat-api/internal/types"
  8. "wechat-api/internal/utils/dberrorhandler"
  9. "github.com/suyuan32/simple-admin-common/msg/errormsg"
  10. "github.com/zeromicro/go-zero/core/logx"
  11. )
  12. type CreateAgentLogic struct {
  13. ctx context.Context
  14. svcCtx *svc.ServiceContext
  15. logx.Logger
  16. }
  17. func NewCreateAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateAgentLogic {
  18. return &CreateAgentLogic{
  19. ctx: ctx,
  20. svcCtx: svcCtx,
  21. Logger: logx.WithContext(ctx),
  22. }
  23. }
  24. func (l *CreateAgentLogic) CreateAgent(req *types.AgentInfo) (*types.BaseMsgResp, error) {
  25. item, err := l.svcCtx.DB.Agent.Query().Where(agent.Name(*req.Name)).First(l.ctx)
  26. if err != nil {
  27. l.Error("DB查询失败")
  28. return nil, err
  29. }
  30. if item != nil && item.ID > 0 {
  31. l.Error("角色名字已经存在")
  32. return nil, errors.New("角色名字已经存在")
  33. }
  34. _, err = l.svcCtx.DB.Agent.Create().
  35. SetNotNilName(req.Name).
  36. SetNotNilRole(req.Role).
  37. //SetNotNilStatus(req.Status).
  38. SetNotNilBackground(req.Background).
  39. SetNotNilExamples(req.Examples).
  40. Save(l.ctx)
  41. if err != nil {
  42. return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
  43. }
  44. return &types.BaseMsgResp{Msg: errormsg.CreateSuccess}, nil
  45. }