hongze_task/services/company_permission.go

package services

import (
	"fmt"
	"hongze/hongze_task/models"
	"time"
)

//客户权限设置
func SetCompanyPermission() {
	var err error
	defer func() {
		if err != nil {
			fmt.Println("SetCompanyPermission Err:", err.Error())
		}
	}()
	companyList, err := models.GetCompany()
	fmt.Println("len:", len(companyList))
	time.Sleep(10 * time.Second)
	if err != nil {
		return
	}
	for k, v := range companyList {
		fmt.Println("line 24:",k,v.CompanyName)
		permissionList, err := models.GetCompanyReportPermission(v.CompanyId)
		if err != nil {
			return
		}
		if len(permissionList) == 14 {
			fmt.Println(k, v.CompanyName, "大套餐客户")
		} else {
			fmt.Println(k, v.CompanyName, "小套餐客户")
			flag := 0
			for m, n := range permissionList {
				fmt.Println(m, n.ChartPermissionId)
				if n.ChartPermissionId == 17 {
					flag = 1
				}
				if n.ChartPermissionId == 8 {
					flag = 2
				}
			}

			if flag == 0 || flag == 1 {
				//新增炉料权限
				item := new(models.CompanyReportPermission)
				item.CompanyId = v.CompanyId
				item.ChartPermissionId = 8
				item.CreatedTime = time.Now()
				item.LastUpdatedTime = time.Now()
				item.ReportPermissionId = 8
				err = models.AddCompanyReportPermission(item)
				if err != nil {
					fmt.Println("AddCompanyReportPermission Err:",err.Error())
					//return
				}
			}
		}
	}
}