legion-data 1.6.9 → 1.6.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 79d855773e661254d6a8036ff1dc06a4310214503d331813b55008d57d3fc23a
-  data.tar.gz: 6209e686da5f791d968ed734cb2ffc53b92db78cc2d6154afa96c4fb294c6a87
+  metadata.gz: c0e428fc5d4a92e8fe8b3c24f2277f5842e13b0c9814f712c82cc7cc41a43cfc
+  data.tar.gz: b9d1dd28db840e02b4308d2b29b20a1ff89b7932c07952f243b64cad6b45d5c1
 SHA512:
-  metadata.gz: dc3d56af6edd0b31cb3b8b673af3f1dc582723869aca751957ee18d5efee413344284c6e0fa0c27ac58a638370193dc018f79b02d29cd6ef2356910d582f8798
-  data.tar.gz: 630e4c8f408dcd503ed7d1e9f49ada6df92f9060cafd7a581b7b21d2c3fad1d626713f38d532d77d92cf90d8236b6caa7aa6cb8c1fc72c61ab50be75d95bca35
+  metadata.gz: df6941f0d373dbb1d18aadd1b75ee092ec58d8da8fb82b273e6b847f8cf420e999212d0323171bfdf965fe4b105e0e4f13bbe62d3f29ac77f63c35477a4f78c7
+  data.tar.gz: d64f91ecafd301f33153e5bc77988014f11ceb2a3cc0eacf0201bcb19250adc7f71888004643d45ab967cf230ca82fb1986b166b389f1b08a4c098890fed98c1

data/CHANGELOG.md

@@ -1,5 +1,25 @@
 # Legion::Data Changelog
 
+## [1.6.11] - 2026-03-28
+
+### Added
+- Migration 050: critical missing indexes across 13 tables (runners, tasks, digital_workers, audit_log, webhook_deliveries, webhook_dead_letters, conversations, approval_queue, rbac_role_assignments, rbac_cross_team_grants, memory_traces, agent_cluster_tasks, finlog_executions)
+- Migration 051: fix tasks archival column mismatch — adds `created_at` column (PG: generated from `created`, SQLite/MySQL: backfilled)
+- Migration 052: drop redundant Apollo indexes (PG only) — auto-named duplicates from migration 012 superseded by explicit indexes in migration 047
+- Migration 053: FK constraint for `tasks.relationship_id` (PG only) with orphan cleanup and ON DELETE SET NULL
+- Migration 054: add `component_type` column to functions table (v3.0 naming convention — runner/hook/absorber)
+- Migration 055: add nullable `definition` text column to functions table (v3.0 method contract storage)
+- Migration 056: add `absorber_patterns` table for pattern-matched content acquisition (v3.0)
+- Migration 057: add `routing_key` column to runners table (v3.0 AMQP routing key storage)
+
+### Fixed
+- `Archival::Policy` now includes `DATE_COLUMN_OVERRIDES` map for legacy tables using non-standard date columns
+
+## [1.6.10] - 2026-03-28
+
+### Added
+- Migration 049: adds `remote_invocable` boolean column (default: true) to the `functions` table. Allows per-function control over whether a registered function can be dispatched remotely via AMQP from the `LexDispatch` API layer.
+
 ## [1.6.9] - 2026-03-27
 
 ### Added

data/CLAUDE.md

@@ -8,7 +8,7 @@
 Manages persistent database storage for the LegionIO framework. Supports SQLite (default), MySQL, and PostgreSQL via Sequel ORM. Provides automatic schema migrations and data models for extensions, functions, runners, nodes, tasks, settings, digital workers, task relationships, Apollo shared knowledge tables (PostgreSQL only), tenants, webhooks, audit log, and archive tables. Also provides a parallel local SQLite database (`Legion::Data::Local`) for agentic cognitive state persistence.
 
 **GitHub**: https://github.com/LegionIO/legion-data
-**Version**: 1.6.6
+**Version**: 1.6.11
 **License**: Apache-2.0
 
 ## Supported Databases
