迁移代码到Git

etl_billiards/loaders/facts/order.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+"""订单事实表加载器"""
+from ..base_loader import BaseLoader
+
+class OrderLoader(BaseLoader):
+    """订单数据加载器"""
+    
+    def upsert_orders(self, records: list, store_id: int) -> tuple:
+        """加载订单数据"""
+        if not records:
+            return (0, 0, 0)
+        
+        sql = """
+        INSERT INTO billiards.fact_order (
+            store_id, order_id, order_no, member_id, table_id,
+            order_time, end_time, total_amount, discount_amount,
+            final_amount, pay_status, order_status, remark, raw_data
+        )
+        VALUES (
+            %(store_id)s, %(order_id)s, %(order_no)s, %(member_id)s, %(table_id)s,
+            %(order_time)s, %(end_time)s, %(total_amount)s, %(discount_amount)s,
+            %(final_amount)s, %(pay_status)s, %(order_status)s, %(remark)s, %(raw_data)s
+        )
+        ON CONFLICT (store_id, order_id) DO UPDATE SET
+            order_no = EXCLUDED.order_no,
+            member_id = EXCLUDED.member_id,
+            table_id = EXCLUDED.table_id,
+            order_time = EXCLUDED.order_time,
+            end_time = EXCLUDED.end_time,
+            total_amount = EXCLUDED.total_amount,
+            discount_amount = EXCLUDED.discount_amount,
+            final_amount = EXCLUDED.final_amount,
+            pay_status = EXCLUDED.pay_status,
+            order_status = EXCLUDED.order_status,
+            remark = EXCLUDED.remark,
+            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)

etl_billiards/loaders/facts/payment.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+"""支付记录加载器"""
+from ..base_loader import BaseLoader
+
+class PaymentLoader(BaseLoader):
+    """支付记录加载器"""
+    
+    def upsert_payments(self, records: list, store_id: int) -> tuple:
+        """加载支付记录"""
+        if not records:
+            return (0, 0, 0)
+        
+        sql = """
+        INSERT INTO billiards.fact_payment (
+            store_id, pay_id, order_id, pay_time, pay_amount,
+            pay_type, pay_status, remark, raw_data
+        )
+        VALUES (
+            %(store_id)s, %(pay_id)s, %(order_id)s, %(pay_time)s, %(pay_amount)s,
+            %(pay_type)s, %(pay_status)s, %(remark)s, %(raw_data)s
+        )
+        ON CONFLICT (store_id, pay_id) DO UPDATE SET
+            order_id = EXCLUDED.order_id,
+            pay_time = EXCLUDED.pay_time,
+            pay_amount = EXCLUDED.pay_amount,
+            pay_type = EXCLUDED.pay_type,
+            pay_status = EXCLUDED.pay_status,
+            remark = EXCLUDED.remark,
+            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)