package logic import ( "context" "lone-services/pkg/utils" "lone-services/services/water/dao" "lone-services/services/water/model" "lone-services/services/water/validator" "lone-services/rpc/water/pb" "lone-services/services/water/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type AddLogLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewAddLogLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AddLogLogic { return &AddLogLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *AddLogLogic) AddLog(in *water.AddLogRequest) (*water.Response, error) { var v validator.LogAddValidator 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) } info := dao.LogCreate{ Data: in.Data, HostCode: in.HostCode, UserId: adminInfo.ID, } err := model.LogModel{}.Init().Create(&info) if err != nil { return l.fail(utils.Fail) } return l.ok(utils.NumberOne) }