@@ -56,7 +56,7 @@ Legion::Data (singleton module)
 │   ├── .shutdown      # Close local connection
 │   └── .reset!        # Clear all state (testing)
 │
-├── Migration          # Auto-migration system (47 migrations, Sequel DSL)
+├── Migration          # Auto-migration system (57 migrations, Sequel DSL)
 │   └── migrations/
 │       ├── 001_add_schema_columns
 │       ├── 002_add_nodes
@@ -104,7 +104,17 @@ Legion::Data (singleton module)
 │       ├── 044_expand_memory_traces
 │       ├── 045_add_memory_associations
 │       ├── 046_add_metering_hourly_rollup
-│       └── 047_apollo_knowledge_capture     # identity cols, ops table, archive table, 25+ indexes
+│       ├── 047_apollo_knowledge_capture     # identity cols, ops table, archive table, 25+ indexes
+│       ├── 048_add_financial_logging        # 7 UAIS cost recovery tables (identity, asset, environment, accounting, execution, tags, usage)
+│       ├── 049_add_remote_invocable_to_functions  # remote_invocable boolean on functions (v3.0)
+│       ├── 050_add_missing_indexes         # critical indexes across 13 tables
+│       ├── 051_fix_tasks_created_at        # created_at alias for archival (PG generated, SQLite backfill)
+│       ├── 052_drop_redundant_apollo_indexes   # PG only: remove duplicate auto-named indexes
+│       ├── 053_add_tasks_relationship_fk   # PG only: FK constraint on tasks.relationship_id
+│       ├── 054_add_component_type_to_functions  # component_type on functions (runner/hook/absorber, v3.0)
+│       ├── 055_add_definition_to_functions      # definition text column on functions (v3.0)
+│       ├── 056_add_absorber_patterns       # absorber_patterns table for pattern-matched acquisition
+│       └── 057_add_routing_key_to_runners  # routing_key on runners (v3.0 AMQP)
 │
 ├── Model              # Sequel model loader
 │   └── Models/
@@ -269,7 +279,7 @@ Per-adapter credential defaults are defined in `Settings::CREDS`:
 | `lib/legion/data.rb` | Module entry, setup/shutdown lifecycle |
 | `lib/legion/data/connection.rb` | Sequel database connection (adapter selection) |
 | `lib/legion/data/migration.rb` | Migration runner |
-| `lib/legion/data/migrations/` | 47 numbered migration files (Sequel DSL) |
+| `lib/legion/data/migrations/` | 48 numbered migration files (Sequel DSL) |
 | `lib/legion/data/model.rb` | Model autoloader |
 | `lib/legion/data/local.rb` | Local SQLite module for agentic cognitive state |
 | `lib/legion/data/models/` | Sequel models (Extension, Function, Runner, Node, Task, TaskLog, Setting, DigitalWorker, Relationship, ApolloEntry, ApolloRelation, ApolloExpertise, ApolloAccessLog, AuditLog, RbacRoleAssignment, RbacRunnerGrant, RbacCrossTeamGrant) |
@@ -309,6 +319,7 @@ Optional persistent storage initialized during `Legion::Service` startup (after
 12. Webhook subscription storage (migration 020)
 13. Archive, memory traces, and tenant partition tables (migrations 021–025)
 14. Function embeddings for semantic runner discovery (migration 026 — description + vector columns on functions table)
+15. Financial logging for UAIS cost recovery (migration 048 — 7 tables: identity, asset, environment, accounting, execution, tags, usage rollup)
 
 ---
 

data/lib/legion/data/archival/policy.rb

@@ -11,6 +11,14 @@ module Legion
           tables:          %w[tasks metering_records].freeze
         }.freeze
 
+        # Per-table date column overrides. The Retention module defaults to :created_at,
+        # but legacy tables (like tasks) use :created. Migration 051 may add a created_at
+        # column/alias for tasks (implementation varies by adapter); this map forces use of
+        # :created so behavior is consistent across legacy schemas and adapters.
+        DATE_COLUMN_OVERRIDES = {
+          'tasks' => :created
+        }.freeze
+
         attr_reader :warm_after_days, :cold_after_days, :batch_size, :tables
 
         def initialize(**opts)

