- ExportShopsHealthInfo

business/jxstore/cms/store.go

@@ -8,10 +8,13 @@ import (
 	"strings"
 	"time"
 
+	"git.rosy.net.cn/jx-callback/business/partner/purchase/ebai"
+
 	"git.rosy.net.cn/baseapi/platformapi/dadaapi"
 	"git.rosy.net.cn/baseapi/platformapi/feieapi"
 	"git.rosy.net.cn/baseapi/utils"
 	"git.rosy.net.cn/jx-callback/business/jxutils"
+	"git.rosy.net.cn/jx-callback/business/jxutils/excel"
 	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
 	"git.rosy.net.cn/jx-callback/business/jxutils/netprinter"
 	"git.rosy.net.cn/jx-callback/business/jxutils/tasksch"
@@ -1039,3 +1042,85 @@ func updateOrCreateDadaStore(storeDetail *dao.StoreDetail2) (err error) {
 func composeDadaStoreName(storeDetail *dao.StoreDetail2) (storeName string) {
 	return storeDetail.Name + "-" + storeDetail.DadaStoreID
 }
+
+func ExportShopsHealthInfo(ctx *jxcontext.Context, vendorIDs, storeIDs []int, isAsync, isContinueWhenError bool) (hint string, err error) {
+	db := dao.GetDB()
+	vendorID := model.VendorIDEBAI
+	storeMapList, err := dao.GetStoresMapList(db, []int{vendorID}, storeIDs, model.StoreStatusAll)
+	if err != nil {
+		return "", err
+	}
+	storeMap2 := make(map[string]*model.StoreMap)
+	for _, v := range storeMapList {
+		storeMap2[v.VendorStoreID] = v
+	}
+	if len(storeMapList) > 0 {
+		var healthInfoList []interface{}
+		var excelBin []byte
+		var excelURL string
+		task := tasksch.NewSeqTask(fmt.Sprintf("ExportShopHealthInfo[%s]", model.VendorChineseNames[vendorID]), ctx,
+			func(task *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) {
+				switch step {
+				case 0:
+					subTask := tasksch.NewParallelTask(fmt.Sprintf("ExportShopHealthInfo2[%s]", model.VendorChineseNames[vendorID]), tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError), ctx,
+						func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
+							storeMap := batchItemList[0].(*model.StoreMap)
+							healthInfo, err := ebai.CurPurchaseHandler.GetShopHealthInfo(storeMap.VendorStoreID)
+							if err == nil {
+								retVal = []map[string]interface{}{healthInfo}
+							}
+							return retVal, err
+						}, storeMapList)
+					tasksch.AddChild(task, subTask).Run()
+					healthInfoList, err = subTask.GetResult(0)
+					if isContinueWhenError && err != nil && len(healthInfoList) > 0 {
+						err = nil
+					}
+				case 1:
+					var healthInfoList2 []map[string]interface{}
+					for _, v := range healthInfoList {
+						mapInfo := v.(map[string]interface{})
+						mapInfo["real_shop_id"] = storeMap2[utils.Interface2String(mapInfo["merchant_id"])].StoreID
+						healthInfoList2 = append(healthInfoList2, mapInfo)
+					}
+					excelConf := &excel.Obj2ExcelSheetConfig{
+						Title: "饿百门店情况导出",
+						Data:  healthInfoList2,
+						CaptionList: []string{
+							"real_shop_id",
+							"merchant_id",
+							"merchant_name",
+							"hours",
+							"sku_num",
+							"target_jiedan",
+							"is_healthy",
+						},
+					}
+					excelBin = excel.Obj2Excel([]*excel.Obj2ExcelSheetConfig{excelConf})
+				case 2:
+					keyPart := []string{
+						ctx.GetUserName(),
+						"饿百门店情况",
+					}
+					keyPart = append(keyPart, time.Now().Format("20060102T150405")+".xlsx")
+					key := "export/" + strings.Join(keyPart, "_")
+					excelURL, err = jxutils.UploadExportContent(excelBin, key)
+					if err == nil {
+						task.SetNoticeMsg(excelURL)
+					}
+					globals.SugarLogger.Debugf("导出饿百门店情况excelURL:%s, err:%v", excelURL, err)
+				}
+				return nil, err
+			}, 3)
+		tasksch.HandleTask(task, nil, true).Run()
+		if !isAsync {
+			_, err = task.GetResult(0)
+			if err == nil {
+				hint = excelURL
+			}
+		} else {
+			hint = task.GetID()
+		}
+	}
+	return hint, err
+}