作者 tangxvhui

优化改造

... ... @@ -31,181 +31,25 @@ func GetChanceCheckResultInfo(chanceId int64, companyid int64) (protocol.Reponse
log.Error("机会的自查内容筛选状态错误")
return rsp, protocol.NewErrWithMessage("1")
}
rsp, err = getChanceCheckResultData(chanceId)
return rsp, err
}
//getChanceCheckResultData 机会自查内容筛选详情数据输出
// func getChanceCheckResultData(chanceId int64) (protocol.ReponseChanceCheckResult, error) {
// var (
// checkResultData []models.ChanceCheckResult
// err error
// )
// rsp := protocol.ReponseChanceCheckResult{
// CustomItem: []string{},
// CheckData: []protocol.ChanceCheckResultData{},
// }
// checkResultData, err = models.GetChanceCheckResultByChancce(chanceId)
// if err != nil {
// log.Error("未获得指定的机会自查内容筛选结果,err;%s", err)
// return rsp, nil
// }
// //组装响应数据
// //自查内容中的用户id
// userIds := utils.NewArraySetInt64()
// //一维自查内容项
// checkItemsLv1 := utils.NewArraySetInt64()
// checkItemsLv2 := map[int64]*utils.ArraySetInt64{}
// checkResultMap := make(map[int64]*protocol.ChanceCheckResultData)
// //处理一维自查内容数据
// for _, v := range checkResultData {
// if v.CheckPid > 0 {
// continue
// }
// if _, ok := checkResultMap[v.CheckId]; ok {
// checkResultMap[v.CheckId].CustomItemData[fmt.Sprint(v.UserCompanyId)] = v.Answer
// switch v.Answer {
// case "是":
// checkResultMap[v.CheckId].AnswerYes++
// case "否":
// checkResultMap[v.CheckId].AnswerNo++
// case "不清楚":
// checkResultMap[v.CheckId].AnswerAny++
// }
// } else {
// userAnswer := map[string]string{
// fmt.Sprint(v.UserCompanyId): v.Answer,
// }
// r := &protocol.ChanceCheckResultData{
// CheckPid: v.CheckPid,
// CheckId: v.CheckId,
// CheckItem: v.CheckItem,
// CustomItemData: userAnswer,
// }
// switch v.Answer {
// case "是":
// r.AnswerYes++
// case "否":
// r.AnswerNo++
// case "不清楚":
// r.AnswerAny++
// }
// checkResultMap[v.CheckId] = r
// userIds.Add(v.UserCompanyId)
// checkItemsLv1.Add(v.CheckId)
// }
// }
// //处理二维自查内容数据
// for _, v := range checkResultData {
// if v.CheckPid == 0 {
// continue
// }
// _, ok := checkResultMap[v.CheckPid]
// if !ok {
// continue
// }
// _, ok = checkResultMap[v.CheckPid].Child[v.CheckId]
// if ok {
// checkResultMap[v.CheckPid].Child[v.CheckId].CustomItemData[fmt.Sprint(v.UserCompanyId)] = v.Answer
// switch v.Answer {
// case "是":
// checkResultMap[v.CheckId].AnswerYes++
// case "否":
// checkResultMap[v.CheckId].AnswerNo++
// case "不清楚":
// checkResultMap[v.CheckId].AnswerAny++
// }
// } else {
// userAnswer := map[string]string{
// fmt.Sprint(v.UserCompanyId): v.Answer,
// }
// r := &protocol.ChanceCheckResultData{
// CheckPid: v.CheckPid,
// CheckId: v.CheckId,
// CheckItem: v.CheckItem,
// CustomItemData: userAnswer,
// }
// switch v.Answer {
// case "是":
// r.AnswerYes++
// case "否":
// r.AnswerNo++
// case "不清楚":
// r.AnswerAny++
// }
// checkResultMap[v.CheckPid].Child[v.CheckId] = r
// userIds.Add(v.UserCompanyId)
// if _, ok := checkItemsLv2[v.CheckPid]; ok {
// checkItemsLv2[v.CheckPid].Add(v.CheckId)
// } else {
// checkItemsLv2[v.CheckPid] = utils.NewArraySetInt64()
// checkItemsLv2[v.CheckPid].Add(v.CheckId)
// }
// }
// }
// usernameMap := make(map[string]string)
// for _, v := range userIds.ToSlice() {
// idString := fmt.Sprint(v)
// u, err := models.GetUserCompanyById(v)
// if err == nil {
// usernameMap[idString] = u.NickName
// } else {
// usernameMap[idString] = idString
// }
// rsp.CustomItem = append(rsp.CustomItem, usernameMap[idString])
// }
// for _, Lv1 := range checkItemsLv1.ToSlice() {
// // 处理一级数据
// newCustomItemData := map[string]string{}
// for uid, uname := range usernameMap {
// if answer, ok := checkResultMap[Lv1].CustomItemData[uid]; ok {
// newCustomItemData[uname] = answer
// } else {
// newCustomItemData[uname] = ""
// }
// }
// checkResultMap[Lv1].CustomItemData = newCustomItemData
// checkResultMap[Lv1].Total = fmt.Sprintf("是:%d 否:%d 不清楚:%d",
// checkResultMap[Lv1].AnswerYes, checkResultMap[Lv1].AnswerNo, checkResultMap[Lv1].AnswerAny)
// rsp.CheckData = append(rsp.CheckData, *checkResultMap[Lv1])
// if _, ok := checkItemsLv2[Lv1]; ok {
// //处理二级数据
// for _, Lv2 := range checkItemsLv2[Lv1].ToSlice() {
// newCustomItemData := map[string]string{}
// for uid, uname := range usernameMap {
// if answer, ok := checkResultMap[Lv1].Child[Lv2].CustomItemData[uid]; ok {
// newCustomItemData[uname] = answer
// } else {
// newCustomItemData[uname] = ""
// }
// }
// checkResultMap[Lv1].Child[Lv2].CustomItemData = newCustomItemData
// // checkResultMap[Lv1].Child[Lv2].Total = fmt.Sprintf("是:%d 否:%d 不清楚:%d",
// // checkResultMap[Lv1].Child[Lv2].AnswerYes, checkResultMap[Lv1].Child[Lv2].AnswerNo,
// // checkResultMap[Lv1].Child[Lv2].AnswerAny)
// rsp.CheckData = append(rsp.CheckData, *checkResultMap[Lv1].Child[Lv2])
// }
// }
// }
// return rsp, nil
// }
//getChanceCheckResultData 机会自查内容筛选详情数据输出
func getChanceCheckResultData(chanceId int64) (protocol.ReponseChanceCheckResult, error) {
var (
checkResultData []models.ChanceCheckResult
err error
)
rsp := protocol.ReponseChanceCheckResult{
CustomItem: []string{},
CheckData: []protocol.ChanceCheckResultData{},
}
checkResultData, err = models.GetChanceCheckResultByChancce(chanceId)
if err != nil {
log.Error("未获得指定的机会自查内容筛选结果,err;%s", err)
return rsp, nil
}
rsp, err = buildChanceCheckResultData(checkResultData)
return rsp, err
}
//buildChanceCheckResultData 构建机会自查内容筛选详情数据输出
func buildChanceCheckResultData(checkResultData []models.ChanceCheckResult) (protocol.ReponseChanceCheckResult, error) {
rsp := protocol.ReponseChanceCheckResult{
CustomItem: []string{},
CheckData: []protocol.ChanceCheckResultData{},
}
//人的id
userIds := utils.NewArraySetInt64()
checkResultMap := map[int64]*protocol.ChanceCheckResultData{}
... ... @@ -266,6 +110,7 @@ func getChanceCheckResultData(chanceId int64) (protocol.ReponseChanceCheckResult
checkReusltSlice[i].AnswerYes, checkReusltSlice[i].AnswerNo, checkReusltSlice[i].AnswerAny)
}
if checkReusltSlice[i].CheckPid > 0 {
//子集归并到父级
pid := checkReusltSlice[i].CheckPid
checkResultMap[pid].Child = append(checkResultMap[pid].Child, checkReusltSlice[i])
}
... ...
package utils
import "testing"
func TestArraySetInt64(t *testing.T) {
arr := NewArraySetInt64()
arr.Add(21)
arr.Add(11)
arr.Add(11)
arr.Add(13)
arr.Add(11)
arr.Add(21)
arr.Add(13)
arr.Add(14)
t.Log("结果", arr.ToSlice())
}
... ...