data/lib/legion/data/migrations/049_add_remote_invocable_to_functions.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next unless table_exists?(:functions)
+    next if schema(:functions).any? { |c, _| c == :remote_invocable }
+
+    alter_table(:functions) do
+      add_column :remote_invocable, TrueClass, default: true, null: false
+    end
+  end
+
+  down do
+    next unless table_exists?(:functions)
+    next unless schema(:functions).any? { |c, _| c == :remote_invocable }
+
+    alter_table(:functions) do
+      drop_column :remote_invocable
+    end
+  end
+end

data/lib/legion/data/migrations/050_add_missing_indexes.rb

@@ -0,0 +1,176 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    # runners: FK without index, hot-path lookups, duplicate prevention
+    if table_exists?(:runners)
+      # Remove any duplicate (extension_id, name) rows before adding the unique index.
+      # Keep the active and most recently updated row per pair; use id DESC as tie-breaker.
+      run <<~SQL
+        WITH ranked AS (
+          SELECT
+            id,
+            ROW_NUMBER() OVER (
+              PARTITION BY extension_id, name
+              ORDER BY
+                active DESC,
+                updated DESC,
+                id DESC
+            ) AS rn
+          FROM runners
+        )
+        DELETE FROM runners
+        WHERE id IN (
+          SELECT id
+          FROM ranked
+          WHERE rn > 1
+        )
+      SQL
+
+      alter_table(:runners) do
+        add_index :extension_id, name: :idx_runners_extension_id, if_not_exists: true
+        add_index :namespace, name: :idx_runners_namespace, if_not_exists: true
+        add_index :name, name: :idx_runners_name, if_not_exists: true
+        add_index %i[extension_id name], name: :idx_runners_extension_name, unique: true, if_not_exists: true
+      end
+    end
+
+    # tasks: plain Integer relationship_id used by ORM association
+    if table_exists?(:tasks)
+      alter_table(:tasks) do
+        add_index :relationship_id, name: :idx_tasks_relationship_id, if_not_exists: true
+      end
+    end
+
+    # digital_workers: consent/trust-based queries
+    if table_exists?(:digital_workers)
+      alter_table(:digital_workers) do
+        add_index :consent_tier, name: :idx_digital_workers_consent_tier, if_not_exists: true
+        add_index :trust_score, name: :idx_digital_workers_trust_score, if_not_exists: true
+      end
+    end
+
+    # audit_log: composite principal+time query, action/node lookups
+    if table_exists?(:audit_log)
+      alter_table(:audit_log) do
+        add_index %i[principal_id created_at], name: :idx_audit_log_principal_time, if_not_exists: true
+        add_index :action, name: :idx_audit_log_action, if_not_exists: true
+        add_index :node, name: :idx_audit_log_node, if_not_exists: true
+      end
+    end
+
+    # webhook_deliveries: event/time/success filtering
+    if table_exists?(:webhook_deliveries)
+      alter_table(:webhook_deliveries) do
+        add_index :event_name, name: :idx_webhook_deliveries_event_name, if_not_exists: true
+        add_index :delivered_at, name: :idx_webhook_deliveries_delivered_at, if_not_exists: true
+        add_index :success, name: :idx_webhook_deliveries_success, if_not_exists: true
+      end
+    end
+
+    # webhook_dead_letters: event/time filtering
+    if table_exists?(:webhook_dead_letters)
+      alter_table(:webhook_dead_letters) do
+        add_index :event_name, name: :idx_webhook_dead_letters_event_name, if_not_exists: true
+        add_index :created_at, name: :idx_webhook_dead_letters_created_at, if_not_exists: true
+      end
+    end
+
+    # conversations: identity and recency lookups
+    if table_exists?(:conversations)
+      alter_table(:conversations) do
+        add_index :caller_identity, name: :idx_conversations_caller_identity, if_not_exists: true
+        add_index :updated_at, name: :idx_conversations_updated_at, if_not_exists: true
+      end
+    end
+
+    # approval_queue: requester/reviewer lookups
+    if table_exists?(:approval_queue)
+      alter_table(:approval_queue) do
+        add_index :requester_id, name: :idx_approval_queue_requester_id, if_not_exists: true
+        add_index :reviewer_id, name: :idx_approval_queue_reviewer_id, if_not_exists: true
+      end
+    end
+
+    # rbac_role_assignments: role and expiry lookups
+    if table_exists?(:rbac_role_assignments)
+      alter_table(:rbac_role_assignments) do
+        add_index :role, name: :idx_rbac_role_assignments_role, if_not_exists: true
+        add_index :expires_at, name: :idx_rbac_role_assignments_expires_at, if_not_exists: true
+      end
+    end
+
+    # rbac_cross_team_grants: target team and expiry lookups
+    if table_exists?(:rbac_cross_team_grants)
+      alter_table(:rbac_cross_team_grants) do
+        add_index :target_team, name: :idx_rbac_cross_team_grants_target_team, if_not_exists: true
+        add_index :expires_at, name: :idx_rbac_cross_team_grants_expires_at, if_not_exists: true
+      end
+    end
+
+    # memory_traces: consolidation and source agent lookups
+    if table_exists?(:memory_traces)
+      existing_cols = schema(:memory_traces).map(&:first)
+
+      if existing_cols.include?(:consolidation_candidate)
+        alter_table(:memory_traces) do
+          add_index :consolidation_candidate, name: :idx_memory_traces_consolidation, if_not_exists: true
+        end
+      end
+
+      if existing_cols.include?(:source_agent_id)
+        alter_table(:memory_traces) do
+          add_index :source_agent_id, name: :idx_memory_traces_source_agent_id, if_not_exists: true
+        end
+      end
+    end
+
+    # agent_cluster_tasks: time-based querying
+    if table_exists?(:agent_cluster_tasks)
+      alter_table(:agent_cluster_tasks) do
+        add_index :created_at, name: :idx_agent_cluster_tasks_created_at, if_not_exists: true
+      end
+    end
+
+    # finlog_executions: environment_id FK without index
+    if table_exists?(:finlog_executions)
+      alter_table(:finlog_executions) do
+        add_index :environment_id, name: :idx_finlog_exec_environment_id, if_not_exists: true
+      end
+    end
+  end
+
+  down do
+    [
+      [:runners, %i[
+        idx_runners_extension_id idx_runners_namespace idx_runners_name idx_runners_extension_name
+      ]],
+      [:tasks, %i[idx_tasks_relationship_id]],
+      [:digital_workers, %i[idx_digital_workers_consent_tier idx_digital_workers_trust_score]],
+      [:audit_log, %i[idx_audit_log_principal_time idx_audit_log_action idx_audit_log_node]],
+      [:webhook_deliveries, %i[
+        idx_webhook_deliveries_event_name idx_webhook_deliveries_delivered_at idx_webhook_deliveries_success
+      ]],
+      [:webhook_dead_letters, %i[
+        idx_webhook_dead_letters_event_name idx_webhook_dead_letters_created_at
+      ]],
+      [:conversations, %i[idx_conversations_caller_identity idx_conversations_updated_at]],
+      [:approval_queue, %i[idx_approval_queue_requester_id idx_approval_queue_reviewer_id]],
+      [:rbac_role_assignments, %i[idx_rbac_role_assignments_role idx_rbac_role_assignments_expires_at]],
+      [:rbac_cross_team_grants, %i[
+        idx_rbac_cross_team_grants_target_team idx_rbac_cross_team_grants_expires_at
+      ]],
+      [:memory_traces, %i[idx_memory_traces_consolidation idx_memory_traces_source_agent_id]],
+      [:agent_cluster_tasks, %i[idx_agent_cluster_tasks_created_at]],
+      [:finlog_executions, %i[idx_finlog_exec_environment_id]]
+    ].each do |table, indexes|
+      next unless table_exists?(table)
+
+      alter_table(table) do
+        indexes.each do |idx_name|
+          drop_index nil, name: idx_name, if_exists: true
+        end
+      end
+    end
+  end
+end

