18 changed files with 1125 additions and 303 deletions
@ -0,0 +1,256 @@ |
|||||
|
package personnelapi |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
"hr_server/grocerystore" |
||||
|
"hr_server/models" |
||||
|
"hr_server/overall" |
||||
|
"hr_server/overall/overallhandle" |
||||
|
"path" |
||||
|
"strconv" |
||||
|
"strings" |
||||
|
"time" |
||||
|
|
||||
|
"encoding/json" |
||||
|
|
||||
|
"github.com/360EntSecGroup-Skylar/excelize" |
||||
|
"github.com/gin-gonic/gin" |
||||
|
) |
||||
|
|
||||
|
/* |
||||
|
* |
||||
|
@ 作者: 秦东 |
||||
|
@ 时间: 2024-02-03 15:38:58 |
||||
|
@ 功能: 解析上传列表 |
||||
|
@ 参数 |
||||
|
|
||||
|
# |
||||
|
|
||||
|
@ 返回值 |
||||
|
|
||||
|
# |
||||
|
|
||||
|
@ 方法原型 |
||||
|
|
||||
|
# |
||||
|
*/ |
||||
|
func (s *StaffApi) UploadUserFilesRedis(c *gin.Context) { |
||||
|
fileInfo, fileHeader, fileErr := c.Request.FormFile("file") |
||||
|
if fileErr != nil { |
||||
|
overallhandle.Result(1, fileErr, c) |
||||
|
return |
||||
|
} |
||||
|
tageExt := path.Ext(fileHeader.Filename) |
||||
|
fmt.Printf("tageExt:%v\n", tageExt) |
||||
|
fileType := overallhandle.JudgeUpFileType(tageExt) |
||||
|
if fileType != 5 { |
||||
|
overallhandle.Result(1, fileType, c, "您上传的不是电子表格！请上传正确的文件！") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
xlsx, err := excelize.OpenReader(fileInfo) |
||||
|
if err != nil { |
||||
|
overallhandle.Result(2, err, c) |
||||
|
return |
||||
|
} |
||||
|
redisListKey := fmt.Sprintf("ExcelImport:PeopleList:%v_%v", overall.CONSTANT_CONFIG.RedisPrefixStr.Alias, overallhandle.OnlyOneNumber(1)) |
||||
|
var userInfoData GroupParsingData |
||||
|
totalNum, meritsYearIng, rewPunYearsIng := userInfoData.ReadExcelRedis(xlsx, redisListKey) |
||||
|
sendData := overallhandle.MapOut() |
||||
|
sendData["redisListKey"] = redisListKey |
||||
|
sendData["totalNum"] = totalNum |
||||
|
sendData["meritsYearIng"] = meritsYearIng |
||||
|
sendData["rewPunYearsIng"] = rewPunYearsIng |
||||
|
overallhandle.Result(0, sendData, c) |
||||
|
} |
||||
|
|
||||
|
// ReadExcel .读取excel 转成切片
|
||||
|
func (g *GroupParsingData) ReadExcelRedis(xlsx *excelize.File, redisKey string) (totalNum int64, meritsYearIng, rewPunYearsIng map[int]int) { |
||||
|
rows := xlsx.GetRows(xlsx.GetSheetName(xlsx.GetActiveSheetIndex())) |
||||
|
isDateYear := time.Now().Year() |
||||
|
//获取绩效年度
|
||||
|
meritsYear := make(map[int]int) |
||||
|
if rows[1][289] != "" { |
||||
|
yearAry := strings.Split(rows[1][289], "年") |
||||
|
if len(yearAry) > 0 { |
||||
|
|
||||
|
yearVal, _ := strconv.Atoi(yearAry[0]) |
||||
|
meritsYear[0] = yearVal |
||||
|
} |
||||
|
} else { |
||||
|
meritsYear[0] = isDateYear - 3 |
||||
|
} |
||||
|
if rows[1][290] != "" { |
||||
|
yearAry := strings.Split(rows[1][290], "年") |
||||
|
if len(yearAry) > 0 { |
||||
|
yearVal, _ := strconv.Atoi(yearAry[0]) |
||||
|
meritsYear[1] = yearVal |
||||
|
} |
||||
|
} else { |
||||
|
meritsYear[1] = isDateYear - 2 |
||||
|
} |
||||
|
if rows[1][291] != "" { |
||||
|
yearAry := strings.Split(rows[1][291], "年") |
||||
|
if len(yearAry) > 0 { |
||||
|
yearVal, _ := strconv.Atoi(yearAry[0]) |
||||
|
meritsYear[2] = yearVal |
||||
|
} |
||||
|
} else { |
||||
|
meritsYear[2] = isDateYear - 1 |
||||
|
} |
||||
|
meritsYearIng = meritsYear |
||||
|
//获取奖惩年度
|
||||
|
rewPunYears := make(map[int]int) |
||||
|
if rows[2][299] != "" { |
||||
|
jcYearAry := strings.Split(rows[2][299], "奖惩") |
||||
|
if len(jcYearAry) > 0 { |
||||
|
|
||||
|
jcYearVal, _ := strconv.Atoi(jcYearAry[0]) |
||||
|
rewPunYears[0] = jcYearVal |
||||
|
} |
||||
|
} else { |
||||
|
rewPunYears[0] = isDateYear - 3 |
||||
|
} |
||||
|
if rows[2][304] != "" { |
||||
|
jcYearAry := strings.Split(rows[2][304], "奖惩") |
||||
|
if len(jcYearAry) > 0 { |
||||
|
|
||||
|
jcYearVal, _ := strconv.Atoi(jcYearAry[0]) |
||||
|
rewPunYears[1] = jcYearVal |
||||
|
} |
||||
|
} else { |
||||
|
rewPunYears[1] = isDateYear - 2 |
||||
|
} |
||||
|
if rows[2][309] != "" { |
||||
|
jcYearAry := strings.Split(rows[2][309], "奖惩") |
||||
|
if len(jcYearAry) > 0 { |
||||
|
|
||||
|
jcYearVal, _ := strconv.Atoi(jcYearAry[0]) |
||||
|
rewPunYears[2] = jcYearVal |
||||
|
} |
||||
|
} else { |
||||
|
rewPunYears[2] = isDateYear - 1 |
||||
|
} |
||||
|
rewPunYearsIng = rewPunYears |
||||
|
|
||||
|
redisClient := grocerystore.RunRedis(overall.CONSTANT_REDIS4) |
||||
|
redisClient.SetRedisTime(86400) |
||||
|
totalNum = 0 |
||||
|
for i, row := range rows { |
||||
|
if i < 4 { |
||||
|
continue |
||||
|
} |
||||
|
if row[1] != "" { |
||||
|
jsonStr, _ := json.Marshal(row) |
||||
|
// fmt.Printf("%v出生日期:%v\n", row[1], string(jsonStr))
|
||||
|
redisClient.Lpush(redisKey, jsonStr) |
||||
|
totalNum++ |
||||
|
} |
||||
|
} |
||||
|
// redisVal, err := redisClient.Lindex(redisKey, 9)
|
||||
|
// fmt.Printf("row:%v,%v\n", redisVal, err)
|
||||
|
return |
||||
|
} |
||||
|
|
||||
|
/* |
||||
|
* |
||||
|
@ 作者: 秦东 |
||||
|
@ 时间: 2024-02-03 16:22:38 |
||||
|
@ 功能: 解析Redis列表中的数据 |
||||
|
@ 参数 |
||||
|
|
||||
|
# |
||||
|
|
||||
|
@ 返回值 |
||||
|
|
||||
|
# |
||||
|
|
||||
|
@ 方法原型 |
||||
|
|
||||
|
# |
||||
|
*/ |
||||
|
func (s *StaffApi) AnalysisRedisExelect(c *gin.Context) { |
||||
|
var requestData AnaRedisKB |
||||
|
err := c.ShouldBindJSON(&requestData) |
||||
|
if err != nil { |
||||
|
overallhandle.Result(100, err, c) |
||||
|
return |
||||
|
} |
||||
|
redisClient := grocerystore.RunRedis(overall.CONSTANT_REDIS4) |
||||
|
redisListLenght, err := redisClient.Llen(requestData.RedisListKey) |
||||
|
if err != nil { |
||||
|
overallhandle.Result(1, err, c, "数据获取识别！没有找到数据！") |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
redisVal, err := redisClient.Lindex(requestData.RedisListKey, int64(requestData.Number)) |
||||
|
if err != nil { |
||||
|
overallhandle.Result(1, err, c, "数据获取识别！没有找到数据！") |
||||
|
return |
||||
|
} |
||||
|
var jieguo []string |
||||
|
err = json.Unmarshal([]byte(redisVal), &jieguo) |
||||
|
if err != nil { |
||||
|
overallhandle.Result(1, err, c, "数据格式不正取！") |
||||
|
} |
||||
|
manCont := make(map[int]string) |
||||
|
for i, v := range jieguo { |
||||
|
manCont[i] = v |
||||
|
} |
||||
|
meritsYear := make(map[int]int) |
||||
|
for i, v := range requestData.MeritsYearIng { |
||||
|
iInt, _ := strconv.Atoi(i) |
||||
|
meritsYear[iInt] = v |
||||
|
} |
||||
|
rewPunYears := make(map[int]int) |
||||
|
for i, v := range requestData.RewPunYearsIng { |
||||
|
iInt, _ := strconv.Atoi(i) |
||||
|
rewPunYears[iInt] = v |
||||
|
} |
||||
|
var groupManCont GroupParsingData |
||||
|
groupManCont.GroupParsingDataInfoRedis(manCont, meritsYear, rewPunYears) |
||||
|
// fmt.Printf("执行完222毕删除列表键%v-%v-%v\n", redisListLenght, requestData.Number, redisListLenght-1 == int64(requestData.Number))
|
||||
|
if redisListLenght-1 == int64(requestData.Number) { |
||||
|
//执行完毕删除列表键
|
||||
|
redisClient.Lrange(requestData.RedisListKey, 0, -1) |
||||
|
redisClient.DelKey(requestData.RedisListKey) |
||||
|
// fmt.Printf("执行完毕删除列表键%v\n", requestData.RedisListKey)
|
||||
|
} |
||||
|
overallhandle.Result(0, groupManCont, c) |
||||
|
} |
||||
|
|
||||
|
/* |
||||
|
* |
||||
|
@ 作者: 秦东 |
||||
|
@ 时间: 2024-01-30 08:10:16 |
||||
|
@ 功能: 分组数据解析 |
||||
|
@ 参数 |
||||
|
|
||||
|
#info //人员信息
|
||||
|
#meritsYear 绩效年度 |
||||
|
#rewPunYearsmap 奖惩年度 |
||||
|
|
||||
|
@ 返回值 |
||||
|
|
||||
|
# |
||||
|
|
||||
|
@ 方法原型 |
||||
|
|
||||
|
# |
||||
|
*/ |
||||
|
func (g *GroupParsingData) GroupParsingDataInfoRedis(list map[int]string, meritsYear, rewPunYearsmap map[int]int) { |
||||
|
g.IsOk = false |
||||
|
g.UserNum = list[1] |
||||
|
if list[1] != "" { |
||||
|
var myCont models.PersonArchives |
||||
|
err := myCont.GetCont(map[string]interface{}{"`number`": list[1]}, "`key`") |
||||
|
if err != nil { |
||||
|
g.IsOk = true |
||||
|
} else { |
||||
|
g.UserKey = myCont.Key |
||||
|
} |
||||
|
g.ProcessMainTable(list, meritsYear, rewPunYearsmap) |
||||
|
} else { |
||||
|
g.Msg = append(g.Msg, fmt.Sprintf("序号：%v--->%v：没有工号--->%v", list[0], list[2], time.Now().Unix())) |
||||
|
} |
||||
|
} |
||||
Binary file not shown.
Loading…
Reference in new issue