epuiment

services/equipment/internal/logic/editLogic.go

@@ -0,0 +1,81 @@
+package logic
+
+import (
+	"context"
+	"lone-services/pkg/modelbase"
+	"lone-services/pkg/utils"
+	"lone-services/services/equipment/internal/dao"
+	"lone-services/services/equipment/internal/model"
+	"lone-services/services/task/validator"
+	"strconv"
+
+	"lone-services/rpc/equipment/pb"
+	"lone-services/services/equipment/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type EditLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+	BaseLogic
+}
+
+func NewEditLogic(ctx context.Context, svcCtx *svc.ServiceContext) *EditLogic {
+	return &EditLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *EditLogic) Edit(in *equipment.EditRequest) (*equipment.Response, error) {
+	var v validator.TaskEditValidator
+	if fail := l.checkParams(in, &v); fail != nil {
+		return fail, nil
+	}
+	adminInfo := utils.GetUserFromCtx(l.ctx)
+	if adminInfo.ID < utils.NumberOne {
+		return l.fail(utils.ErrorNoLoginInfo)
+	}
+
+	var info dao.Version
+	w := modelbase.Params{Eq: map[string]string{"id": strconv.Itoa(int(in.Id))}}
+	err := model.VersionModel{}.Init().GetOne(w, &info)
+	if err != nil {
+		return l.fail(utils.ErrorNotFund)
+	}
+	if info.Id < utils.NumberOne {
+		return l.fail(utils.ErrorNotFund)
+	}
+	action := utils.ActionAdd{
+		OldContent: info,
+		Type:       utils.LogActionTypeEdit,
+		ModuleName: utils.LogActionModuleEquipment,
+	}
+
+	info.AdminId = adminInfo.ID
+	info.AdminName = adminInfo.Name
+	info.HardwareVersion = in.HardwareVersion
+	info.Version = in.Version
+	info.FirmwareSrc = in.FirmwareSrc
+	info.Describe = in.Describe
+	info.SoftSrc = in.SoftSrc
+	info.AddSrc = in.AddSrc
+	info.SoftConfig = in.SoftConfig
+	info.Type = uint8(in.Type)
+
+	row, err := model.VersionModel{}.Init().Edit(w, &info)
+	if err != nil {
+		l.Logger.Error(err)
+		return l.fail(utils.Fail)
+	}
+
+	if row > utils.NumberZero {
+		action.NewContent = info
+		utils.SetActionLog(adminInfo, action)
+	}
+
+	return l.ok(info.Id)
+}