mh_goadmin_server/app/admin/models/supplier.go

package models

import (
	"fmt"
	"go-admin/app/admin/models/common"
	orm "go-admin/common/global"
	"go-admin/logger"
	"gorm.io/gorm"
)

type Supplier struct {
	Model

	Number                string `json:"number" gorm:"index"`     //编号
	Name                  string `json:"name"`                    //供应商名称
	Contact               string `json:"contact"`                 //联系人
	Tel                   string `json:"tel"`                     //手机号
	Province              string `json:"province"`                //省
	City                  string `json:"city"`                    //市
	Area                  string `json:"area"`                    //区
	Address               string `json:"address"`                 //详细地址
	AccountHolder         string `json:"account_holder"`          //开户人
	OpeningBank           string `json:"opening_bank"`            //开户银行
	BankAccount           string `json:"bank_account"`            //银行账号
	PaymentCycle          uint32 `json:"payment_cycle"`           //支付周期
	TaxNumber             string `json:"tax_number"`              //税号
	Landline              string `json:"landline"`                //固定电话
	Email                 string `json:"email"`                   //邮件
	CompanyWebsite        string `json:"company_website"`         //网站
	CooperativeBusinessId uint32 `json:"cooperative_business_id"` //合作商id
}

func (s *Supplier) TableName() string {
	return "erp_supplier"
}

func (s *Supplier) BeforeCreate(tx *gorm.DB) error {
	if s.Number == "" {
		var count int64
		err := tx.Table(s.TableName()).
			Unscoped().
			Scopes(common.ScopeCooperativeBusiness(s.CooperativeBusinessId)).
			Count(&count).Error
		if err != nil {
			logger.Error("get supplier count failed", logger.Field("err", err), logger.Field("s", s))
			return err
		}
		s.Number = fmt.Sprintf("%03d", count)
	}
	return nil
}

type GetSupplierRequest struct {
	Name                  string //供应商名称
	Number                string //供应商编号
	CooperativeBusinessId uint32 //合作商id
}

func GetSupplier(req GetSupplierRequest) ([]*Supplier, error) {
	var list []*Supplier
	m := orm.Eloquent.Model(Supplier{})
	if req.Name != "" {
		m = m.Where("name = ?", req.Number)
	}

	if req.Number != "" {
		m = m.Where("number = ?", req.Number)
	}

	m = m.Scopes(common.ScopeCooperativeBusiness(req.CooperativeBusinessId))

	err := m.
		Order("created_at desc").
		Find(&list).
		Error
	if err != nil {
		logger.Error("get supplier list err", logger.Field("err", err), logger.Field("req", req))
		return nil, err
	}

	return list, nil
}