补全任务与测试

etl_billiards/loaders/facts/table_discount.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+"""台费打折事实表"""
+
+from ..base_loader import BaseLoader
+
+
+class TableDiscountLoader(BaseLoader):
+    """写入 fact_table_discount"""
+
+    def upsert_discounts(self, records: list) -> tuple:
+        if not records:
+            return (0, 0, 0)
+
+        sql = """
+        INSERT INTO billiards.fact_table_discount (
+            store_id,
+            discount_id,
+            adjust_type,
+            applicant_id,
+            applicant_name,
+            operator_id,
+            operator_name,
+            ledger_amount,
+            ledger_count,
+            ledger_name,
+            ledger_status,
+            order_settle_id,
+            order_trade_no,
+            site_table_id,
+            table_area_id,
+            table_area_name,
+            create_time,
+            is_delete,
+            raw_data
+        )
+        VALUES (
+            %(store_id)s,
+            %(discount_id)s,
+            %(adjust_type)s,
+            %(applicant_id)s,
+            %(applicant_name)s,
+            %(operator_id)s,
+            %(operator_name)s,
+            %(ledger_amount)s,
+            %(ledger_count)s,
+            %(ledger_name)s,
+            %(ledger_status)s,
+            %(order_settle_id)s,
+            %(order_trade_no)s,
+            %(site_table_id)s,
+            %(table_area_id)s,
+            %(table_area_name)s,
+            %(create_time)s,
+            %(is_delete)s,
+            %(raw_data)s
+        )
+        ON CONFLICT (store_id, discount_id) DO UPDATE SET
+            adjust_type     = EXCLUDED.adjust_type,
+            applicant_id    = EXCLUDED.applicant_id,
+            applicant_name  = EXCLUDED.applicant_name,
+            operator_id     = EXCLUDED.operator_id,
+            operator_name   = EXCLUDED.operator_name,
+            ledger_amount   = EXCLUDED.ledger_amount,
+            ledger_count    = EXCLUDED.ledger_count,
+            ledger_name     = EXCLUDED.ledger_name,
+            ledger_status   = EXCLUDED.ledger_status,
+            order_settle_id = EXCLUDED.order_settle_id,
+            order_trade_no  = EXCLUDED.order_trade_no,
+            site_table_id   = EXCLUDED.site_table_id,
+            table_area_id   = EXCLUDED.table_area_id,
+            table_area_name = EXCLUDED.table_area_name,
+            create_time     = EXCLUDED.create_time,
+            is_delete       = EXCLUDED.is_delete,
+            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)