fonchain-approval-fork/pkg/model/approval.go

package model

import (
	"errors"
	"github.com/fonchain_enterprise/fonchain-approval/api/approval"
	"github.com/fonchain_enterprise/fonchain-approval/pkg/m"
	"gorm.io/gorm"
	"gorm.io/plugin/soft_delete"
	"time"
)

// Approval 审批
type Approval struct {
	ID            uint64                `gorm:"primaryKey;column:id" json:"id"`             // ID
	DeletedAt     soft_delete.DeletedAt `gorm:"column:deleted_at" json:"deletedAt"`         // 删除时间
	CreatedAt     time.Time             `gorm:"column:created_at" json:"createdAt"`         // 创建时间
	UpdatedAt     time.Time             `gorm:"column:updated_at" json:"updatedAt"`         // 更新时间
	SubmitterID   uint64                `gorm:"column:submitter_id" json:"submitterId"`     // 提交人ID
	ApproverID    uint64                `gorm:"column:approver_id" json:"approverId"`       // 审核人ID
	Status        int8                  `gorm:"column:status" json:"status"`                // 状态0-待 1-可 2-否
	SubmitterName string                `gorm:"column:submitter_name" json:"submitterName"` // 提交人名称缓存
	ApproverName  string                `gorm:"column:approver_name" json:"approverName"`   // 审核人名称缓存
	CopyUsers     CopyUsers             `gorm:"column:copy_users" json:"copyUsers"`         // 抄送人
	Type          string                `gorm:"column:type" json:"type"`                    // 类型
	Content       string                `gorm:"column:content" json:"content"`              // 提交审批内容
	Reply         string                `gorm:"column:reply" json:"reply"`                  // 回复内容
	Domain        *string               `gorm:"column:domain" json:"domain"`                // 环境变量
	Show          *Show                 `gorm:"foreignKey:ApprovalID" json:"Show"`          // 环境变量
	Work          *Work                 `gorm:"foreignKey:ApprovalID" json:"Work"`          // 环境变量
	Bundle        *Bundle               `gorm:"foreignKey:ApprovalID" json:"Bundle"`        // 环境变量
	Exhibition    *Exhibition           `gorm:"foreignKey:ApprovalID" json:"Exhibition"`    // 环境变量
}

const (
	TypeContent    = "content"
	TypeWork       = "work"
	TypeExhibition = "exhibition"
	TypeShow       = "show"
	TypeBundle     = "bundle"
)
const (
	StatusDoing = 0
	StatusOk    = 1
	StatusFail  = 2
)

type ApprovalContentInterface interface {
	SaveApprovalContent(in *approval.CreateRequest, a *Approval) error
	UpdateApprovalContent(in *approval.CreateRequest, a *Approval) error
	BuildResContent(a *Approval, request *approval.CreateRequest)
	DeleteApproval(p *Approval) error
}

// TableName get sql table name.获取数据库表名
func (m *Approval) TableName() string {
	return "approval"
}

func StoreApproval(in *approval.CreateRequest) (*Approval, error) {
	var entity *Approval

	err := DB.Transaction(func(tx *gorm.DB) error {

		// 保存基本信息
		entity = &Approval{
			Domain:        &in.Domain,
			SubmitterID:   in.SubmitterID,
			SubmitterName: in.SubmitterName,
			ApproverID:    in.ApproverID,
			ApproverName:  in.ApproverName,
			CopyUsers:     FormatCopyUsers(in.CopyUsers),
			Type:          in.Type,
			Content:       in.Content,
		}

		if err := DB.Create(&entity).Error; err != nil {
			return err
		}

		// 不同类型保存不同的结构体
		err := entity.SaveContent(in)

		return err
	})

	return entity, err

}

func Viewed(in *approval.ViewedRequest) error {

	var entity *Approval
	var err error

	if err = DB.First(&entity, in.ID).Error; err != nil {
		return errors.New(m.ErrorNotFound)
	}

	copyUsers := entity.CopyUsers

	for i, temp := range copyUsers {
		if temp.ID == in.UserID {
			temp.IsViewed = true
			copyUsers[i] = temp
		}
	}

	return DB.Model(&Approval{}).Where(&Approval{ID: in.ID}).Updates(&Approval{CopyUsers: copyUsers}).Error

}

func UpdateApproval(in *approval.CreateRequest) (*Approval, error) {
	var entity *Approval

	err := DB.Transaction(func(tx *gorm.DB) error {

		if departErr := DB.First(&entity, in.ID).Error; departErr != nil {
			return errors.New(m.ErrorNotFound)
		}

		if entity.Type != in.Type {
			return errors.New(m.ErrorChangeType)
		}

		if entity.Status != StatusDoing {
			return errors.New(m.ErrorWrongStatus)
		} else if in.Status != StatusDoing { //审批
			// 保存基本信息
			entity = &Approval{
				ID:     in.ID,
				Status: int8(in.Status),
				Reply:  in.Reply,
			}

			return DB.Save(&entity).Error
		}

		// 保存基本信息
		entity = &Approval{
			ID: in.ID,
			//Domain:        in.Domain,
			//SubmitterID:   in.SubmitterId,
			//SubmitterName: in.SubmitterName,
			//ApproverID:   in.ApproverId,
			//ApproverName: in.ApproverName,
			//CopyUsers: FormatCopyUsers(in.CopyUsers),
			//Type:          in.Type,
			Content: in.Content,
		}

		if err := DB.Where(&Approval{ID: in.ID}).Updates(&entity).Error; err != nil {
			return err
		}

		// 不同类型保存不同的结构体
		err := entity.UpdateContent(in)

		return err
	})

	return entity, err

}

