eta_mini_bridge/models/chart_permission.go

package models

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

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

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

// 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
}

// 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
}