diff --git a/src/transformers/data_transformer.py b/src/transformers/data_transformer.py index 0c5f4bd..dfc6294 100644 --- a/src/transformers/data_transformer.py +++ b/src/transformers/data_transformer.py @@ -259,6 +259,7 @@ class DataTransformer: "event_timestamp": event_timestamp, "measurements": measurements, "created_at": mysql_row.get("created_at"), + "updated_at": mysql_row.get("updated_at"), } return pg_row @@ -477,6 +478,7 @@ class DataTransformer: "event_timestamp", "measurements", "created_at", + "updated_at", ] else: raise ValueError(f"Unknown table: {table}") diff --git a/src/transformers/schema_transformer.py b/src/transformers/schema_transformer.py index 768813c..c8046e1 100644 --- a/src/transformers/schema_transformer.py +++ b/src/transformers/schema_transformer.py @@ -89,7 +89,8 @@ CREATE TABLE IF NOT EXISTS elabdatadisp ( tool_name_id VARCHAR(32) NOT NULL, event_timestamp TIMESTAMP NOT NULL, measurements JSONB, - created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP + created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, + updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ) PARTITION BY RANGE (EXTRACT(YEAR FROM event_timestamp)); -- Note: PostgreSQL doesn't support PRIMARY KEY or UNIQUE constraints