cas_tt_cloud_backend/app/basic/controller/shelf.go

package controller

import (
	"SciencesServer/app/common/model"
	"SciencesServer/app/session"
	"errors"
	"time"
)

type Shelf struct {
	*session.Admin
	*session.Enterprise
}

type ShelfOption func(object *Shelf)

type IModelInfo struct {
	model.Model
	UID uint64 `json:"uid"`
	model.ModelTenant
	model.Shelf
}

func WithShelfSessionAdmin(session *session.Admin) ShelfOption {
	return func(object *Shelf) {
		object.Admin = session
	}
}

func WithShelfSessionEnterprise(session *session.Enterprise) ShelfOption {
	return func(object *Shelf) {
		object.Enterprise = session
	}
}

func (c *Shelf) Handle(iModel model.IModel, id uint64, callback func(kind model.ShelfStatusKind) error) error {
	if c.Admin == nil && c.Enterprise == nil {
		return errors.New("操作错误，未知的人员操作")
	}
	iModel.SetID(id)
	out := new(IModelInfo)

	err := model.ScanFields(iModel, out, []string{"id", "tenant_id", "uid", "shelf_status"},
		&model.ModelWhereOrder{Where: model.NewWhere("id", id)})

	if err != nil {
		return err
	} else if out.ID <= 0 {
		return errors.New("操作错误，数据信息不存在或已被删除")
	}

	if c.Admin != nil {
		if c.Admin.TenantID > 0 && c.Admin.TenantID != out.TenantID {
			return errors.New("操作错误，无权限操作")
		}
	} else {
		if c.Enterprise.UID != out.UID {
			return errors.New("操作错误，无权限操作")
		}
	}
	shelfStatus := model.ShelfStatusForUp

	if out.ShelfStatus == model.ShelfStatusForUp {
		shelfStatus = model.ShelfStatusForDown
	}
	values := map[string]interface{}{
		"shelf_status": shelfStatus,
		"updated_at":   time.Now(),
	}
	if err := model.Updates(iModel, values); err != nil {
		return err
	}
	if callback != nil {
		return callback(shelfStatus)
	}
	return nil
}

func NewShelf(options ...ShelfOption) *Shelf {
	out := new(Shelf)

	for _, option := range options {
		option(out)
	}
	return out
}