迁移代码到Git

etl_billiards/loaders/dimensions/member.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+"""会员维度表加载器"""
+from ..base_loader import BaseLoader
+
+class MemberLoader(BaseLoader):
+    """会员维度加载器"""
+    
+    def upsert_members(self, records: list, store_id: int) -> tuple:
+        """加载会员数据"""
+        if not records:
+            return (0, 0, 0)
+        
+        sql = """
+        INSERT INTO billiards.dim_member (
+            store_id, member_id, member_name, phone, balance,
+            status, register_time, raw_data
+        )
+        VALUES (
+            %(store_id)s, %(member_id)s, %(member_name)s, %(phone)s, %(balance)s,
+            %(status)s, %(register_time)s, %(raw_data)s
+        )
+        ON CONFLICT (store_id, member_id) DO UPDATE SET
+            member_name = EXCLUDED.member_name,
+            phone = EXCLUDED.phone,
+            balance = EXCLUDED.balance,
+            status = EXCLUDED.status,
+            register_time = EXCLUDED.register_time,
+            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)