data/lib/legion/data/migrations/051_fix_tasks_created_at.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next unless table_exists?(:tasks)
+
+    existing_cols = schema(:tasks).map(&:first)
+    next if existing_cols.include?(:created_at)
+
+    if adapter_scheme == :postgres
+      # Add a generated column so retention/archival queries using created_at work transparently
+      run 'ALTER TABLE tasks ADD COLUMN created_at TIMESTAMPTZ GENERATED ALWAYS AS (created) STORED'
+      run 'CREATE INDEX IF NOT EXISTS idx_tasks_created_at ON tasks (created_at)'
+    else
+      # SQLite/MySQL: add real column and backfill from created
+      alter_table(:tasks) do
+        add_column :created_at, DateTime, default: Sequel::CURRENT_TIMESTAMP
+      end
+
+      run 'UPDATE tasks SET created_at = created WHERE created_at IS NULL'
+
+      alter_table(:tasks) do
+        add_index :created_at, name: :idx_tasks_created_at, if_not_exists: true
+      end
+    end
+  end
+
+  down do
+    next unless table_exists?(:tasks)
+
+    existing_cols = schema(:tasks).map(&:first)
+    next unless existing_cols.include?(:created_at)
+
+    if adapter_scheme == :postgres
+      run 'DROP INDEX IF EXISTS idx_tasks_created_at'
+      run 'ALTER TABLE tasks DROP COLUMN IF EXISTS created_at'
+    else
+      alter_table(:tasks) do
+        drop_index :created_at, name: :idx_tasks_created_at, if_exists: true
+        drop_column :created_at
+      end
+    end
+  end
+end

