jx-callback/business/jxstore/initdata/initdata.go

package initdata

import (
	"fmt"

	"git.rosy.net.cn/baseapi/platformapi/autonavi"
	"git.rosy.net.cn/baseapi/utils"
	"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
	"git.rosy.net.cn/jx-callback/business/model"
	"git.rosy.net.cn/jx-callback/business/model/dao"
	"git.rosy.net.cn/jx-callback/globals/api"
)

func TruncateTable(db *dao.DaoDB, tableName string) (err error) {
	_, err = dao.ExecuteSQL(db, "TRUNCATE TABLE "+tableName)
	return err
}

func insertPlace(ctx *jxcontext.Context, db *dao.DaoDB, parent *autonavi.District, placeList []*autonavi.District) (err error) {
	for _, v := range placeList {
		if v.Level <= autonavi.DistrictLevelDistrict {
			place := &model.Place{
				Code:    int(utils.Str2Int64(v.Adcode)),
				Name:    v.Name,
				Level:   int8(v.Level),
				TelCode: v.CityCode,
				Enabled: 1,
			}
			if parent != nil {
				place.ParentCode = int(utils.Str2Int64(parent.Adcode))
			}
			dao.WrapAddIDCULEntity(place, ctx.GetUserName())
			if err = dao.CreateEntity(db, place); err != nil {
				return err
			}
			if err = insertPlace(ctx, db, v, v.Districts); err != nil {
				return err
			}
		}
	}
	return nil
}

func InitPlace(ctx *jxcontext.Context) (err error) {
	placeList, err2 := api.AutonaviAPI.GetDistricts(autonavi.DistrictLevelDistrict, "")
	if err = err2; err != nil {
		return err
	}
	placeList = placeList[0].Districts
	db := dao.GetDB()
	dao.Begin(db)
	defer func() {
		if r := recover(); r != nil || err != nil {
			dao.Rollback(db)
			if r != nil {
				panic(r)
			}
		}
	}()
	if _, err = dao.ExecuteSQL(db, `
	DELETE t1
	FROM place t1
	WHERE code < 9000000;
	`); err != nil {
		return err
	}
	if err = insertPlace(ctx, db, nil, placeList); err != nil {
		return err
	}
	dao.Commit(db)
	return err
}

func UploadImg4Vendors(ctx *jxcontext.Context, isAsync, isContinueWhenError bool) (hint string, err error) {
	// db := dao.GetDB()
	// rootTask := tasksch.NewSeqTask("合并SkuName图片至DataResource", ctx,
	// 	func(task *tasksch.SeqTask, step int, params ...interface{}) (result interface{}, err error) {
	// 		switch step {
	// 		case 0: // 计算SkuName中缺失的hashCode（注意是DescImg不是Img）
	// 			var skuNameList []*model.SkuName
	// 			if err = dao.GetRows(db, &skuNameList, `
	// 			SELECT t1.*
	// 			FROM sku_name t1
	// 			WHERE t1.desc_img <> '' AND t1.img_hash_code = ''
	// 			`); err == nil && len(skuNameList) > 0 {
	// 				calcTask := tasksch.NewParallelTask("UploadImg4Vendors calc hashCode",
	// 					tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError).SetParallelCount(5), ctx,
	// 					func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
	// 						skuName := batchItemList[0].(*model.SkuName)
	// 						_, skuName.ImgHashCode, err = jxutils.DownloadFileByURL(skuName.DescImg)
	// 						if err == nil {
	// 							dao.UpdateEntity(db, skuName, "ImgHashCode")
	// 						}
	// 						return retVal, err
	// 					}, skuNameList)
	// 				tasksch.HandleTask(calcTask, task, false).Run()
	// 				_, err = calcTask.GetResult(0)
	// 			}
	// 		case 1: // 从SkuName添加缺失的图片至DataResource
	// 			_, err = dao.ExecuteSQL(db, `
	// 			INSERT INTO data_resource(created_at, updated_at, last_operator, hash_code,
	// 				resource_type, name, main_url, ebai_url, qiniu_url, use_type)
	// 			SELECT t1.created_at, t1.created_at, t1.last_operator, t1.img_hash_code,
	// 			CASE
	// 				WHEN INSTR(t1.desc_img, ".jpg") > 0 OR INSTR(t1.desc_img, ".jpeg") > 0 THEN
	// 					'image/jpeg'
	// 				WHEN INSTR(t1.desc_img, ".png") > 0 OR INSTR(t1.desc_img, ".peg") > 0 THEN
	// 					'image/png'
	// 				WHEN INSTR(t1.desc_img, ".gif") THEN
	// 					'image/gif'
	// 				ELSE
	// 					''
	// 			END resource_type,
	// 			CONCAT(t1.name, '_desc'), desc_img main_url, t1.desc_img_ebai ebai_url,
	// 			IF(INSTR(t1.desc_img, "image.jxc4.com") > 0, t1.desc_img, '') qiniu_url, 2
	// 			FROM sku_name t1
	// 			JOIN (
	// 				SELECT img_hash_code, MAX(id) id, COUNT(*) ct
	// 				FROM sku_name
	// 				WHERE img_hash_code <> '' AND desc_img <> ''
	// 				GROUP BY 1
	// 			) t3 ON t3.id = t1.id
	// 			LEFT JOIN data_resource t2 ON (t2.main_url <> '' AND t2.main_url = t1.desc_img)
	// 			WHERE t1.desc_img <> '' AND t1.img_hash_code <> '' AND t2.id IS NULL;
	// 			`)
	// 		case 2: // 统一SkuName中同hashCode，不同图片地址至同一地址
	// 			_, err = dao.ExecuteSQL(db, `
	// 			UPDATE sku_name t1
	// 			JOIN data_resource t2 ON t2.hash_code = t1.img_hash_code AND t2.main_url <> ''
	// 			SET t1.desc_img = t2.main_url
	// 			WHERE t1.img_hash_code <> '' AND t1.desc_img <> t2.main_url;
	// 			`)
	// 		case 3: // 上传DataResource中缺失的平台图片
	// 			dataResList, err2 := dao.GetNeedUploadDataResource(db)
	// 			if err = err2; err == nil && len(dataResList) > 0 {
	// 				uploadTask := tasksch.NewParallelTask("批量上传图片至平台",
	// 					tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError).SetParallelCount(2), ctx,
	// 					func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
	// 						dataRes := batchItemList[0].(*model.DataResource)
	// 						_, err = datares.UploadImage2Vendors(ctx, task, dataRes, nil, false)
	// 						return nil, err
	// 					}, dataResList)
	// 				tasksch.HandleTask(uploadTask, task, true).Run()
	// 				_, err = uploadTask.GetResult(0)
	// 			}
	// 		}
	// 		return result, err
	// 	}, 4)
	// tasksch.HandleTask(rootTask, nil, true).Run()
	// if !isAsync {
	// 	if _, err = rootTask.GetResult(0); err == nil {
	// 		hint = "1"
	// 	}
	// } else {
	// 	hint = rootTask.ID
	// }
	return hint, err
}

func getSkuNameKey(prefix, name, comment, specUnit, unit string, specQuality float32) string {
	return fmt.Sprintf("%s-%s-%f-%s-%s", prefix, name, specQuality, specUnit, unit)
}