pm/services/work/workcraft/workcraft.go

package workcraft

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

var WorkCraft models.WorkCraft

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

	if CheckWorkCraftDuplicate(form.NodeId, form.Name) {
		return 0, errors.New("工艺已存在")
	}
	workCraftMap := map[string]interface{}{
		"`node_id`":  form.NodeId,
		"`name`":     form.Name,
		"`cover`":    form.Cover,
		"`pictures`": form.Pictures,
		"`purpose`":  form.Purpose,
		"`content`":  form.Content,
		"`state`":    form.State,
	}
	workCraftID, err := db.InsertOne(WorkCraft.TableName(), workCraftMap)
	if err != nil {
		return 0, nil
	}
	return workCraftID, nil
}

func EditByID(form form.WorkCraftAdd, id int) error {
	info, err := GetInfoByID(id, nil, nil)
	if info == nil {
		return errors.New("invalid id")
	}
	if info.Name != form.Name && CheckWorkCraftDuplicate(form.NodeId, form.Name) {
		return errors.New("工艺已存在")
	}
	workCraftMap := map[string]interface{}{
		"`node_id`":  form.NodeId,
		"`name`":     form.Name,
		"`cover`":    form.Cover,
		"`pictures`": form.Pictures,
		"`purpose`":  form.Purpose,
		"`content`":  form.Content,
		"`state`":    form.State,
	}
	_, err = db.Update(WorkCraft.TableName(), map[string]interface{}{"id": id}, workCraftMap)
	return err
}

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

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

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

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

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

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

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

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