data/lib/legion/data/migrations/052_drop_redundant_apollo_indexes.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    # PostgreSQL only — these auto-named indexes from migration 012 are exact duplicates
+    # of explicitly named indexes added in migration 047.
+    next unless adapter_scheme == :postgres
+
+    run 'DROP INDEX IF EXISTS apollo_entries_status_index'
+    run 'DROP INDEX IF EXISTS apollo_relations_from_entry_id_index'
+    run 'DROP INDEX IF EXISTS apollo_relations_to_entry_id_index'
+    run 'DROP INDEX IF EXISTS apollo_expertise_agent_id_index'
+    run 'DROP INDEX IF EXISTS apollo_expertise_domain_index'
+  end
+
+  down do
+    next unless adapter_scheme == :postgres
+
+    # Recreate the auto-named indexes that migration 012 created inline.
+    # idx_apollo_status, idx_apollo_rel_from, etc. from migration 047 remain in place.
+    run 'CREATE INDEX IF NOT EXISTS apollo_entries_status_index ON apollo_entries (status)' \
+      if table_exists?(:apollo_entries)
+    run 'CREATE INDEX IF NOT EXISTS apollo_relations_from_entry_id_index ON apollo_relations (from_entry_id)' \
+      if table_exists?(:apollo_relations)
+    run 'CREATE INDEX IF NOT EXISTS apollo_relations_to_entry_id_index ON apollo_relations (to_entry_id)' \
+      if table_exists?(:apollo_relations)
+    run 'CREATE INDEX IF NOT EXISTS apollo_expertise_agent_id_index ON apollo_expertise (agent_id)' \
+      if table_exists?(:apollo_expertise)
+    run 'CREATE INDEX IF NOT EXISTS apollo_expertise_domain_index ON apollo_expertise (domain)' \
+      if table_exists?(:apollo_expertise)
+  end
+end

data/lib/legion/data/migrations/053_add_tasks_relationship_fk.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    # PostgreSQL only — add FK constraint for tasks.relationship_id with ON DELETE SET NULL.
+    # Orphaned values must be cleaned first.
+    next unless adapter_scheme == :postgres
+    next unless table_exists?(:tasks)
+    next unless table_exists?(:relationships)
+
+    # Clean orphaned relationship_id values before adding constraint
+    run <<~SQL
+      UPDATE tasks
+      SET relationship_id = NULL
+      WHERE relationship_id IS NOT NULL
+        AND relationship_id NOT IN (SELECT id FROM relationships);
+    SQL
+
+    # Skip if constraint already exists (idempotency guard)
+    constraint_exists = self[:pg_constraint].where(conname: 'fk_tasks_relationship_id').any?
+
+    unless constraint_exists
+      run <<~SQL
+        ALTER TABLE tasks
+          ADD CONSTRAINT fk_tasks_relationship_id
+          FOREIGN KEY (relationship_id) REFERENCES relationships(id)
+          ON DELETE SET NULL;
+      SQL
+    end
+  end
+
+  down do
+    next unless adapter_scheme == :postgres
+    next unless table_exists?(:tasks)
+
+    run 'ALTER TABLE tasks DROP CONSTRAINT IF EXISTS fk_tasks_relationship_id'
+  end
+end

