381 lines
11 KiB
Go
381 lines
11 KiB
Go
package initdata
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"git.rosy.net.cn/baseapi/platformapi/autonavi"
|
|
"git.rosy.net.cn/baseapi/utils"
|
|
"git.rosy.net.cn/jx-callback/business/jxstore/cms"
|
|
"git.rosy.net.cn/jx-callback/business/jxutils"
|
|
"git.rosy.net.cn/jx-callback/business/jxutils/jxcontext"
|
|
"git.rosy.net.cn/jx-callback/business/jxutils/tasksch"
|
|
"git.rosy.net.cn/jx-callback/business/model"
|
|
"git.rosy.net.cn/jx-callback/business/model/dao"
|
|
"git.rosy.net.cn/jx-callback/business/partner"
|
|
"git.rosy.net.cn/jx-callback/business/partner/purchase/mtwm"
|
|
"git.rosy.net.cn/jx-callback/globals"
|
|
"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) {
|
|
db := dao.GetDB()
|
|
if err = TruncateTable(db, "place"); err == nil {
|
|
placeList, err2 := api.AutonaviAPI.GetDistricts(autonavi.DistrictLevelDistrict, "")
|
|
if err = err2; err != nil {
|
|
return err
|
|
}
|
|
placeList = placeList[0].Districts
|
|
dao.Begin(db)
|
|
defer func() {
|
|
dao.Rollback(db)
|
|
}()
|
|
if err = insertPlace(ctx, db, nil, placeList); err != nil {
|
|
return err
|
|
}
|
|
updateSqls := []string{
|
|
`
|
|
UPDATE place t1
|
|
JOIN jde_city t2 ON t1.code = t2.col_tencentAddressCode
|
|
SET t1.jd_code = t2.col_areaCode;
|
|
`,
|
|
`
|
|
UPDATE place t1
|
|
JOIN place t2 ON t1.parent_code = t2.code AND t2.jd_code != 0
|
|
JOIN jde_district t3 ON t1.name = t3.col_areaName AND t2.jd_code = t3.col_cityCode
|
|
SET t1.jd_code = t3.col_areaCode
|
|
WHERE t1.level = 3;
|
|
`,
|
|
`
|
|
UPDATE
|
|
place t1
|
|
JOIN ebde_places t2 ON t1.name = t2.col_city_name
|
|
SET t1.ebai_code = t2.col_city_id
|
|
WHERE t1.level = 1 OR t1.level = 2;
|
|
`,
|
|
`
|
|
UPDATE
|
|
place t1
|
|
JOIN place t1p ON t1.parent_code = t1p.code
|
|
JOIN ebde_places t2 ON t1.name = t2.col_city_name
|
|
JOIN ebde_places t2p ON t2.col_parent_id = t2p.col_city_id AND t1p.ebai_code = t2p.col_city_id
|
|
SET t1.ebai_code = t2.col_city_id
|
|
WHERE t1.level = 3;
|
|
`,
|
|
`
|
|
UPDATE
|
|
place t1
|
|
JOIN mtpsdeliveryprice t2 ON t1.code = t2.citycode
|
|
SET t1.mtps_price = t2.price;
|
|
`,
|
|
`
|
|
UPDATE
|
|
place t1
|
|
JOIN mtpsdeliveryprice t2 ON t1.name LIKE CONCAT(t2.cityname, '%')
|
|
SET t1.mtps_price = t2.price
|
|
WHERE t1.level = 2 AND t1.mtps_price = 0;
|
|
`,
|
|
`
|
|
UPDATE place t1
|
|
LEFT JOIN (
|
|
SELECT DISTINCT city_code
|
|
FROM store
|
|
UNION DISTINCT
|
|
SELECT DISTINCT place_code city_code
|
|
FROM sku_name_place_bind
|
|
) t2 ON t1.code = t2.city_code
|
|
SET t1.enabled = 0
|
|
WHERE t1.level = 2 AND t2.city_code IS NULL;
|
|
`,
|
|
}
|
|
for _, v := range updateSqls {
|
|
if _, err = dao.ExecuteSQL(db, v); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
dao.Commit(db)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func InitSkuName(ctx *jxcontext.Context, isForce, isAsync, isContinueWhenError bool) (hint string, err error) {
|
|
db := dao.GetDB()
|
|
var skuNameList []*model.SkuName
|
|
if err = dao.GetRows(db, &skuNameList, `
|
|
SELECT *
|
|
FROM sku_name
|
|
WHERE deleted_at = ? AND img <> ''
|
|
ORDER BY id
|
|
`, utils.DefaultTimeValue); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
task := tasksch.NewParallelTask("InitSkuName calculate md5", tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError), ctx.GetUserName(), func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
|
|
skuName := batchItemList[0].(*model.SkuName)
|
|
if skuName.ImgHashCode == "" || isForce {
|
|
response, err := http.Get(skuName.Img)
|
|
if err == nil {
|
|
defer response.Body.Close()
|
|
data, err2 := ioutil.ReadAll(response.Body)
|
|
if err = err2; err == nil {
|
|
skuName.ImgHashCode = fmt.Sprintf("%X", md5.Sum(data))
|
|
db := dao.GetDB()
|
|
_, err = dao.UpdateEntity(db, skuName, "ImgHashCode")
|
|
}
|
|
}
|
|
}
|
|
return nil, err
|
|
}, skuNameList)
|
|
tasksch.ManageTask(task).Run()
|
|
if !isAsync {
|
|
_, err = task.GetResult(0)
|
|
} else {
|
|
hint = task.ID
|
|
}
|
|
return hint, err
|
|
}
|
|
|
|
func InitVendorCategory(ctx *jxcontext.Context) (num int64, err error) {
|
|
if handler, ok := partner.PurchasePlatformHandlers[model.VendorIDMTWM].(*mtwm.PurchaseHandler); ok {
|
|
cats, err2 := handler.GetVendorCategories()
|
|
if err2 != nil {
|
|
return num, err2
|
|
}
|
|
|
|
db := dao.GetDB()
|
|
dao.Begin(db)
|
|
defer dao.Rollback(db)
|
|
sql := `
|
|
DELETE
|
|
FROM sku_vendor_category
|
|
WHERE vendor_id = ?
|
|
`
|
|
if _, err = dao.ExecuteSQL(db, sql, model.VendorIDMTWM); err != nil {
|
|
return num, err
|
|
}
|
|
for _, cat := range cats {
|
|
dao.WrapAddIDCULEntity(cat, ctx.GetUserName())
|
|
if err = dao.CreateEntity(db, cat); err != nil {
|
|
return num, err
|
|
}
|
|
}
|
|
dao.Commit(db)
|
|
num = int64(len(cats))
|
|
}
|
|
return num, err
|
|
}
|
|
|
|
func Convert2JDSPU(ctx *jxcontext.Context, count int, isAsync, isContinueWhenError bool) (hint string, err error) {
|
|
sql := `
|
|
SELECT t1.*
|
|
FROM sku_name t1
|
|
LEFT JOIN sku_name t2 ON t2.link_id = t1.id
|
|
WHERE t1.deleted_at = ? AND t1.status <> ? AND t1.is_spu = 0 AND t1.unit = '份'
|
|
AND t2.id IS NULL
|
|
ORDER BY t1.id
|
|
`
|
|
if count > 0 {
|
|
sql += " LIMIT " + utils.Int2Str(count)
|
|
}
|
|
sqlParams := []interface{}{
|
|
utils.DefaultTimeValue,
|
|
model.SkuStatusDeleted,
|
|
}
|
|
|
|
db := dao.GetDB()
|
|
var skuNameList []*model.SkuName
|
|
if err = dao.GetRows(db, &skuNameList, sql, sqlParams...); err != nil {
|
|
return "", err
|
|
}
|
|
for _, skuName := range skuNameList {
|
|
sql = `
|
|
SELECT *
|
|
FROM sku
|
|
WHERE name_id = ? AND deleted_at = ? AND status = ?;
|
|
`
|
|
sqlParams := []interface{}{
|
|
skuName.ID,
|
|
utils.DefaultTimeValue,
|
|
model.SkuStatusNormal,
|
|
}
|
|
var skuList []*model.Sku
|
|
if err = dao.GetRows(db, &skuList, sql, sqlParams...); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
sql = `
|
|
SELECT t1.*
|
|
FROM sku_name_place_bind t1
|
|
WHERE t1.name_id = ?
|
|
`
|
|
sqlParams = []interface{}{
|
|
skuName.ID,
|
|
}
|
|
var skuNamePlaceBindList []*model.SkuNamePlaceBind
|
|
if err = dao.GetRows(db, &skuNamePlaceBindList, sql, sqlParams...); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(skuList) > 0 {
|
|
globals.SugarLogger.Debugf("Convert2JDSPU, name:%s", skuName.Name)
|
|
dao.Begin(db)
|
|
skuNameNew2 := *skuName
|
|
skuNameNew := &skuNameNew2
|
|
dao.WrapAddIDCULEntity(skuNameNew, ctx.GetUserName())
|
|
skuNameNew.JdID = 0
|
|
skuNameNew.LinkID = skuName.ID
|
|
skuNameNew.IsSpu = 1
|
|
skuNameNew.JdSyncStatus = model.SyncFlagNewMask
|
|
// skuNameNew.Status = model.SkuStatusDontSale
|
|
if err = dao.CreateEntity(db, skuNameNew); err != nil {
|
|
dao.Rollback(db)
|
|
return "", err
|
|
}
|
|
for _, sku := range skuList {
|
|
skuNew2 := *sku
|
|
skuNew := &skuNew2
|
|
dao.WrapAddIDCULEntity(skuNew, ctx.GetUserName())
|
|
skuNew.JdID = jxutils.GenFakeID()
|
|
skuNew.LinkID = sku.ID
|
|
skuNew.NameID = skuNameNew.ID
|
|
skuNew.JdSyncStatus = model.SyncFlagNewMask
|
|
if skuNameNew.Status == model.SkuStatusDontSale {
|
|
skuNew.Status = model.SkuStatusDontSale
|
|
}
|
|
globals.SugarLogger.Debugf("Convert2JDSPU, sku:%s", utils.Format4Output(skuNew, false))
|
|
if err = dao.CreateEntity(db, skuNew); err != nil {
|
|
dao.Rollback(db)
|
|
return "", err
|
|
}
|
|
}
|
|
for _, placeBind := range skuNamePlaceBindList {
|
|
dao.WrapAddIDCULEntity(placeBind, ctx.GetUserName())
|
|
placeBind.NameID = skuNameNew.ID
|
|
globals.SugarLogger.Debugf("Convert2JDSPU, placeBind:%s", utils.Format4Output(placeBind, false))
|
|
if err = dao.CreateEntity(db, placeBind); err != nil {
|
|
dao.Rollback(db)
|
|
return "", err
|
|
}
|
|
}
|
|
dao.Commit(db)
|
|
}
|
|
}
|
|
sql = `
|
|
SELECT t1.*
|
|
FROM sku_name t1
|
|
WHERE t1.link_id <> 0;
|
|
`
|
|
skuNameList = []*model.SkuName{}
|
|
if err = dao.GetRows(db, &skuNameList, sql); err != nil {
|
|
return "", err
|
|
}
|
|
rootTask := tasksch.NewParallelTask("Convert2JDSPU", tasksch.NewParallelConfig().SetIsContinueWhenError(isContinueWhenError), ctx.GetUserName(), func(task *tasksch.ParallelTask, batchItemList []interface{}, params ...interface{}) (retVal interface{}, err error) {
|
|
skuName := batchItemList[0].(*model.SkuName)
|
|
_, err = cms.CurVendorSync.SyncSku(ctx, db, skuName.ID, -1, false, ctx.GetUserName())
|
|
return nil, err
|
|
}, skuNameList)
|
|
tasksch.ManageTask(rootTask).Run()
|
|
|
|
if !isAsync {
|
|
_, err = rootTask.GetResult(0)
|
|
} else {
|
|
hint = rootTask.ID
|
|
}
|
|
return hint, err
|
|
}
|
|
|
|
func Change2JDSPU4Store(ctx *jxcontext.Context, storeIDs []int, isAsync, isContinueWhenError bool) (hint string, err error) {
|
|
sql := `
|
|
DELETE t1
|
|
FROM store_sku_bind t1
|
|
JOIN sku t2 ON t2.id = t1.sku_id AND t2.link_id <> 0
|
|
WHERE 1 = 1
|
|
`
|
|
sqlParams := []interface{}{}
|
|
if len(storeIDs) > 0 {
|
|
sql += " AND store_id IN (" + dao.GenQuestionMarks(len(storeIDs)) + ")"
|
|
sqlParams = append(sqlParams, storeIDs)
|
|
}
|
|
db := dao.GetDB()
|
|
if _, err = dao.ExecuteSQL(db, sql, sqlParams...); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
dao.Begin(db)
|
|
sql = `
|
|
INSERT INTO store_sku_bind(created_at, updated_at, last_operator, deleted_at, store_id, sku_id, price, unit_price, status, ebai_id, mtwm_id, jd_sync_status, ebai_sync_status, mtwm_sync_status)
|
|
SELECT NOW(), NOW(), ?, ?, t1.store_id, t2.id, t1.price, t1.unit_price, t1.status , 0, 0, ?, ?, ?
|
|
FROM store_sku_bind t1
|
|
JOIN sku t2 ON t2.link_id = t1.sku_id AND t2.deleted_at = ?
|
|
WHERE t1.deleted_at = ?
|
|
`
|
|
sqlParams = []interface{}{
|
|
ctx.GetUserName(),
|
|
utils.DefaultTimeValue,
|
|
// model.SkuStatusDontSale,
|
|
model.SyncFlagNewMask,
|
|
0, //model.SyncFlagNewMask,
|
|
0, //model.SyncFlagNewMask,
|
|
utils.DefaultTimeValue,
|
|
utils.DefaultTimeValue,
|
|
}
|
|
if len(storeIDs) > 0 {
|
|
sql += " AND t1.store_id IN (" + dao.GenQuestionMarks(len(storeIDs)) + ")"
|
|
sqlParams = append(sqlParams, storeIDs)
|
|
}
|
|
globals.SugarLogger.Debug(sql)
|
|
globals.SugarLogger.Debug(utils.Format4Output(sqlParams, false))
|
|
var num int64
|
|
if num, err = dao.ExecuteSQL(db, sql, sqlParams...); err != nil {
|
|
dao.Rollback(db)
|
|
return "", err
|
|
}
|
|
globals.SugarLogger.Debug(num)
|
|
var skuIDs []int
|
|
sql = `
|
|
SELECT id
|
|
FROM sku
|
|
WHERE link_id <> 0 AND deleted_at = ?
|
|
`
|
|
sqlParams = []interface{}{
|
|
utils.DefaultTimeValue,
|
|
}
|
|
if err = dao.GetRows(db, &skuIDs, sql, sqlParams...); err != nil {
|
|
return "", err
|
|
}
|
|
hint, err = cms.CurVendorSync.SyncStoresSkus(ctx, db, []int{model.VendorIDJD}, storeIDs, skuIDs, isAsync, isContinueWhenError)
|
|
return hint, err
|
|
}
|