fix:edit API

internal/logic/agent/get_agent_by_id_logic.go

@@ -2,6 +2,7 @@ package agent
 
 import (
 	"context"
+	"wechat-api/ent"
 	"wechat-api/ent/agent"
 
 	"wechat-api/internal/svc"
@@ -31,7 +32,7 @@ func NewGetAgentByIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetA
 func (l *GetAgentByIdLogic) GetAgentById(req *types.IDReq) (*types.AgentInfoResp, error) {
 	organizationId := l.ctx.Value("organizationId").(uint64)
 	data, err := l.svcCtx.DB.Agent.Query().Where(agent.ID(req.Id), agent.OrganizationID(organizationId)).First(l.ctx)
-	if err != nil {
+	if err != nil && !ent.IsNotFound(err) {
 		return nil, dberrorhandler.DefaultEntError(l.Logger, err, req)
 	}
 

internal/logic/agent/update_agent_logic.go

@@ -3,6 +3,7 @@ package agent
 import (
 	"context"
 	"errors"
+	"wechat-api/ent"
 	"wechat-api/ent/agent"
 
 	"wechat-api/internal/svc"
@@ -35,8 +36,8 @@ func (l *UpdateAgentLogic) UpdateAgent(req *types.AgentInfo) (*types.BaseMsgResp
 		return nil, errors.New("记录ID不存在")
 	}
 
-	item, err := l.svcCtx.DB.Agent.Query().Where(agent.ID(*req.Id), agent.OrganizationID(organizationId)).First(l.ctx)
-	if item.ID == 0 {
+	_, err := l.svcCtx.DB.Agent.Query().Where(agent.ID(*req.Id), agent.OrganizationID(organizationId)).First(l.ctx)
+	if err != nil && ent.IsNotFound(err) {
 		return nil, errors.New("记录不存在")
 	}