data/lib/legion/data/migrations/054_add_component_type_to_functions.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next unless table_exists?(:functions)
+    next if schema(:functions).any? { |c, _| c == :component_type }
+
+    alter_table(:functions) do
+      add_column :component_type, String, size: 32, null: false, default: 'runner'
+      add_index :component_type, name: :idx_functions_component_type, if_not_exists: true
+    end
+  end
+
+  down do
+    next unless table_exists?(:functions)
+    next unless schema(:functions).any? { |c, _| c == :component_type }
+
+    alter_table(:functions) do
+      drop_index :component_type, name: :idx_functions_component_type, if_exists: true
+      drop_column :component_type
+    end
+  end
+end

data/lib/legion/data/migrations/055_add_definition_to_functions.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next unless table_exists?(:functions)
+    next if schema(:functions).any? { |c, _| c == :definition }
+
+    alter_table(:functions) do
+      add_column :definition, String, text: true, null: true
+    end
+  end
+
+  down do
+    next unless table_exists?(:functions)
+    next unless schema(:functions).any? { |c, _| c == :definition }
+
+    alter_table(:functions) do
+      drop_column :definition
+    end
+  end
+end

data/lib/legion/data/migrations/056_add_absorber_patterns.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next if table_exists?(:absorber_patterns)
+
+    create_table(:absorber_patterns) do
+      primary_key :id
+      foreign_key :function_id, :functions, null: false, on_delete: :cascade, index: true
+      String :pattern_type, size: 32, null: false, default: 'url'
+      String :pattern, size: 1024, null: false
+      Integer :priority, null: false, default: 0
+      TrueClass :active, null: false, default: true
+      String :tenant_id, size: 64, null: true
+      DateTime :created_at, null: false, default: Sequel::CURRENT_TIMESTAMP
+      DateTime :updated_at, null: true
+
+      index :pattern_type, name: :idx_absorber_patterns_pattern_type
+      index :active, name: :idx_absorber_patterns_active
+      index :tenant_id, name: :idx_absorber_patterns_tenant_id
+      index %i[pattern_type active], name: :idx_absorber_patterns_type_active
+    end
+  end
+
+  down do
+    drop_table?(:absorber_patterns)
+  end
+end

data/lib/legion/data/migrations/057_add_routing_key_to_runners.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+Sequel.migration do
+  up do
+    next unless table_exists?(:runners)
+    next if schema(:runners).any? { |c, _| c == :routing_key }
+
+    alter_table(:runners) do
+      add_column :routing_key, String, size: 512, null: true
+      add_index :routing_key, name: :idx_runners_routing_key, if_not_exists: true
+    end
+  end
+
+  down do
+    next unless table_exists?(:runners)
+    next unless schema(:runners).any? { |c, _| c == :routing_key }
+
+    alter_table(:runners) do
+      drop_index :routing_key, name: :idx_runners_routing_key, if_exists: true
+      drop_column :routing_key
+    end
+  end
+end

data/lib/legion/data/retention.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative 'archival/policy'
+
 module Legion
   module Data
     module Retention
@@ -7,10 +9,11 @@ module Legion
       DEFAULT_ARCHIVE_AFTER_DAYS = 90
 
       class << self
-        def archive_old_records(table:, date_column: :created_at, archive_after_days: DEFAULT_ARCHIVE_AFTER_DAYS)
+        def archive_old_records(table:, date_column: nil, archive_after_days: DEFAULT_ARCHIVE_AFTER_DAYS)
           db = Legion::Data.connection
           return { archived: 0, table: table } unless db
 
