mmm-go / partnermg · 提交

转到一个项目

作者 tangxvhui
提交 1660aaa46e869ca2d00832e6e472a263ad586a26 1 个父辈 dbbc11b7

增加基本的请求参数校验

内嵌 并排对比
正在显示 3 个修改的文件 包含 29 行增加 和 0 行删除
1 package command 1 package command
2 2
  3 +import (
  4 + "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/lib"
  5 +)
  6 +
3 type SaveAdminUserCommand struct { 7 type SaveAdminUserCommand struct {
4 Id int64 `json:"id"` 8 Id int64 `json:"id"`
5 // 员工姓名 9 // 员工姓名
@@ -13,3 +17,13 @@ type SaveAdminUserCommand struct { @@ -13,3 +17,13 @@ type SaveAdminUserCommand struct {
13 17
14 IsUsable bool `json:"isUsable"` 18 IsUsable bool `json:"isUsable"`
15 } 19 }
  20 +
  21 +func (command SaveAdminUserCommand) ValidateCommand() error {
  22 + if len(command.Name) == 0 {
  23 + return lib.ThrowError(lib.ARG_ERROR, "用户名称必填")
  24 + }
  25 + if len(command.Account) == 0 {
  26 + return lib.ThrowError(lib.ARG_ERROR, "账号必填")
  27 + }
  28 + return nil
  29 +}
1 package command 1 package command
2 2
  3 +import "gitlab.fjmaimaimai.com/mmm-go/partnermg/pkg/lib"
  4 +
3 //UpdateAdminUserPwdCommand 修改密码 5 //UpdateAdminUserPwdCommand 修改密码
4 type UpdateAdminUserPwdCommand struct { 6 type UpdateAdminUserPwdCommand struct {
5 Id int64 `json:"id"` 7 Id int64 `json:"id"`
6 //密码 8 //密码
7 Password string `json:"password"` 9 Password string `json:"password"`
8 } 10 }
  11 +
  12 +func (command UpdateAdminUserPwdCommand) ValidateCommand() error {
  13 + if len(command.Password) == 0 {
  14 + return lib.ThrowError(lib.ARG_ERROR, "密码必填")
  15 + }
  16 + return nil
  17 +}
@@ -49,6 +49,9 @@ func (adminUserSrv AdminUserService) GetAdminUser(getAdminUserQuery *query.GetAd @@ -49,6 +49,9 @@ func (adminUserSrv AdminUserService) GetAdminUser(getAdminUserQuery *query.GetAd
49 } 49 }
50 50
51 func (adminUserSrv AdminUserService) SaveAdminUser(saveUserCmd *command.SaveAdminUserCommand) (*domain.AdminUser, error) { 51 func (adminUserSrv AdminUserService) SaveAdminUser(saveUserCmd *command.SaveAdminUserCommand) (*domain.AdminUser, error) {
  52 + if err := saveUserCmd.ValidateCommand(); err != nil {
  53 + return nil, lib.ThrowError(lib.ARG_ERROR, err.Error())
  54 + }
52 //实际业务 55 //实际业务
53 transactionContext, err := factory.CreateTransactionContext(nil) 56 transactionContext, err := factory.CreateTransactionContext(nil)
54 if err != nil { 57 if err != nil {
@@ -192,6 +195,9 @@ func (adminUserSrv AdminUserService) PageListAdminUser(listAdminUserQuery *query @@ -192,6 +195,9 @@ func (adminUserSrv AdminUserService) PageListAdminUser(listAdminUserQuery *query
192 } 195 }
193 196
194 func (adminUserSrv AdminUserService) UpdateAdminPassword(updatecmd command.UpdateAdminUserPwdCommand) error { 197 func (adminUserSrv AdminUserService) UpdateAdminPassword(updatecmd command.UpdateAdminUserPwdCommand) error {
  198 + if err := updatecmd.ValidateCommand(); err != nil {
  199 + return lib.ThrowError(lib.ARG_ERROR, err.Error())
  200 + }
195 transactionContext, err := factory.CreateTransactionContext(nil) 201 transactionContext, err := factory.CreateTransactionContext(nil)
196 if err != nil { 202 if err != nil {
197 return lib.ThrowError(lib.TRANSACTION_ERROR, err.Error()) 203 return lib.ThrowError(lib.TRANSACTION_ERROR, err.Error())