pm/services/work/workprocess/workprocess.go

package workprocess

import (
	"errors"
	"zhiyuan/models"
	"zhiyuan/pkg/app"
	"zhiyuan/pkg/db"
	"zhiyuan/services/form"
)

var WorkProcess models.WorkProcess

func Add(form form.WorkProcessAdd) (int64, error) {

	if CheckWorkProcessDuplicate(form.NodeId, form.Name) {
		return 0, errors.New("工序已存在")
	}
	workProcessMap := map[string]interface{}{
		"`node_id`": form.NodeId,
		"`name`":    form.Name,
		"`state`":   form.State,
	}
	workProcessID, err := db.InsertOne(WorkProcess.TableName(), workProcessMap)
	if err != nil {
		return 0, nil
	}
	return workProcessID, nil
}

func EditByID(form form.WorkProcessAdd, id int) error {
	info, err := GetInfoByID(id, nil, nil)
	if info == nil {
		return errors.New("invalid id")
	}
	if info.Name != form.Name && CheckWorkProcessDuplicate(form.NodeId, form.Name) {
		return errors.New("工序已存在")
	}
	workProcessMap := map[string]interface{}{
		"`node_id`": form.NodeId,
		"`name`":    form.Name,
		"`state`":   form.State,
	}
	_, err = db.Update(WorkProcess.TableName(), map[string]interface{}{"id": id}, workProcessMap)
	return err
}

func DeleteByID(id int) error {
	info, _ := GetInfoByID(id, nil, nil)
	if info == nil {
		return errors.New("invalid id")
	}
	_, err := db.DeleteSoft(WorkProcess.TableName(), map[string]interface{}{"id": id})
	return err
}

func Count(where map[string]interface{}) (int64, error) {
	return db.Count(WorkProcess.TableName(), where)
}

func CountRaw(where string, param map[string]interface{}) (int64, error) {
	query := "`zy_work_process` WHERE " + where
	return db.CountRaw(query, param)
}

func GetList(where map[string]interface{}, fields []string, page app.Page, retVal interface{}) ([]*models.WorkProcess, error) {
	if page.PageNum > 0 && page.PageSize > 0 {
		where["_limit"] = db.GetOffset(uint(page.PageNum), uint(page.PageSize))
	}
	return WorkProcess.GetMulti(where, fields, retVal)
}

func GetWorkProcesssRaw(where map[string]string, param map[string]interface{}, retVal interface{}) ([]*models.WorkProcess, error) {
	return WorkProcess.GetMultiRaw(where, param, retVal)
}

func GetOne(where map[string]interface{}, fields []string, retVal interface{}) (*models.WorkProcess, error) {
	return WorkProcess.GetOne(where, fields, retVal)
}

func GetInfoByID(id int, fields []string, retVal interface{}) (*models.WorkProcess, error) {
	return GetOne(map[string]interface{}{"id": id}, fields, retVal)
}

func CheckWorkProcessDuplicate(nodeId int, name string) bool {
	workProcessInfo, err := GetOne(map[string]interface{}{"node_id": nodeId, "name": name, "deleted_at": 0}, nil, nil)
	return workProcessInfo != nil && err == nil
}