增加租户套餐
This commit is contained in:
@@ -0,0 +1,837 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
"server/pkg/jwtutil"
|
||||
"server/services"
|
||||
|
||||
beego "github.com/beego/beego/v2/server/web"
|
||||
)
|
||||
|
||||
// PlatformTenantPackageController 平台端「租户套餐」管理
|
||||
//
|
||||
// 两块能力：
|
||||
// 1. 功能套餐（yz_system_tenant_package + yz_system_tenant_package_module）：
|
||||
// 勾选套餐包含的功能模块（erp / oa / crm / cms ...），租户绑定后按其开通功能。
|
||||
// 2. 用户数配额（yz_system_tenant_user_quota_package + yz_system_tenant_quota_order）：
|
||||
// 每个租户有用户数上限（默认 20），支持按单个用户或按加购套餐增购。
|
||||
type PlatformTenantPackageController struct {
|
||||
beego.Controller
|
||||
}
|
||||
|
||||
func (c *PlatformTenantPackageController) jsonErr(httpStatus, bizCode int, msg string) {
|
||||
c.Ctx.Output.SetStatus(httpStatus)
|
||||
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
func (c *PlatformTenantPackageController) jsonOK(data interface{}, msg string) {
|
||||
if msg == "" {
|
||||
msg = "success"
|
||||
}
|
||||
resp := map[string]interface{}{"code": 200, "msg": msg}
|
||||
if data != nil {
|
||||
resp["data"] = data
|
||||
}
|
||||
c.Data["json"] = resp
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
// currentOperator 解析当前平台账号名（用于增购记录留痕；取不到时返回空串，不阻断请求）
|
||||
func (c *PlatformTenantPackageController) currentOperator() string {
|
||||
auth := c.Ctx.Request.Header.Get("Authorization")
|
||||
if auth == "" {
|
||||
return ""
|
||||
}
|
||||
parts := strings.SplitN(auth, " ", 2)
|
||||
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||
return ""
|
||||
}
|
||||
claims, err := jwtutil.ParseToken(parts[1])
|
||||
if err != nil || claims == nil {
|
||||
return ""
|
||||
}
|
||||
if strings.TrimSpace(claims.Username) != "" {
|
||||
return claims.Username
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func (c *PlatformTenantPackageController) readBody(target interface{}) bool {
|
||||
raw, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return false
|
||||
}
|
||||
if err := json.Unmarshal(raw, target); err != nil {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func parseUint64Param(raw string) uint64 {
|
||||
v, _ := strconv.ParseUint(strings.TrimSpace(raw), 10, 64)
|
||||
return v
|
||||
}
|
||||
|
||||
// ============================ 功能套餐 ============================
|
||||
|
||||
type tenantPackageModuleItem struct {
|
||||
ModuleCode string `json:"module_code"`
|
||||
ModuleName string `json:"module_name"`
|
||||
}
|
||||
|
||||
type tenantPackageDTO struct {
|
||||
models.SystemTenantPackage
|
||||
Modules []tenantPackageModuleItem `json:"modules"`
|
||||
}
|
||||
|
||||
type tenantPackagePayload struct {
|
||||
Name string `json:"name"`
|
||||
Code string `json:"code"`
|
||||
Description string `json:"description"`
|
||||
Price *float64 `json:"price"`
|
||||
UserQuota *int `json:"user_quota"`
|
||||
ExtraUserPrice *float64 `json:"extra_user_price"`
|
||||
IsDefault *int8 `json:"is_default"`
|
||||
Sort *int `json:"sort"`
|
||||
Status *int8 `json:"status"`
|
||||
Remark string `json:"remark"`
|
||||
Modules []string `json:"modules"`
|
||||
}
|
||||
|
||||
// loadPackageModules 批量加载套餐模块（packageID -> 模块列表）
|
||||
func loadPackageModules(packageIDs []uint64) map[uint64][]tenantPackageModuleItem {
|
||||
result := map[uint64][]tenantPackageModuleItem{}
|
||||
if len(packageIDs) == 0 {
|
||||
return result
|
||||
}
|
||||
var rows []models.SystemTenantPackageModule
|
||||
_, err := models.Orm.QueryTable(new(models.SystemTenantPackageModule)).
|
||||
Filter("package_id__in", packageIDs).
|
||||
OrderBy("id").
|
||||
All(&rows)
|
||||
if err != nil {
|
||||
return result
|
||||
}
|
||||
for _, r := range rows {
|
||||
result[r.PackageID] = append(result[r.PackageID], tenantPackageModuleItem{
|
||||
ModuleCode: r.ModuleCode,
|
||||
ModuleName: r.ModuleName,
|
||||
})
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// GetPackageList GET /platform/tenantPackage/list
|
||||
func (c *PlatformTenantPackageController) GetPackageList() {
|
||||
keyword := strings.TrimSpace(c.GetString("keyword"))
|
||||
status := strings.TrimSpace(c.GetString("status"))
|
||||
|
||||
qs := models.Orm.QueryTable(new(models.SystemTenantPackage)).Filter("delete_time__isnull", true)
|
||||
if keyword != "" {
|
||||
qs = qs.Filter("name__icontains", keyword)
|
||||
}
|
||||
if status == "0" || status == "1" {
|
||||
v, _ := strconv.Atoi(status)
|
||||
qs = qs.Filter("status", v)
|
||||
}
|
||||
|
||||
var rows []models.SystemTenantPackage
|
||||
if _, err := qs.OrderBy("sort", "id").All(&rows); err != nil {
|
||||
c.jsonErr(500, 500, "获取失败："+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
ids := make([]uint64, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
ids = append(ids, r.ID)
|
||||
}
|
||||
moduleMap := loadPackageModules(ids)
|
||||
|
||||
// 每个套餐的租户数
|
||||
tenantCount := map[uint64]int{}
|
||||
if len(ids) > 0 {
|
||||
var tenants []models.SystemTenant
|
||||
if _, err := models.Orm.QueryTable(new(models.SystemTenant)).All(&tenants, "ID", "PackageID"); err == nil {
|
||||
for _, t := range tenants {
|
||||
tenantCount[t.PackageID]++
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
list := make([]map[string]interface{}, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
modules := moduleMap[r.ID]
|
||||
if modules == nil {
|
||||
modules = []tenantPackageModuleItem{}
|
||||
}
|
||||
list = append(list, map[string]interface{}{
|
||||
"id": r.ID,
|
||||
"name": r.Name,
|
||||
"code": r.Code,
|
||||
"description": r.Description,
|
||||
"price": r.Price,
|
||||
"user_quota": r.UserQuota,
|
||||
"extra_user_price": r.ExtraUserPrice,
|
||||
"is_default": r.IsDefault,
|
||||
"sort": r.Sort,
|
||||
"status": r.Status,
|
||||
"remark": r.Remark,
|
||||
"create_time": r.CreateTime,
|
||||
"update_time": r.UpdateTime,
|
||||
"modules": modules,
|
||||
"module_codes": moduleCodes(modules),
|
||||
"tenant_count": tenantCount[r.ID],
|
||||
})
|
||||
}
|
||||
|
||||
c.jsonOK(map[string]interface{}{"list": list, "total": len(list)}, "获取成功")
|
||||
}
|
||||
|
||||
func moduleCodes(items []tenantPackageModuleItem) []string {
|
||||
codes := make([]string, 0, len(items))
|
||||
for _, it := range items {
|
||||
codes = append(codes, it.ModuleCode)
|
||||
}
|
||||
return codes
|
||||
}
|
||||
|
||||
// GetPackageSelectList GET /platform/tenantPackage/select/list
|
||||
func (c *PlatformTenantPackageController) GetPackageSelectList() {
|
||||
var rows []models.SystemTenantPackage
|
||||
_, err := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("delete_time__isnull", true).
|
||||
Filter("status", 1).
|
||||
OrderBy("sort", "id").
|
||||
All(&rows)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "获取失败："+err.Error())
|
||||
return
|
||||
}
|
||||
ids := make([]uint64, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
ids = append(ids, r.ID)
|
||||
}
|
||||
moduleMap := loadPackageModules(ids)
|
||||
|
||||
list := make([]map[string]interface{}, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
modules := moduleMap[r.ID]
|
||||
if modules == nil {
|
||||
modules = []tenantPackageModuleItem{}
|
||||
}
|
||||
list = append(list, map[string]interface{}{
|
||||
"id": r.ID,
|
||||
"name": r.Name,
|
||||
"code": r.Code,
|
||||
"description": r.Description,
|
||||
"price": r.Price,
|
||||
"user_quota": r.UserQuota,
|
||||
"extra_user_price": r.ExtraUserPrice,
|
||||
"is_default": r.IsDefault,
|
||||
"modules": modules,
|
||||
"module_codes": moduleCodes(modules),
|
||||
})
|
||||
}
|
||||
c.jsonOK(list, "获取成功")
|
||||
}
|
||||
|
||||
// GetPackageDetail GET /platform/tenantPackage/detail/:id
|
||||
func (c *PlatformTenantPackageController) GetPackageDetail() {
|
||||
id := parseUint64Param(c.Ctx.Input.Param(":id"))
|
||||
if id == 0 {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
var row models.SystemTenantPackage
|
||||
if err := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
One(&row); err != nil {
|
||||
c.jsonErr(404, 404, "套餐不存在")
|
||||
return
|
||||
}
|
||||
modules := loadPackageModules([]uint64{id})[id]
|
||||
if modules == nil {
|
||||
modules = []tenantPackageModuleItem{}
|
||||
}
|
||||
c.jsonOK(tenantPackageDTO{SystemTenantPackage: row, Modules: modules}, "获取成功")
|
||||
}
|
||||
|
||||
// GetModuleOptions GET /platform/tenantPackage/moduleOptions
|
||||
// 可加入套餐的功能模块清单（取 yz_system_modules 中启用且未删除的模块）
|
||||
func (c *PlatformTenantPackageController) GetModuleOptions() {
|
||||
var rows []models.SystemModules
|
||||
_, err := models.Orm.QueryTable(new(models.SystemModules)).
|
||||
Filter("delete_time__isnull", true).
|
||||
Filter("status", 1).
|
||||
OrderBy("sort", "id").
|
||||
All(&rows)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "获取失败："+err.Error())
|
||||
return
|
||||
}
|
||||
list := make([]map[string]interface{}, 0, len(rows))
|
||||
for _, r := range rows {
|
||||
list = append(list, map[string]interface{}{
|
||||
"id": r.ID,
|
||||
"name": r.Name,
|
||||
"code": r.Code,
|
||||
"path": r.Path,
|
||||
"icon": r.Icon,
|
||||
"description": r.Description,
|
||||
"type": r.Type,
|
||||
"sort": r.Sort,
|
||||
})
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"list": list, "total": len(list)}, "获取成功")
|
||||
}
|
||||
|
||||
// savePackageModules 覆盖式保存套餐模块
|
||||
func savePackageModules(packageID uint64, modules []string) error {
|
||||
if _, err := models.Orm.QueryTable(new(models.SystemTenantPackageModule)).
|
||||
Filter("package_id", packageID).
|
||||
Delete(); err != nil {
|
||||
return err
|
||||
}
|
||||
if len(modules) == 0 {
|
||||
return nil
|
||||
}
|
||||
nameMap := map[string]string{}
|
||||
var allModules []models.SystemModules
|
||||
if _, err := models.Orm.QueryTable(new(models.SystemModules)).
|
||||
Filter("delete_time__isnull", true).
|
||||
All(&allModules, "ID", "Name", "Code"); err == nil {
|
||||
for _, m := range allModules {
|
||||
nameMap[m.Code] = m.Name
|
||||
}
|
||||
}
|
||||
seen := map[string]bool{}
|
||||
for _, code := range modules {
|
||||
code = strings.TrimSpace(code)
|
||||
if code == "" || seen[code] {
|
||||
continue
|
||||
}
|
||||
seen[code] = true
|
||||
row := &models.SystemTenantPackageModule{
|
||||
PackageID: packageID,
|
||||
ModuleCode: code,
|
||||
ModuleName: nameMap[code],
|
||||
}
|
||||
if _, err := models.Orm.Insert(row); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CreatePackage POST /platform/tenantPackage/create
|
||||
func (c *PlatformTenantPackageController) CreatePackage() {
|
||||
var p tenantPackagePayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
p.Name = strings.TrimSpace(p.Name)
|
||||
p.Code = strings.TrimSpace(p.Code)
|
||||
if p.Name == "" {
|
||||
c.jsonErr(400, 400, "套餐名称不能为空")
|
||||
return
|
||||
}
|
||||
if p.Code == "" {
|
||||
c.jsonErr(400, 400, "套餐编码不能为空")
|
||||
return
|
||||
}
|
||||
cnt, _ := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("code", p.Code).
|
||||
Filter("delete_time__isnull", true).
|
||||
Count()
|
||||
if cnt > 0 {
|
||||
c.jsonErr(400, 400, "套餐编码已存在")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
row := &models.SystemTenantPackage{
|
||||
Name: p.Name,
|
||||
Code: p.Code,
|
||||
Description: strings.TrimSpace(p.Description),
|
||||
Price: valueFloat(p.Price, 0),
|
||||
UserQuota: valueInt(p.UserQuota, services.DefaultTenantUserQuota),
|
||||
ExtraUserPrice: valueFloat(p.ExtraUserPrice, services.DefaultTenantExtraUserPrice),
|
||||
IsDefault: valueInt8(p.IsDefault, 0),
|
||||
Sort: valueInt(p.Sort, 0),
|
||||
Status: valueInt8(p.Status, 1),
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
UpdateTime: &now,
|
||||
}
|
||||
if row.UserQuota <= 0 {
|
||||
row.UserQuota = services.DefaultTenantUserQuota
|
||||
}
|
||||
id, err := models.Orm.Insert(row)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "创建失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if err := savePackageModules(uint64(id), p.Modules); err != nil {
|
||||
c.jsonErr(500, 500, "套餐功能保存失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if row.IsDefault == 1 {
|
||||
clearOtherDefaultPackages(uint64(id))
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"id": uint64(id)}, "创建成功")
|
||||
}
|
||||
|
||||
// EditPackage POST /platform/tenantPackage/edit/:id
|
||||
func (c *PlatformTenantPackageController) EditPackage() {
|
||||
id := parseUint64Param(c.Ctx.Input.Param(":id"))
|
||||
if id == 0 {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
var p tenantPackagePayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
p.Name = strings.TrimSpace(p.Name)
|
||||
p.Code = strings.TrimSpace(p.Code)
|
||||
if p.Name == "" {
|
||||
c.jsonErr(400, 400, "套餐名称不能为空")
|
||||
return
|
||||
}
|
||||
if p.Code != "" {
|
||||
cnt, _ := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("code", p.Code).
|
||||
Filter("id__ne", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Count()
|
||||
if cnt > 0 {
|
||||
c.jsonErr(400, 400, "套餐编码已存在")
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
update := map[string]interface{}{
|
||||
"name": p.Name,
|
||||
"description": strings.TrimSpace(p.Description),
|
||||
"extra_user_price": valueFloat(p.ExtraUserPrice, services.DefaultTenantExtraUserPrice),
|
||||
"sort": valueInt(p.Sort, 0),
|
||||
"status": valueInt8(p.Status, 1),
|
||||
"remark": strings.TrimSpace(p.Remark),
|
||||
"update_time": now,
|
||||
}
|
||||
if p.Code != "" {
|
||||
update["code"] = p.Code
|
||||
}
|
||||
if p.Price != nil {
|
||||
update["price"] = *p.Price
|
||||
}
|
||||
if p.UserQuota != nil {
|
||||
quota := *p.UserQuota
|
||||
if quota <= 0 {
|
||||
quota = services.DefaultTenantUserQuota
|
||||
}
|
||||
update["user_quota"] = quota
|
||||
}
|
||||
if p.IsDefault != nil {
|
||||
update["is_default"] = *p.IsDefault
|
||||
update["status"] = 1
|
||||
}
|
||||
|
||||
if _, err := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(update); err != nil {
|
||||
c.jsonErr(500, 500, "编辑失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if p.Modules != nil {
|
||||
if err := savePackageModules(id, p.Modules); err != nil {
|
||||
c.jsonErr(500, 500, "套餐功能保存失败："+err.Error())
|
||||
return
|
||||
}
|
||||
}
|
||||
if p.IsDefault != nil && *p.IsDefault == 1 {
|
||||
clearOtherDefaultPackages(id)
|
||||
}
|
||||
c.jsonOK(nil, "编辑成功")
|
||||
}
|
||||
|
||||
// clearOtherDefaultPackages 保证全局只有一个默认套餐
|
||||
func clearOtherDefaultPackages(keepID uint64) {
|
||||
_, _ = models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("id__ne", keepID).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(map[string]interface{}{"is_default": 0})
|
||||
}
|
||||
|
||||
// DeletePackage DELETE /platform/tenantPackage/delete/:id（软删）
|
||||
func (c *PlatformTenantPackageController) DeletePackage() {
|
||||
id := parseUint64Param(c.Ctx.Input.Param(":id"))
|
||||
if id == 0 {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
// 有租户正在使用该套餐时不允许删除
|
||||
used, _ := models.Orm.QueryTable(new(models.SystemTenant)).Filter("package_id", id).Count()
|
||||
if used > 0 {
|
||||
c.jsonErr(400, 400, fmt.Sprintf("该套餐已被 %d 个租户使用，请先调整租户套餐", used))
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
n, err := models.Orm.QueryTable(new(models.SystemTenantPackage)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(map[string]interface{}{"delete_time": now, "update_time": now, "is_default": 0})
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "删除失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "套餐不存在")
|
||||
return
|
||||
}
|
||||
_, _ = models.Orm.QueryTable(new(models.SystemTenantPackageModule)).Filter("package_id", id).Delete()
|
||||
c.jsonOK(nil, "删除成功")
|
||||
}
|
||||
|
||||
func valueFloat(v *float64, def float64) float64 {
|
||||
if v == nil {
|
||||
return def
|
||||
}
|
||||
return *v
|
||||
}
|
||||
|
||||
func valueInt(v *int, def int) int {
|
||||
if v == nil {
|
||||
return def
|
||||
}
|
||||
return *v
|
||||
}
|
||||
|
||||
// ============================ 用户数加购套餐 ============================
|
||||
|
||||
type quotaPackagePayload struct {
|
||||
Name string `json:"name"`
|
||||
UserCount *int `json:"user_count"`
|
||||
Price *float64 `json:"price"`
|
||||
Sort *int `json:"sort"`
|
||||
Status *int8 `json:"status"`
|
||||
Remark string `json:"remark"`
|
||||
}
|
||||
|
||||
// GetQuotaPackageList GET /platform/tenantQuotaPackage/list
|
||||
func (c *PlatformTenantPackageController) GetQuotaPackageList() {
|
||||
onlyEnabled := c.GetString("enabled") == "1"
|
||||
qs := models.Orm.QueryTable(new(models.SystemTenantUserQuotaPackage)).Filter("delete_time__isnull", true)
|
||||
if onlyEnabled {
|
||||
qs = qs.Filter("status", 1)
|
||||
}
|
||||
var rows []models.SystemTenantUserQuotaPackage
|
||||
if _, err := qs.OrderBy("sort", "id").All(&rows); err != nil {
|
||||
c.jsonErr(500, 500, "获取失败："+err.Error())
|
||||
return
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"list": rows, "total": len(rows)}, "获取成功")
|
||||
}
|
||||
|
||||
// CreateQuotaPackage POST /platform/tenantQuotaPackage/create
|
||||
func (c *PlatformTenantPackageController) CreateQuotaPackage() {
|
||||
var p quotaPackagePayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
p.Name = strings.TrimSpace(p.Name)
|
||||
if p.Name == "" {
|
||||
c.jsonErr(400, 400, "规格名称不能为空")
|
||||
return
|
||||
}
|
||||
if p.UserCount == nil || *p.UserCount <= 0 {
|
||||
c.jsonErr(400, 400, "包含用户数必须大于 0")
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
row := &models.SystemTenantUserQuotaPackage{
|
||||
Name: p.Name,
|
||||
UserCount: *p.UserCount,
|
||||
Price: valueFloat(p.Price, 0),
|
||||
Sort: valueInt(p.Sort, 0),
|
||||
Status: valueInt8(p.Status, 1),
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
UpdateTime: &now,
|
||||
}
|
||||
id, err := models.Orm.Insert(row)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "创建失败："+err.Error())
|
||||
return
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"id": uint64(id)}, "创建成功")
|
||||
}
|
||||
|
||||
// EditQuotaPackage POST /platform/tenantQuotaPackage/edit/:id
|
||||
func (c *PlatformTenantPackageController) EditQuotaPackage() {
|
||||
id := parseUint64Param(c.Ctx.Input.Param(":id"))
|
||||
if id == 0 {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
var p quotaPackagePayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
p.Name = strings.TrimSpace(p.Name)
|
||||
if p.Name == "" {
|
||||
c.jsonErr(400, 400, "规格名称不能为空")
|
||||
return
|
||||
}
|
||||
if p.UserCount != nil && *p.UserCount <= 0 {
|
||||
c.jsonErr(400, 400, "包含用户数必须大于 0")
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
update := map[string]interface{}{
|
||||
"name": p.Name,
|
||||
"sort": valueInt(p.Sort, 0),
|
||||
"status": valueInt8(p.Status, 1),
|
||||
"remark": strings.TrimSpace(p.Remark),
|
||||
"update_time": now,
|
||||
}
|
||||
if p.UserCount != nil {
|
||||
update["user_count"] = *p.UserCount
|
||||
}
|
||||
if p.Price != nil {
|
||||
update["price"] = *p.Price
|
||||
}
|
||||
n, err := models.Orm.QueryTable(new(models.SystemTenantUserQuotaPackage)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(update)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "编辑失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "规格不存在")
|
||||
return
|
||||
}
|
||||
c.jsonOK(nil, "编辑成功")
|
||||
}
|
||||
|
||||
// DeleteQuotaPackage DELETE /platform/tenantQuotaPackage/delete/:id（软删）
|
||||
func (c *PlatformTenantPackageController) DeleteQuotaPackage() {
|
||||
id := parseUint64Param(c.Ctx.Input.Param(":id"))
|
||||
if id == 0 {
|
||||
c.jsonErr(400, 400, "参数错误")
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
n, err := models.Orm.QueryTable(new(models.SystemTenantUserQuotaPackage)).
|
||||
Filter("id", id).
|
||||
Filter("delete_time__isnull", true).
|
||||
Update(map[string]interface{}{"delete_time": now, "update_time": now})
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "删除失败："+err.Error())
|
||||
return
|
||||
}
|
||||
if n == 0 {
|
||||
c.jsonErr(404, 404, "规格不存在")
|
||||
return
|
||||
}
|
||||
c.jsonOK(nil, "删除成功")
|
||||
}
|
||||
|
||||
// ============================ 租户套餐 / 用户数配额 ============================
|
||||
|
||||
type setTenantPackagePayload struct {
|
||||
Tid uint64 `json:"tid"`
|
||||
PackageID uint64 `json:"package_id"`
|
||||
SyncQuota *bool `json:"sync_quota"`
|
||||
}
|
||||
|
||||
// SetTenantPackage POST /platform/tenant/setPackage
|
||||
// 为租户绑定套餐；sync_quota 为 true（默认）时按套餐包含用户数与现有上限取较大值同步。
|
||||
func (c *PlatformTenantPackageController) SetTenantPackage() {
|
||||
var p setTenantPackagePayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
if p.Tid == 0 || p.PackageID == 0 {
|
||||
c.jsonErr(400, 400, "租户与套餐不能为空")
|
||||
return
|
||||
}
|
||||
syncQuota := true
|
||||
if p.SyncQuota != nil {
|
||||
syncQuota = *p.SyncQuota
|
||||
}
|
||||
newQuota, err := services.ApplyPackageToTenant(p.Tid, p.PackageID, syncQuota)
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"user_quota": newQuota}, "套餐设置成功")
|
||||
}
|
||||
|
||||
// GetTenantQuotaInfo GET /platform/tenant/quotaInfo?tid=1
|
||||
// 返回租户套餐、用户数使用情况、可增购规格与单价
|
||||
func (c *PlatformTenantPackageController) GetTenantQuotaInfo() {
|
||||
tid, _ := c.GetUint64("tid")
|
||||
if tid == 0 {
|
||||
c.jsonErr(400, 400, "tid 不能为空")
|
||||
return
|
||||
}
|
||||
info, err := services.GetTenantQuotaInfo(tid)
|
||||
if err != nil {
|
||||
c.jsonErr(404, 404, "租户不存在")
|
||||
return
|
||||
}
|
||||
|
||||
var quotaPackages []models.SystemTenantUserQuotaPackage
|
||||
_, _ = models.Orm.QueryTable(new(models.SystemTenantUserQuotaPackage)).
|
||||
Filter("delete_time__isnull", true).
|
||||
Filter("status", 1).
|
||||
OrderBy("sort", "id").
|
||||
All("aPackages)
|
||||
|
||||
modules, _ := services.GetTenantPackageModules(tid)
|
||||
moduleList := make([]tenantPackageModuleItem, 0, len(modules))
|
||||
for _, m := range modules {
|
||||
moduleList = append(moduleList, tenantPackageModuleItem{ModuleCode: m.ModuleCode, ModuleName: m.ModuleName})
|
||||
}
|
||||
|
||||
c.jsonOK(map[string]interface{}{
|
||||
"tid": info.Tid,
|
||||
"package_id": info.PackageID,
|
||||
"effective_package_id": info.EffectivePackageID,
|
||||
"package_name": info.PackageName,
|
||||
"modules": moduleList,
|
||||
"quota": info.Quota,
|
||||
"used": info.Used,
|
||||
"remaining": info.Remaining,
|
||||
"extra_user_price": info.ExtraUserPrice,
|
||||
"quota_packages": quotaPackages,
|
||||
}, "获取成功")
|
||||
}
|
||||
|
||||
type rechargeQuotaPayload struct {
|
||||
Tid uint64 `json:"tid"`
|
||||
Type int8 `json:"type"` // 1 单个增购 2 套餐增购
|
||||
QuotaPackageID uint64 `json:"quota_package_id"`
|
||||
UserCount int `json:"user_count"`
|
||||
UnitPrice float64 `json:"unit_price"`
|
||||
Remark string `json:"remark"`
|
||||
}
|
||||
|
||||
// RechargeTenantQuota POST /platform/tenant/rechargeQuota
|
||||
// 用户数增购：单个增购（单价 x 人数）或 按加购套餐（如 5人/500元）
|
||||
func (c *PlatformTenantPackageController) RechargeTenantQuota() {
|
||||
var p rechargeQuotaPayload
|
||||
if !c.readBody(&p) {
|
||||
return
|
||||
}
|
||||
if p.Tid == 0 {
|
||||
c.jsonErr(400, 400, "租户不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
order := models.SystemTenantQuotaOrder{
|
||||
Tid: p.Tid,
|
||||
Type: p.Type,
|
||||
Operator: c.currentOperator(),
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
}
|
||||
if order.Type != 2 {
|
||||
order.Type = 1
|
||||
}
|
||||
|
||||
if order.Type == 2 {
|
||||
var pkg models.SystemTenantUserQuotaPackage
|
||||
if err := models.Orm.QueryTable(new(models.SystemTenantUserQuotaPackage)).
|
||||
Filter("id", p.QuotaPackageID).
|
||||
Filter("delete_time__isnull", true).
|
||||
One(&pkg); err != nil {
|
||||
c.jsonErr(400, 400, "加购规格不存在")
|
||||
return
|
||||
}
|
||||
if pkg.UserCount <= 0 {
|
||||
c.jsonErr(400, 400, "加购规格用户数配置有误")
|
||||
return
|
||||
}
|
||||
order.QuotaPackageID = pkg.ID
|
||||
order.PackageName = pkg.Name
|
||||
order.UserCount = pkg.UserCount
|
||||
order.Amount = pkg.Price
|
||||
order.UnitPrice = round2(pkg.Price / float64(pkg.UserCount))
|
||||
} else {
|
||||
if p.UserCount <= 0 {
|
||||
c.jsonErr(400, 400, "增购用户数必须大于 0")
|
||||
return
|
||||
}
|
||||
unitPrice := p.UnitPrice
|
||||
if unitPrice <= 0 {
|
||||
// 未传单价时取租户套餐的单个用户增购价
|
||||
if info, err := services.GetTenantQuotaInfo(p.Tid); err == nil && info.ExtraUserPrice > 0 {
|
||||
unitPrice = info.ExtraUserPrice
|
||||
} else {
|
||||
unitPrice = services.DefaultTenantExtraUserPrice
|
||||
}
|
||||
}
|
||||
order.UserCount = p.UserCount
|
||||
order.UnitPrice = unitPrice
|
||||
order.Amount = round2(unitPrice * float64(p.UserCount))
|
||||
order.PackageName = "单个增购"
|
||||
}
|
||||
|
||||
before, after, err := services.IncreaseTenantQuota(p.Tid, order.UserCount)
|
||||
if err != nil {
|
||||
c.jsonErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
order.QuotaBefore = before
|
||||
order.QuotaAfter = after
|
||||
|
||||
if _, err := models.Orm.Insert(&order); err != nil {
|
||||
c.jsonErr(500, 500, "增购记录写入失败："+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
if info, err := services.GetTenantQuotaInfo(p.Tid); err == nil {
|
||||
c.jsonOK(map[string]interface{}{
|
||||
"quota": info.Quota,
|
||||
"used": info.Used,
|
||||
"order": order,
|
||||
"amount": order.Amount,
|
||||
"add": order.UserCount,
|
||||
}, "增购成功")
|
||||
return
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"quota": after, "add": order.UserCount, "amount": order.Amount}, "增购成功")
|
||||
}
|
||||
|
||||
// GetQuotaOrders GET /platform/tenant/quotaOrders?tid=1
|
||||
func (c *PlatformTenantPackageController) GetQuotaOrders() {
|
||||
tid, _ := c.GetUint64("tid")
|
||||
if tid == 0 {
|
||||
c.jsonErr(400, 400, "tid 不能为空")
|
||||
return
|
||||
}
|
||||
var rows []models.SystemTenantQuotaOrder
|
||||
_, err := models.Orm.QueryTable(new(models.SystemTenantQuotaOrder)).
|
||||
Filter("tid", tid).
|
||||
OrderBy("-id").
|
||||
All(&rows)
|
||||
if err != nil {
|
||||
c.jsonErr(500, 500, "获取失败："+err.Error())
|
||||
return
|
||||
}
|
||||
c.jsonOK(map[string]interface{}{"list": rows, "total": len(rows)}, "获取成功")
|
||||
}
|
||||
Reference in New Issue
Block a user