construction_management/services/rule_service.go

package services

import (
	"encoding/json"
	"errors"
	"fmt"
	"log"
	"reflect"
	"strconv"

	"go.mod/web/utils"

	"github.com/kataras/iris/v12"
	"go.mod/dao"
	"go.mod/datasource"
	"go.mod/web/viewmodels"
)

type RuleService interface {
	Get(pid int, id int) viewmodels.ViewRule
	Post(pid int, id int, key string, value string) error
	AutoCode(bid int, pid int, codeType string) (string, error)
	ValidRule(ctx iris.Context) (viewmodels.ValidField, error)
}

// //返回service操作类
type ruleService struct {
	daoRule       *dao.RuleDao
	daoSafe       *dao.SafeDao
	daoQuality    *dao.QualityDao
	daoContract   *dao.ContractDao
	validAutoPath string
}

//创建项目用户service
func NewRuleService() RuleService {
	return &ruleService{
		validAutoPath: "/api/rule/auto",
		daoRule:       dao.NewRuleDao(datasource.InstanceDbMaster()),
		daoSafe:       dao.NewSafeDao(datasource.InstanceDbMaster()),
		daoQuality:    dao.NewQualityDao(datasource.InstanceDbMaster()),
		daoContract:   dao.NewContractDao(datasource.InstanceDbMaster()),
	}
}

func (s *ruleService) Get(pid int, id int) viewmodels.ViewRule {
	data := s.daoRule.FindByPidWithBid(pid, id)
	viewData := viewmodels.ViewRule{SafeRule: data.SafeRule, QualityRule: data.QualityRule, ContractPaidRule: data.ContractPaidRule, ContractReturnRule: data.ContractReturnRule}
	return viewData
}

func (s *ruleService) Post(pid int, id int, key string, value string) error {
	err := s.daoRule.UpdateOrCreate(pid, id, key, value)
	return err
}

// 生成code
func (s *ruleService) AutoCode(bid int, pid int, codeType string) (string, error) {
	fmt.Println("codeType---------------------------------------", codeType)
	// 获取该标段的规则
	rule := s.daoRule.FindByPidWithBid(pid, bid)
	if codeType == "safeRule" {
		if rule.SafeRule == "" {
			return "", errors.New("该标段未设置编号规则!")
		}
		var code viewmodels.RuleCode
		err := json.Unmarshal([]byte(rule.SafeRule), &code)
		if err == nil {
			total, err := s.daoSafe.CountRuleCode(bid)
			value := reflect.ValueOf(code)
			for i := 0; i < value.NumField(); i++ {
				b := fmt.Sprint(value.Field(i))
				if b == code.Code {
					k, _ := strconv.Atoi(b)
					code.Code = utils.CreateRuleCode(int64(k), total, len(code.Code))
				}
			}

			e, err := json.Marshal(code)
			if err != nil {
				return "", err
			}
			return string(e), err
		}
	} else if codeType == "contractReturnRule" {
		if rule.ContractReturnRule == "" {
			return "", errors.New("该标段未设置编号规则!")
		}
		var code viewmodels.RuleCode
		err := json.Unmarshal([]byte(rule.ContractReturnRule), &code)
		if err == nil {
			total, err := s.daoContract.CountRuleCode(bid, 0)
			if err != nil {
				return "", err
			}
			value := reflect.ValueOf(code)
			for i := 0; i < value.NumField(); i++ {
				b := fmt.Sprint(value.Field(i))
				if b == code.Code {
					k, _ := strconv.Atoi(b)
					code.Code = utils.CreateRuleCode(int64(k), total, len(code.Code))
				}
			}

			e, err := json.Marshal(code)
			if err != nil {
				return "", err
			}
			return string(e), err
		}
	} else if codeType == "contractPaidRule" {
		if rule.ContractReturnRule == "" {
			return "", errors.New("该标段未设置编号规则!")
		}
		var code viewmodels.RuleCode
		err := json.Unmarshal([]byte(rule.ContractPaidRule), &code)
		if err == nil {
			total, err := s.daoContract.CountRuleCode(bid, 1)
			if err != nil {
				return "", err
			}
			value := reflect.ValueOf(code)
			for i := 0; i < value.NumField(); i++ {
				b := fmt.Sprint(value.Field(i))
				if b == code.Code {
					k, _ := strconv.Atoi(b)
					code.Code = utils.CreateRuleCode(int64(k), total, len(code.Code))
				}
			}

			e, err := json.Marshal(code)
			if err != nil {
				return "", err
			}
			return string(e), err
		}
	} else {
		if rule.QualityRule == "" {
			return "", errors.New("该标段未设置编号规则!")
		}
		var code viewmodels.RuleCode
		err := json.Unmarshal([]byte(rule.QualityRule), &code)
		if err == nil {
			total, err := s.daoQuality.CountRuleCode(bid)
			value := reflect.ValueOf(code)
			for i := 0; i < value.NumField(); i++ {
				b := fmt.Sprint(value.Field(i))
				if b == code.Code {
					k, _ := strconv.Atoi(b)
					code.Code = utils.CreateRuleCode(int64(k), total, len(code.Code))
				}
			}

			e, err := json.Marshal(code)
			if err != nil {
				return "", err
			}
			return string(e), err
		}
	}
	return "", errors.New("生成code失败")
}

func (s *ruleService) ValidRule(ctx iris.Context) (viewmodels.ValidField, error) {
	safeVaild := viewmodels.ValidField{}
	if ctx.Method() == "GET" {
		err := ctx.ReadForm(&safeVaild)
		if err != nil {
			log.Println("safe-ValidRule-ReadForm转换异常, error=", err)
			return safeVaild, err
		}
		err = safeVaild.Validate()
		return safeVaild, err
	}

	if ctx.Method() == "POST" {
		err := ctx.ReadJSON(&safeVaild)
		if err != nil {
			log.Println("safe-ValidRule-ReadJson转换异常, error=", err)
			return safeVaild, err
		}
		if ctx.Path() == s.validAutoPath {
			err = safeVaild.ValidateAuto()
		} else {
			err = safeVaild.Validate()
		}

	}
	return safeVaild, nil

}