81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""台桌维度加载器"""
|
|
|
|
from ..base_loader import BaseLoader
|
|
|
|
|
|
class TableLoader(BaseLoader):
|
|
"""将台桌档案写入 dim_table"""
|
|
|
|
def upsert_tables(self, records: list) -> tuple:
|
|
"""批量写入台桌档案"""
|
|
if not records:
|
|
return (0, 0, 0)
|
|
|
|
sql = """
|
|
INSERT INTO billiards.dim_table (
|
|
store_id,
|
|
table_id,
|
|
site_id,
|
|
area_id,
|
|
area_name,
|
|
table_name,
|
|
table_price,
|
|
table_status,
|
|
table_status_name,
|
|
light_status,
|
|
is_rest_area,
|
|
show_status,
|
|
virtual_table,
|
|
charge_free,
|
|
only_allow_groupon,
|
|
is_online_reservation,
|
|
created_time,
|
|
raw_data
|
|
)
|
|
VALUES (
|
|
%(store_id)s,
|
|
%(table_id)s,
|
|
%(site_id)s,
|
|
%(area_id)s,
|
|
%(area_name)s,
|
|
%(table_name)s,
|
|
%(table_price)s,
|
|
%(table_status)s,
|
|
%(table_status_name)s,
|
|
%(light_status)s,
|
|
%(is_rest_area)s,
|
|
%(show_status)s,
|
|
%(virtual_table)s,
|
|
%(charge_free)s,
|
|
%(only_allow_groupon)s,
|
|
%(is_online_reservation)s,
|
|
%(created_time)s,
|
|
%(raw_data)s
|
|
)
|
|
ON CONFLICT (store_id, table_id) DO UPDATE SET
|
|
site_id = EXCLUDED.site_id,
|
|
area_id = EXCLUDED.area_id,
|
|
area_name = EXCLUDED.area_name,
|
|
table_name = EXCLUDED.table_name,
|
|
table_price = EXCLUDED.table_price,
|
|
table_status = EXCLUDED.table_status,
|
|
table_status_name = EXCLUDED.table_status_name,
|
|
light_status = EXCLUDED.light_status,
|
|
is_rest_area = EXCLUDED.is_rest_area,
|
|
show_status = EXCLUDED.show_status,
|
|
virtual_table = EXCLUDED.virtual_table,
|
|
charge_free = EXCLUDED.charge_free,
|
|
only_allow_groupon = EXCLUDED.only_allow_groupon,
|
|
is_online_reservation = EXCLUDED.is_online_reservation,
|
|
created_time = COALESCE(EXCLUDED.created_time, dim_table.created_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)
|