115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
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/equipment/validator"
|
|
"strconv"
|
|
|
|
"lone-services/rpc/equipment/pb"
|
|
"lone-services/services/equipment/internal/svc"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type ApiBindAddLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
BaseLogic
|
|
}
|
|
|
|
func NewApiBindAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApiBindAddLogic {
|
|
return &ApiBindAddLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *ApiBindAddLogic) ApiBindAdd(in *equipment.ApiBindAddRequest) (*equipment.Response, error) {
|
|
var v validator.BindValidator
|
|
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)
|
|
}
|
|
|
|
w := modelbase.Params{
|
|
Eq: map[string]string{
|
|
"host_code": v.HostCode,
|
|
"status": utils.StringStatusOk,
|
|
"type": strconv.Itoa(int(in.Type))},
|
|
}
|
|
var info dao.BindCreate
|
|
modelObj := model.BindModel{}.Init()
|
|
err := modelObj.GetOne(w, &info)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return l.fail(utils.Fail)
|
|
}
|
|
if info.Id > utils.NumberZero {
|
|
if info.UserId == adminInfo.ID {
|
|
return l.fail(utils.ErrorBindError)
|
|
}
|
|
|
|
return l.fail(utils.ErrorBindOtherError)
|
|
}
|
|
w = modelbase.Params{Eq: map[string]string{
|
|
"host_code": v.HostCode,
|
|
"user_id": strconv.FormatInt(adminInfo.ID, utils.NumberTen),
|
|
"type": strconv.Itoa(int(in.Type))},
|
|
}
|
|
err = modelObj.GetOne(w, &info)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return l.fail(utils.Fail)
|
|
}
|
|
|
|
info.Status = utils.StatusOk
|
|
info.UserId = adminInfo.ID
|
|
info.UserNickname = adminInfo.Name
|
|
info.HostCode = v.HostCode
|
|
info.HostNickname = v.HostNickname
|
|
info.BindTime = utils.Now()
|
|
info.Type = in.Type
|
|
|
|
w = modelbase.Params{
|
|
Eq: map[string]string{"id": strconv.FormatInt(info.Id, utils.NumberTen)},
|
|
}
|
|
|
|
modelObj.Begin()
|
|
if info.Id < utils.NumberOne {
|
|
err = modelObj.Create(&info)
|
|
} else {
|
|
_, err = modelObj.Edit(w, &info)
|
|
}
|
|
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
modelObj.Rollback()
|
|
return l.fail(utils.Fail)
|
|
}
|
|
|
|
host := dao.BindHistoryCreate{
|
|
HostCode: v.HostCode,
|
|
UserId: adminInfo.ID,
|
|
Status: utils.StatusOk,
|
|
Type: in.Type,
|
|
}
|
|
err = model.BindHistoryModel{}.Init().Create(&host)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
modelObj.Rollback()
|
|
return l.fail(utils.Fail)
|
|
}
|
|
modelObj.Commit()
|
|
return l.ok(utils.NumberOne)
|
|
}
|