补全任务与测试

etl_billiards/loaders/facts/inventory_change.py

@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+"""库存变动事实表"""
+
+from ..base_loader import BaseLoader
+
+
+class InventoryChangeLoader(BaseLoader):
+    """写入 fact_inventory_change"""
+
+    def upsert_changes(self, records: list) -> tuple:
+        if not records:
+            return (0, 0, 0)
+
+        sql = """
+        INSERT INTO billiards.fact_inventory_change (
+            store_id,
+            change_id,
+            site_goods_id,
+            stock_type,
+            goods_name,
+            change_time,
+            start_qty,
+            end_qty,
+            change_qty,
+            unit,
+            price,
+            operator_name,
+            remark,
+            goods_category_id,
+            goods_second_category_id,
+            raw_data
+        )
+        VALUES (
+            %(store_id)s,
+            %(change_id)s,
+            %(site_goods_id)s,
+            %(stock_type)s,
+            %(goods_name)s,
+            %(change_time)s,
+            %(start_qty)s,
+            %(end_qty)s,
+            %(change_qty)s,
+            %(unit)s,
+            %(price)s,
+            %(operator_name)s,
+            %(remark)s,
+            %(goods_category_id)s,
+            %(goods_second_category_id)s,
+            %(raw_data)s
+        )
+        ON CONFLICT (store_id, change_id) DO UPDATE SET
+            site_goods_id            = EXCLUDED.site_goods_id,
+            stock_type               = EXCLUDED.stock_type,
+            goods_name               = EXCLUDED.goods_name,
+            change_time              = EXCLUDED.change_time,
+            start_qty                = EXCLUDED.start_qty,
+            end_qty                  = EXCLUDED.end_qty,
+            change_qty               = EXCLUDED.change_qty,
+            unit                     = EXCLUDED.unit,
+            price                    = EXCLUDED.price,
+            operator_name            = EXCLUDED.operator_name,
+            remark                   = EXCLUDED.remark,
+            goods_category_id        = EXCLUDED.goods_category_id,
+            goods_second_category_id = EXCLUDED.goods_second_category_id,
+            raw_data                 = EXCLUDED.raw_data,
+            updated_at               = now()
+        RETURNING (xmax = 0) AS inserted
+        """
+
+        inserted, updated = self.db.batch_upsert_with_returning(
+            sql, records, page_size=self._batch_size()
+        )
+        return (inserted, updated, 0)