+          date_column = resolve_date_column(table, date_column)
           cutoff = Time.now - (archive_after_days * 86_400)
           archive_table = archive_table_name(table)
 
@@ -18,7 +21,7 @@ module Legion
 
           count = 0
           db.transaction do
-            records = db[table].where(Sequel.lit("#{date_column} < ?", cutoff))
+            records = db[table].where(Sequel.identifier(date_column) < cutoff)
             count = records.count
             if count.positive?
               db[archive_table].multi_insert(records.all)
@@ -30,13 +33,14 @@ module Legion
           { archived: count, table: table }
         end
 
-        def purge_expired_records(table:, date_column: :created_at, retention_years: DEFAULT_RETENTION_YEARS)
+        def purge_expired_records(table:, date_column: nil, retention_years: DEFAULT_RETENTION_YEARS)
           db = Legion::Data.connection
           archive_table = archive_table_name(table)
           return { purged: 0, table: table } unless db&.table_exists?(archive_table)
 
+          date_column = resolve_date_column(table, date_column)
           cutoff = Time.now - (retention_years * 365 * 86_400)
-          expired = db[archive_table].where(Sequel.lit("#{date_column} < ?", cutoff))
+          expired = db[archive_table].where(Sequel.identifier(date_column) < cutoff)
           count = expired.count
           expired.delete if count.positive?
           Legion::Logging.info "Purged #{count} expired row(s) from #{archive_table}" if defined?(Legion::Logging) && count.positive?
@@ -44,9 +48,10 @@ module Legion
           { purged: count, table: table }
         end
 
-        def retention_status(table:, date_column: :created_at)
+        def retention_status(table:, date_column: nil)
           db = Legion::Data.connection
           archive_table = archive_table_name(table)
+          date_column = resolve_date_column(table, date_column)
 
           active_count = db&.table_exists?(table) ? db[table].count : 0
           archived_count = db&.table_exists?(archive_table) ? db[archive_table].count : 0
@@ -70,6 +75,16 @@ module Legion
 
         private
 
+        def resolve_date_column(table, date_column)
+          return date_column if date_column
+
+          if defined?(Legion::Data::Archival::Policy::DATE_COLUMN_OVERRIDES)
+            Legion::Data::Archival::Policy::DATE_COLUMN_OVERRIDES[table.to_s] || :created_at
+          else
+            :created_at
+          end
+        end
+
         def ensure_archive_table!(db, source_table, archive_table)
           return if db.table_exists?(archive_table)
 

data/lib/legion/data/version.rb

@@ -2,6 +2,6 @@
 
 module Legion
   module Data
-    VERSION = '1.6.9'
+    VERSION = '1.6.11'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: legion-data
 version: !ruby/object:Gem::Version
-  version: 1.6.9
+  version: 1.6.11
 platform: ruby
 authors:
 - Esity
@@ -176,6 +176,15 @@ files:
 - lib/legion/data/migrations/046_add_metering_hourly_rollup.rb
 - lib/legion/data/migrations/047_apollo_knowledge_capture.rb
 - lib/legion/data/migrations/048_add_financial_logging.rb
+- lib/legion/data/migrations/049_add_remote_invocable_to_functions.rb
+- lib/legion/data/migrations/050_add_missing_indexes.rb
+- lib/legion/data/migrations/051_fix_tasks_created_at.rb
+- lib/legion/data/migrations/052_drop_redundant_apollo_indexes.rb
+- lib/legion/data/migrations/053_add_tasks_relationship_fk.rb
+- lib/legion/data/migrations/054_add_component_type_to_functions.rb
+- lib/legion/data/migrations/055_add_definition_to_functions.rb
+- lib/legion/data/migrations/056_add_absorber_patterns.rb
+- lib/legion/data/migrations/057_add_routing_key_to_runners.rb
 - lib/legion/data/model.rb
 - lib/legion/data/models/apollo_access_log.rb
 - lib/legion/data/models/apollo_entry.rb