新增权限同步

models/company.go

@@ -6,7 +6,7 @@ import (
 )
 
 type Company struct {
-	CompanyId       int64       `orm:"column(company_id);pk"`
+	CompanyId       int64     `orm:"column(company_id);pk"`
 	CompanyName     string    `description:"客户名称"`
 	CreditCode      string    `description:"社会统一信用码"`
 	CompanyCode     string    `description:"客户编码"`
@@ -46,3 +46,9 @@ func AddCompany(item *Company) (newId int64, err error) {
 	newId, err = o.Insert(item)
 	return
 }
+
+func GetCompany() (items []*Company, err error) {
+	sql := `SELECT * FROM company WHERE type IN(1,2,4) `
+	_, err = orm.NewOrm().Raw(sql).QueryRows(&items)
+	return
+}

models/company_report_permission.go

@@ -1,12 +1,27 @@
 package models
 
-import "time"
+import (
+	"rdluck_tools/orm"
+	"time"
+)
 
 type CompanyReportPermission struct {
 	CompanyReportPermissionId int64     `orm:"column(company_report_permission_id);pk"`
-	CompanyId                 int       `description:"是否付费用户"`
-	report_permission_id      int       `description:"是否付费用户"`
-	created_time              time.Time `description:"是否付费用户"`
-	last_updated_time         time.Time `description:"是否付费用户"`
-	chart_permission_id       int       `description:"是否付费用户"`
+	CompanyId                 int64       `description:"公司id"`
+	ReportPermissionId        int       `description:"权限id(作废)"`
+	CreatedTime               time.Time `description:"创建时间"`
+	LastUpdatedTime           time.Time `description:"更新时间"`
+	ChartPermissionId         int       `description:"大分类ID"`
 }
+
+func GetCompanyReportPermission(companyId int64) (items []*CompanyReportPermission, err error) {
+	sql := `SELECT * FROM company_report_permission WHERE company_id=? `
+	_, err = orm.NewOrm().Raw(sql, companyId).QueryRows(&items)
+	return
+}
+
+func AddCompanyReportPermission(item *CompanyReportPermission) (err error) {
+	o:=orm.NewOrm()
+	_,err=o.Insert(item)
+	return
+}

services/company.go

@@ -156,7 +156,7 @@ func ImportCompanyUsers() {
 			fmt.Println("err:", err.Error())
 		}
 	}()
-	fileName := "./company_user_0803.xlsx"
+	fileName := "./company_user_0806.xlsx"
 	xlFile, err := xlsx.OpenFile(fileName)
 	if err != nil {
 		fmt.Printf("open failed: %s\n", err)
@@ -181,7 +181,7 @@ func ImportCompanyUsers() {
 				mobile = strings.Replace(mobile, " ", "", -1)
 				fmt.Println(mobile)
 				var companyId int64
-				companyId=4537
+				companyId=0
 				if companyId > 0 {
 					wxUser, err := models.GetWxUserByMobile(mobile)
 					if err != nil && err.Error() != utils.ErrNoRow() {

services/company_permission.go

@@ -0,0 +1,60 @@
+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
+				}
+			}
+		}
+	}
+}

services/task.go

@@ -13,7 +13,7 @@ func Task() {
 
 	oneMinute := toolbox.NewTask("oneMinute", "0 */1 * * * * ", OneMinute)
 	toolbox.AddTask("oneMinute", oneMinute)
-
+	//ImportCompanyUsers()
 	toolbox.StartTask()
 	fmt.Println("task end")
 }