func approvalCalc(filter *Approval) *approval.Information {

	var num int64

	info := &approval.Information{}

	DB.Model(&Approval{}).Where(&filter).Count(&num)
	info.Total = uint64(num)

	DB.Model(&Approval{}).Where(&filter).Where(&Approval{Status: StatusOk}).Count(&num)
	info.DoingTotal = uint64(num)

	DB.Model(&Approval{}).Where(&filter).Where(&Approval{Status: StatusOk}).Count(&num)
	info.SuccessTotal = uint64(num)

	DB.Model(&Approval{}).Where(&filter).Where(&Approval{Status: StatusFail}).Count(&num)
	info.FailTotal = uint64(num)

	return info
}

// ApprovalInfo 统计
func ApprovalInfo(in *approval.InformationRequest) (*approval.InformationResponse, error) {

	response := &approval.InformationResponse{
		MySubmitInfo: approvalCalc(&Approval{Domain: &in.Domain, SubmitterID: in.UserID}),
		SubmitMeInfo: approvalCalc(&Approval{Domain: &in.Domain, ApproverID: in.UserID}),
		DomainInfo:   approvalCalc(&Approval{Domain: &in.Domain}),
	}

	return response, nil

}

func (m *Approval) SaveApprovalContent(in *approval.CreateRequest, a *Approval) error {
	return nil
}

func (m *Approval) UpdateApprovalContent(in *approval.CreateRequest, a *Approval) error {
	return nil
}

func (m *Approval) DeleteApproval(p *Approval) error {
	return nil
}

func (m *Approval) DeleteContent() error {

	factory, err := GetApprovalContentFactory(m.Type)

	if err != nil {
		return err
	}

	return factory.DeleteApproval(m)
}

func (m *Approval) BuildResContent(a *Approval, request *approval.CreateRequest) {

}

func (m *Approval) SetResContent(request *approval.CreateRequest) error {

	factory, err := GetApprovalContentFactory(m.Type)

	if err != nil {
		return err
	}

	factory.BuildResContent(m, request)

	return nil
}

func (m *Approval) SaveContent(in *approval.CreateRequest) error {

	factory, err := GetApprovalContentFactory(in.Type)
	if err != nil {
		return err
	}

	err = factory.SaveApprovalContent(in, m)

	return err
}

func (m *Approval) UpdateContent(in *approval.CreateRequest) error {

	factory, err := GetApprovalContentFactory(in.Type)
	if err != nil {
		return err
	}

	err = factory.UpdateApprovalContent(in, m)

	return err
}

func ApprovalList(in *approval.ListRequest) ([]*Approval, int64) {
	var list []*Approval
	var count int64

	//list
	modelObj := DB.Model(&Approval{}).
		Preload("Show").
		Preload("Work.ApprovalWorks").
		Preload("Bundle").
		Preload("Exhibition.ApprovalExhibitions")

	if in.Type != "" {
		modelObj = modelObj.Where(&Approval{Type: in.Type})
	}

	if in.Status != 0 {
		modelObj = modelObj.Where(&Approval{Status: int8(in.Status)})
	}

	modelObj.Count(&count)

	modelObj.Limit(int(in.Limit)).Offset(int(in.Offset)).Order("id desc").Find(&list)

	return list, count
}

func GetApprovalContentFactory(typeFiled string) (ApprovalContentInterface, error) {
	switch typeFiled {
	case TypeContent:
		return &Approval{}, nil
	case TypeShow:
		return &Show{}, nil
	case TypeWork:
		return &Work{}, nil
	case TypeBundle:
		return &Bundle{}, nil
	case TypeExhibition:
		return &Exhibition{}, nil
	default:
		return nil, errors.New(m.ErrorApprovalType)
	}
}

// ApprovalColumns get sql column name.获取数据库列名
var ApprovalColumns = struct {
	ID            string
	DeletedAt     string
	CreatedAt     string
	UpdatedAt     string
	SubmitterID   string
	ApproverID    string
	Status        string
	SubmitterName string
	ApproverName  string
	CopyName      string
	Type          string
	Content       string
	Reply         string
	Domain        string
}{
	ID:            "id",
	DeletedAt:     "deleted_at",
	CreatedAt:     "created_at",
	UpdatedAt:     "updated_at",
	SubmitterID:   "submitter_id",
	ApproverID:    "approver_id",
	Status:        "status",
	SubmitterName: "submitter_name",
	ApproverName:  "approver_name",
	CopyName:      "copy_name",
	Type:          "type",
	Content:       "content",
	Reply:         "reply",
	Domain:        "domain",
}