pm/services/work/worknode/worknode.go

package worknode

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

var WorkNode models.WorkNode

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

	if CheckWorkNodeDuplicate(form.Name) {
		return 0, errors.New("节点已存在")
	}
	workNodeMap := map[string]interface{}{
		"`name`":              form.Name,
		"`color`":             form.Color,
		"`logo`":              form.Logo,
		"`work_type_id`":      form.WorkTypeId,
		"`designer_presence`": form.DesignerPresence,
		"`quality_accept`":    form.QualityAccept,
		"`state`":             form.State,
	}
	workNodeID, err := db.InsertOne(WorkNode.TableName(), workNodeMap)
	if err != nil {
		return 0, nil
	}
	return workNodeID, nil
}

func EditByID(form form.WorkNodeAdd, id int) error {
	info, err := GetInfoByID(id, nil, nil)
	if info == nil {
		return errors.New("invalid id")
	}
	if info.Name != form.Name && CheckWorkNodeDuplicate(form.Name) {
		return errors.New("节点已存在")
	}
	workNodeMap := map[string]interface{}{
		"`name`":              form.Name,
		"`color`":             form.Color,
		"`logo`":              form.Logo,
		"`work_type_id`":      form.WorkTypeId,
		"`designer_presence`": form.DesignerPresence,
		"`quality_accept`":    form.QualityAccept,
		"`state`":             form.State,
	}
	_, err = db.Update(WorkNode.TableName(), map[string]interface{}{"id": id}, workNodeMap)
	return err
}

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

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

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

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

func GetWorkNodesRaw(where map[string]string, param map[string]interface{}, retVal interface{}) error {
	field := "SELECT `zy_work_node`.*, `zy_work_type`.`name` as `work_type_name` FROM `zy_work_node` left join `zy_work_type` on `zy_work_type`.`id` = `zy_work_node`.`work_type_id` "
	return db.GetMultiRaw(field, where, param, retVal)
}

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

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

func CheckWorkNodeDuplicate(name string) bool {
	workNodeInfo, err := GetOne(map[string]interface{}{"name": name, "deleted_at": 0}, nil, nil)
	return workNodeInfo != nil && err == nil
}