audit_company.go 1.4 KB
package command

import (
	"fmt"
	"reflect"
	"strings"

	"github.com/beego/beego/v2/core/validation"
)

type AuditCompanyCommand struct {
	// 企业id
	CompanyId int64 `cname:"企业id" json:"companyId" valid:"Required"`
	// 手机号
	Phone string `cname:"手机号" json:"phone" valid:"Required"`
	// 审核状态
	Status int `cname:"审核状态" json:"status" valid:"Required"`
	// 备注
	Remark string `cname:"备注" json:"remark" valid:"Required"`
}

func (updateCompanyCommand *AuditCompanyCommand) Valid(validation *validation.Validation) {
	if !(updateCompanyCommand.Status == 1 || updateCompanyCommand.Status == 2) {
		validation.Error("状态有误")
		return
	}
	if updateCompanyCommand.Status == 1 {
		updateCompanyCommand.Status = 2
		return
	}
	if updateCompanyCommand.Status == 2 {
		updateCompanyCommand.Status = 3
		return
	}
}

func (updateCompanyCommand *AuditCompanyCommand) ValidateCommand() error {
	valid := validation.Validation{}
	b, err := valid.Valid(updateCompanyCommand)
	if err != nil {
		return err
	}
	if !b {
		elem := reflect.TypeOf(updateCompanyCommand).Elem()
		for _, validErr := range valid.Errors {
			field, isExist := elem.FieldByName(validErr.Field)
			if isExist {
				return fmt.Errorf(strings.Replace(validErr.Message, validErr.Field, field.Tag.Get("cname"), -1))
			} else {
				return fmt.Errorf(validErr.Message)
			}
		}
	}
	return nil
}