123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- 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 CreateAgentLogic struct {
- ctx context.Context
- svcCtx *svc.ServiceContext
- logx.Logger
- }
- func NewCreateAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateAgentLogic {
- return &CreateAgentLogic{
- ctx: ctx,
- svcCtx: svcCtx,
- Logger: logx.WithContext(ctx),
- }
- }
- func (l *CreateAgentLogic) CreateAgent(req *types.AgentInfo) (*types.BaseMsgResp, error) {
- item, err := l.svcCtx.DB.Agent.Query().Where(agent.Name(*req.Name)).First(l.ctx)
- if err != nil {
- l.Error("DB查询失败")
- return nil, err
- }
- if item != nil && item.ID > 0 {
- l.Error("角色名字已经存在")
- return nil, errors.New("角色名字已经存在")
- }
- _, err = l.svcCtx.DB.Agent.Create().
- SetNotNilName(req.Name).
- SetNotNilRole(req.Role).
- //SetNotNilStatus(req.Status).
- SetNotNilBackground(req.Background).
- SetNotNilExamples(req.Examples).
- Save(l.ctx)
- if err != nil {
- return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
- }
- return &types.BaseMsgResp{Msg: errormsg.CreateSuccess}, nil
- }
|