feat: 实现商品目录批量导入接口 (#133)

2026-08-11 10:06:10 +08:00
parent 585d9d7834
commit 273bcc14a6
8 changed files with 695 additions and 0 deletions
@@ -79,6 +79,121 @@ func RecordCatalogImportConflict(q Execer, source, batchID, requestedAt string)
 	return catalogRunUpdateResult(result, err, "记录商品目录批次冲突")
 }

+// CompleteCatalogImportRun 把处理结果摘要和安全的响应 JSON 固化，供幂等重放。
+func CompleteCatalogImportRun(q Execer, run model.CatalogImportRun) error {
+	result, err := q.Exec(`UPDATE catalog_import_runs SET status=?,shopee_created=?,shopee_updated=?,
+		sku_created=?,sku_updated=?,pdd_created=?,pdd_updated=?,association_created=?,
+		association_unchanged=?,failure_count=?,error_summary=NULLIF(?,''),response_body=NULLIF(?,''),
+		finished_at=? WHERE source=? AND batch_id=?`, run.Status, run.ShopeeCreated, run.ShopeeUpdated,
+		run.SKUCreated, run.SKUUpdated, run.PddCreated, run.PddUpdated, run.AssociationCreated,
+		run.AssociationUnchanged, run.FailureCount, run.ErrorSummary, run.ResponseBody,
+		run.FinishedAt, run.Source, run.BatchID)
+	return catalogRunUpdateResult(result, err, "完成商品目录批次")
+}
+
+// UpsertCatalogShopeeProduct 按上游观测时间更新来源字段，绝不覆盖人工 PDD 关联。
+func UpsertCatalogShopeeProduct(q Execer, goodsID, title, status, mainSKU, observedAt, now string) (created, updated bool, err error) {
+	var oldObserved sql.NullString
+	err = q.QueryRow(`SELECT source_observed_at FROM shopee_products WHERE goods_id=?`, goodsID).Scan(&oldObserved)
+	if errors.Is(err, sql.ErrNoRows) {
+		_, err = q.Exec(`INSERT INTO shopee_products(goods_id,title,shopee_status,main_sku_code,source,
+			source_observed_at,created_at,updated_at) VALUES(?,?,NULLIF(?,''),NULLIF(?,''),'api',?,?,?)`,
+			goodsID, title, status, mainSKU, observedAt, now, now)
+		return err == nil, false, err
+	}
+	if err != nil {
+		return false, false, err
+	}
+	if oldObserved.Valid && oldObserved.String > observedAt {
+		return false, false, nil
+	}
+	_, err = q.Exec(`UPDATE shopee_products SET title=?,shopee_status=NULLIF(?,''),main_sku_code=NULLIF(?,''),
+		source='api',source_observed_at=?,updated_at=? WHERE goods_id=?`, title, status, mainSKU, observedAt, now, goodsID)
+	return false, err == nil, err
+}
+
+// UpsertCatalogShopeeSKU 拒绝把同一 SKU 静默挪到另一商品，并保留 is_manual。
+func UpsertCatalogShopeeSKU(q Execer, skuID, goodsID, specRaw, color, size, advice string, parseOK bool, skuCode, observedAt, now string) (created, updated bool, err error) {
+	var oldGoods string
+	var oldObserved sql.NullString
+	err = q.QueryRow(`SELECT goods_id,source_observed_at FROM shopee_skus WHERE sku_id=?`, skuID).Scan(&oldGoods, &oldObserved)
+	parse := 0
+	if parseOK {
+		parse = 1
+	}
+	if errors.Is(err, sql.ErrNoRows) {
+		_, err = q.Exec(`INSERT INTO shopee_skus(sku_id,goods_id,spec_raw,color,size,advice,parse_ok,sku_code,
+			is_manual,source_observed_at,created_at,updated_at) VALUES(?,?,?,NULLIF(?,''),NULLIF(?,''),NULLIF(?,''),?,NULLIF(?,''),0,?,?,?)`,
+			skuID, goodsID, specRaw, color, size, advice, parse, skuCode, observedAt, now, now)
+		return err == nil, false, err
+	}
+	if err != nil {
+		return false, false, err
+	}
+	if oldGoods != goodsID {
+		return false, false, fmt.Errorf("SKU %s 已属于蝦皮商品 %s", skuID, oldGoods)
+	}
+	if oldObserved.Valid && oldObserved.String > observedAt {
+		return false, false, nil
+	}
+	_, err = q.Exec(`UPDATE shopee_skus SET spec_raw=?,color=NULLIF(?,''),size=NULLIF(?,''),advice=NULLIF(?,''),
+		parse_ok=?,sku_code=NULLIF(?,''),source_observed_at=?,updated_at=? WHERE sku_id=?`,
+		specRaw, color, size, advice, parse, skuCode, observedAt, now, skuID)
+	return false, err == nil, err
+}
+
+// UpsertCatalogPddProduct 写入结构化 PDD 数据转换后的规范 JSON；空规格不清除既有采集结果。
+func UpsertCatalogPddProduct(q Execer, goodsID, url, title, shopName, skusJSON, observedAt, now string) (created, updated bool, err error) {
+	var oldObserved, deletedAt sql.NullString
+	err = q.QueryRow(`SELECT source_observed_at,deleted_at FROM pdd_products WHERE goods_id=?`, goodsID).Scan(&oldObserved, &deletedAt)
+	status := "pending"
+	if skusJSON != "" {
+		status = "collected"
+	}
+	if errors.Is(err, sql.ErrNoRows) {
+		_, err = q.Exec(`INSERT INTO pdd_products(goods_id,url,title,shop_name,skus_json,collect_status,collected_at,
+			source,source_observed_at,created_at,updated_at) VALUES(?,?,NULLIF(?,''),NULLIF(?,''),NULLIF(?,''),?,
+			CASE WHEN ?='collected' THEN ? ELSE NULL END,'api',?,?,?)`, goodsID, url, title, shopName, skusJSON, status, status, now, observedAt, now, now)
+		return err == nil, false, err
+	}
+	if err != nil {
+		return false, false, err
+	}
+	if oldObserved.Valid && oldObserved.String > observedAt {
+		return false, false, nil
+	}
+	if skusJSON == "" {
+		_, err = q.Exec(`UPDATE pdd_products SET url=?,title=CASE WHEN TRIM(?)='' THEN title ELSE ? END,
+			shop_name=CASE WHEN TRIM(?)='' THEN shop_name ELSE ? END,deleted_at=NULL,source='api',source_observed_at=?,updated_at=? WHERE goods_id=?`,
+			url, title, title, shopName, shopName, observedAt, now, goodsID)
+	} else {
+		_, err = q.Exec(`UPDATE pdd_products SET url=?,title=NULLIF(?,''),shop_name=NULLIF(?,''),skus_json=?,
+			collect_status='collected',collect_msg=NULL,artifact_ref=NULL,collected_at=?,deleted_at=NULL,
+			source='api',source_observed_at=?,updated_at=? WHERE goods_id=?`, url, title, shopName, skusJSON, now, observedAt, now, goodsID)
+	}
+	return false, err == nil, err
+}
+
+// ApplyCatalogAssociation 只允许空关联建立或相同关联重放。
+func ApplyCatalogAssociation(q Execer, shopeeGoodsID, pddGoodsID, now string) (created, unchanged bool, err error) {
+	var current sql.NullString
+	if err = q.QueryRow(`SELECT pdd_goods_id FROM shopee_products WHERE goods_id=?`, shopeeGoodsID).Scan(&current); err != nil {
+		return
+	}
+	var url string
+	if err = q.QueryRow(`SELECT url FROM pdd_products WHERE goods_id=? AND deleted_at IS NULL`, pddGoodsID).Scan(&url); err != nil {
+		return
+	}
+	if current.Valid && current.String != "" {
+		if current.String == pddGoodsID {
+			return false, true, nil
+		}
+		return false, false, fmt.Errorf("蝦皮商品 %s 已关联 PDD 商品 %s", shopeeGoodsID, current.String)
+	}
+	_, err = q.Exec(`UPDATE shopee_products SET pdd_goods_id=?,pdd_goods_url=?,updated_at=? WHERE goods_id=?`, pddGoodsID, url, now, shopeeGoodsID)
+	return err == nil, false, err
+}
+
 func catalogRunUpdateResult(result sql.Result, err error, action string) error {
 	if err != nil {
 		return fmt.Errorf("%s失败: %w", action, err)