eta_bridge/task/htfutures/sync_staff_task.go

package HTtask

import (
	"errors"
	"eta/eta_bridge/global"
	"eta/eta_bridge/models/ht/oa"
	"eta/eta_bridge/models/ht/wx_crm"
	"fmt"
	"github.com/robfig/cron/v3"
)

func StartCronJob() {
	// 创建一个新的 cron 实例
	c := cron.New(cron.WithSeconds())
	// 添加定时任务
	_, err := c.AddFunc("*/5 * * * * *", func() {
		syncCrmOAInfo()
	})

	if err != nil {
		fmt.Println("添加定时任务失败:", err)
		return
	}

	// 启动定时任务
	c.Start()

	// 在程序结束时停止定时任务
	go func() {
		<-make(chan struct{})
		c.Stop()
	}()
}

func SyncETA() {
	//list, err := oa.GetSubCompanyList()
	//if err != nil {
	//	return
	//}
	//var SysDepartmentList []eta.SysDepartment
	//for _, item := range list {
	//	sysDepartment := eta.SysDepartment{
	//		DepartmentName: item.Subcompanyname,
	//		OutID:          item.Companyid,
	//	}
	//	SysDepartmentList = append(SysDepartmentList, sysDepartment)
	//}
}

func syncCrmOAInfo() {
	//先同步分公司列表
	subCompanylist, err := oa.GetSubCompanyList()
	if err != nil {
		global.FILE_LOG.Error("获取海通OA子公司数据失败:", err.Error())
	}
	var sysSubCompanyList []wx_crm.SysDepartment
	for _, item := range subCompanylist {
		sysDepartment := wx_crm.SysDepartment{
			SysDepartmentName: item.SUBCOMPANYNAME,
			Sort:              item.SHOWORDEROFTREE,
			Level:             item.TLEVEL,
			OutId:             item.ID,
			Type:              wx_crm.SubCompany,
			ParentId:          item.SUPSUBCOMID,
		}
		sysSubCompanyList = append(sysSubCompanyList, sysDepartment)
	}
	err = wx_crm.BatchInsertOrUpdate(sysSubCompanyList)
	if err != nil {
		global.FILE_LOG.Error("同步海通OA子公司数据失败:", err.Error())
		return
	}
	//同步部门信息
	DepartmentList, err := oa.GetDepartMentList()
	if err != nil {
		global.FILE_LOG.Error("获取海通部门数据失败:", err.Error())
	}
	var sysDepartmentList []wx_crm.SysDepartment
	for _, item := range DepartmentList {
		sysDepartment := wx_crm.SysDepartment{
			SysDepartmentName: item.DEPARTMENTNAME,
			Sort:              item.SHOWORDEROFTREE,
			Level:             item.TLEVEL,
			OutId:             item.ID,
			Type:              wx_crm.Department,
			ParentId:          item.SUPDEPID,
		}
		if sysDepartment.Level == 2 && sysDepartment.ParentId == 0 {
			sysDepartment.ParentId = item.SUBCOMPANYID1
		}
		sysDepartmentList = append(sysDepartmentList, sysDepartment)
	}
	err = wx_crm.BatchInsertOrUpdate(sysDepartmentList)
	if err != nil {
		global.FILE_LOG.Error("同步海通OA子公司数据失败:", err.Error())
		return
	}
	//同步员工信息
	staffList, err := oa.GetEmployeeList()
	if err != nil {
		global.FILE_LOG.Error("获取海通OA员工数据失败:", err.Error())
	}
	var Employee []wx_crm.SysUser
	for _, item := range staffList {
		idList, departmentErr := GetEmployeeDepartmentList(item.DEPARTMENTID)
		if departmentErr != nil {
			global.FILE_LOG.Error("获取员工部门组织架构失败:", departmentErr.Error(), "跳过同步员工", item.WORKCODE)
			continue
		}
		fillSize := 4 - len(idList)
		var departments []int
		if fillSize > 0 {
			for i := 0; i < fillSize; i++ {
				departments = append(departments, 0)
			}
		}
		sysUser := wx_crm.SysUser{
			SysUserName:      item.LOGINID,
			SysRealName:      item.LASTNAME,
			Email:            item.EMAIL,
			Phone:            item.MOBILE,
			AreaCode:         "86",
			SysRoleId:        63,
			SysRoleName:      "普通用户",
			SysDepartmentId:  item.DEPARTMENTID,
			SysDepartmentId1: departments[3],
			SysDepartmentId2: departments[2],
			SysDepartmentId3: departments[1],
			SysDepartmentId4: departments[0],
			PositionCode:     item.SECLEVEL,
			IsEnabled:        true,
		}
		Employee = append(Employee, sysUser)
	}
	err = wx_crm.BatchInsertOrUpdateUser(Employee)
	if err != nil {
		global.FILE_LOG.Error("同步海通OA员工数据失败:", err.Error())
		return
	}
}

func GetEmployeeDepartmentList(departmentId int) (departmentIdList []int, err error) {
	departmentList, err := wx_crm.GetDepartmentList()
	if err != nil {
		global.FILE_LOG.Error("获取部门信息失败", err.Error())
		return
	}
	var currentDepartment wx_crm.SysDepartment
	found := false
	for _, item := range departmentList {
		if item.OutId == departmentId {
			currentDepartment = item
			found = true
			break
		}
	}
	if found {
		departmentIdList = append(departmentIdList, currentDepartment.OutId)
		subList := DepartmentList(currentDepartment.ParentId, departmentList, currentDepartment.Level-1)
		departmentIdList = append(departmentIdList, subList...)
	} else {
		err = errors.New("当前部门id不存在")
	}
	return
}

func DepartmentList(departmentId int, departmentList []wx_crm.SysDepartment, level int) (departmentIdList []int) {
	if level == 0 || departmentId == 0 {
		return
	}
	for _, item := range departmentList {
		if item.OutId == departmentId {
			if level > 1 && item.Type == wx_crm.SubCompany {
				continue
			}
			if level == 1 && item.Type == wx_crm.Department {
				continue
			}
			departmentIdList = append(departmentIdList, item.OutId)
			subSubDepartments := DepartmentList(item.ParentId, departmentList, item.Level-1)
			departmentIdList = append(departmentIdList, subSubDepartments...)

		}
	}
	return departmentIdList
}