eta_mini_bridge/models/chart_permission.go

package models

import (
	"eta/eta_mini_bridge/utils"
	"fmt"
	"strings"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

type ChartPermission struct {
	ChartPermissionId     int       `orm:"column(chart_permission_id);pk" description:"问题ID"`
	ChartPermissionName   string    `description:"名称"`
	PermissionName        string    `description:"权限名"`
	Sort                  int       `description:"排序"`
	Enabled               int       `description:"是否可用"`
	CreatedTime           time.Time `description:"创建时间"`
	LastUpdatedTime       time.Time `description:"更新时间"`
	TeleconferenceSort    int       `description:"电话会类型排序"`
	Remark                string    `description:"备注"`
	ClassifyName          string    `description:"分类名称"`
	ProductName           string    `description:"产品名称" `
	ProductId             int       `description:"产品ID"`
	ImageURL              string    `orm:"column(image_url);" description:"图片地址"`
	ShowType              int       `description:"1:查研观向小程序展示"`
	IsOther               int       `description:"是否是其他，用于查研观向小程序后台展示"`
	IsReport              int       `description:"是否是报告，用于查研观向小程序前台报告展示"`
	CygxAuth              int       `description:"是否是权限，用于查研观向小程序前台权限校验"`
	PermissionType        int       `description:"1主观，2客观"`
	YbImgUrl              string    `description:"研报小程序报告列表icon"`
	ProductPermissionName string    `description:"种类权限名称"`
	PriceDrivenState      int       `description:"品种价格驱动开启状态 0-关闭 1-开启"`
	ImageUrlM             string    `description:"图片地址(查研观向移动端)"`
	ParentId              int       `description:"父级权限id"`
	IsPublic              int       `description:"是否是公有权限1:公有权限，0私有权限"`
}

// GetChildChartPermissionListById 获取品种权限列表
func GetChartPermissionList() (items []*ChartPermission, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT * FROM chart_permission WHERE enabled=1 ORDER BY sort ASC`
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

// GetChildChartPermissionListById 获取品种权限列表
func GetChildChartPermissionListById(chartPermissionId int) (items []*ChartPermission, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT * FROM chart_permission WHERE enabled=1 AND parent_id=? ORDER BY sort ASC`
	_, err = o.Raw(sql, chartPermissionId).QueryRows(&items)
	return
}

// GetChartPermissionListById 获取品种权限列表
func GetChartPermissionIdsByIds(chartPermissionIds []string) (items []*ChartPermission, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT chart_permission_id FROM chart_permission WHERE enabled=1 `
	if len(chartPermissionIds) > 0 {
		sql += fmt.Sprintf(" AND chart_permission_id IN (%s) ", strings.Join(chartPermissionIds, ","))
	}
	sql += ` ORDER BY sort ASC `

	_, err = o.Raw(sql).QueryRows(&items)
	return
}

// GetChildChartPermissionListById 获取品种权限列表
func GetChildChartPermissionIdsById(chartPermissionId int) (items []string, err error) {
	o := orm.NewOrmUsingDB("rddp")
	sql := `SELECT chart_permission_id FROM chart_permission WHERE enabled=1 AND parent_id=? ORDER BY sort ASC`
	_, err = o.Raw(sql, chartPermissionId).QueryRows(&items)
	return
}

// GetDefaultChartPermissionCount 获取默认的权限数目
func GetDefaultChartPermissionCount(productId int) (count int, err error) {
	sql := ` select count(1) COUNT from chart_permission where product_id=?`
	o := orm.NewOrm()
	err = o.Raw(sql, productId).QueryRow(&count)
	return
}

func GetDefaultChartPermissionCountByCache(productId int) (maxPermissionCount int, err error) {
	maxPermissionCountKey := fmt.Sprint(`hongze_api:max_permission_count:product_id:`, productId)
	maxPermissionCount, err = utils.Rc.RedisInt(maxPermissionCountKey)
	if err != nil || maxPermissionCount <= 0 {
		maxPermissionCount, err = GetDefaultChartPermissionCount(productId)
		if err != nil {
			return
		}
		utils.Rc.SetNX(maxPermissionCountKey, maxPermissionCount, time.Second*3600)
	}
	//ficc需要移除策略的权限
	if productId == 1 {
		maxPermissionCount = maxPermissionCount - 1
	}
	return
}