Module: Legion::Data

Defined in:
lib/legion/data.rb,
lib/legion/data/rls.rb,
lib/legion/data/local.rb,
lib/legion/data/model.rb,
lib/legion/data/spool.rb,
lib/legion/data/helper.rb,
lib/legion/data/vector.rb,
lib/legion/data/extract.rb,
lib/legion/data/version.rb,
lib/legion/data/archival.rb,
lib/legion/data/archiver.rb,
lib/legion/data/settings.rb,
lib/legion/data/migration.rb,
lib/legion/data/retention.rb,
lib/legion/data/connection.rb,
lib/legion/data/event_store.rb,
lib/legion/data/models/node.rb,
lib/legion/data/models/task.rb,
lib/legion/data/audit_record.rb,
lib/legion/data/models/chain.rb,
lib/legion/data/models/runner.rb,
lib/legion/data/storage_tiers.rb,
lib/legion/data/models/setting.rb,
lib/legion/data/archival/policy.rb,
lib/legion/data/models/function.rb,
lib/legion/data/models/task_log.rb,
lib/legion/data/models/audit_log.rb,
lib/legion/data/models/extension.rb,
lib/legion/data/encryption/cipher.rb,
lib/legion/data/partition_manager.rb,
lib/legion/data/models/apollo_entry.rb,
lib/legion/data/models/audit_record.rb,
lib/legion/data/models/relationship.rb,
lib/legion/data/extract/handlers/csv.rb,
lib/legion/data/extract/handlers/pdf.rb,
lib/legion/data/extract/handlers/vtt.rb,
lib/legion/data/extract/handlers/base.rb,
lib/legion/data/extract/handlers/docx.rb,
lib/legion/data/extract/handlers/html.rb,
lib/legion/data/extract/handlers/json.rb,
lib/legion/data/extract/handlers/pptx.rb,
lib/legion/data/extract/handlers/text.rb,
lib/legion/data/extract/handlers/xlsx.rb,
lib/legion/data/extract/type_detector.rb,
lib/legion/data/models/digital_worker.rb,
lib/legion/data/event_store/projection.rb,
lib/legion/data/extract/handlers/jsonl.rb,
lib/legion/data/models/apollo_relation.rb,
lib/legion/data/encryption/key_provider.rb,
lib/legion/data/models/apollo_expertise.rb,
lib/legion/data/encryption/sequel_plugin.rb,
lib/legion/data/models/apollo_access_log.rb,
lib/legion/data/models/rbac_runner_grant.rb,
lib/legion/data/extract/handlers/markdown.rb,
lib/legion/data/models/rbac_role_assignment.rb,
lib/legion/data/models/rbac_cross_team_grant.rb

Defined Under Namespace

Modules: Archival, Archiver, AuditRecord, Connection, Encryption, EventStore, Extract, Helper, Local, Migration, Model, Models, PartitionManager, Retention, Rls, Settings, Spool, StorageTiers, Vector

Constant Summary collapse

VERSION = 
'1.6.18'

Class Method Summary collapse

Class Method Details

.can_read?(table_name) ⇒ Boolean

Returns:

  • (Boolean) 


83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/legion/data.rb', line 83

def can_read?(table_name)
  return false unless connected?

  adapter = Legion::Settings[:data][:adapter]&.to_s
  return true if adapter == 'sqlite'

  @read_privileges ||= {}
  return @read_privileges[table_name] unless @read_privileges[table_name].nil?

  @read_privileges[table_name] = connection
                                 .fetch("SELECT has_table_privilege(current_user, ?, 'SELECT') AS can", table_name.to_s)
                                 .first[:can] == true
rescue StandardError
  @read_privileges[table_name] = false if @read_privileges
  false
end

.can_write?(table_name) ⇒ Boolean

Returns:

  • (Boolean) 


66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/legion/data.rb', line 66

def can_write?(table_name)
  return false unless connected?

  adapter = Legion::Settings[:data][:adapter]&.to_s
  return true if adapter == 'sqlite'

  @write_privileges ||= {}
  return @write_privileges[table_name] unless @write_privileges[table_name].nil?

  @write_privileges[table_name] = connection
                                  .fetch("SELECT has_table_privilege(current_user, ?, 'INSERT') AS can", table_name.to_s)
                                  .first[:can] == true
rescue StandardError
  @write_privileges[table_name] = false if @write_privileges
  false
end

.connected? ⇒ Boolean

Returns:

  • (Boolean) 


60
61
62
63
64
 
# File 'lib/legion/data.rb', line 60

def connected?
  Legion::Settings[:data][:connected] == true
rescue StandardError
  false
end

.connection ⇒ Object 



45
46
47
 
# File 'lib/legion/data.rb', line 45

def connection
  Legion::Data::Connection.sequel
end

.connection_setup ⇒ Object 



31
32
33
34
35
 
# File 'lib/legion/data.rb', line 31

def connection_setup
  return if Legion::Settings[:data][:connected]

  Legion::Data::Connection.setup
end

.load_models ⇒ Object 



41
42
43
 
# File 'lib/legion/data.rb', line 41

def load_models
  Legion::Data::Models.load
end

.local ⇒ Object 



49
50
51
 
# File 'lib/legion/data.rb', line 49

def local
  Legion::Data::Local
end

.migrate ⇒ Object 



37
38
39
 
# File 'lib/legion/data.rb', line 37

def migrate
  Legion::Data::Migration.migrate
end

.reload_static_cache ⇒ Object 



121
122
123
124
125
 
# File 'lib/legion/data.rb', line 121

def reload_static_cache
  [Model::Extension, Model::Runner, Model::Function].each do |model|
    model.load_cache if model.respond_to?(:load_cache)
  end
end

.reset_privileges! ⇒ Object 



100
101
102
103
 
# File 'lib/legion/data.rb', line 100

def reset_privileges!
  @write_privileges = nil
  @read_privileges = nil
end

.setup ⇒ Object 



22
23
24
25
26
27
28
29
 
# File 'lib/legion/data.rb', line 22

def setup
  connection_setup
  migrate
  load_models
  setup_cache
  setup_local
  Legion::Logging.info 'Legion::Data setup complete' if defined?(Legion::Logging)
end

.setup_cache ⇒ Object 



105
106
107
108
109
 
# File 'lib/legion/data.rb', line 105

def setup_cache
  cache_settings = Legion::Settings[:data][:cache]
  setup_static_cache if cache_settings[:static_cache]
  setup_external_cache if cache_settings[:auto_enable] && defined?(::Legion::Cache)
end

.setup_external_cache ⇒ Object 



127
128
129
130
131
132
133
134
135
136
137
138
139
140
 
# File 'lib/legion/data.rb', line 127

def setup_external_cache
  ttl = Legion::Settings[:data][:cache][:ttl] || 60
  {
    Model::Relationship => 10,
    Model::Node         => 10,
    Model::Setting      => ttl
  }.each do |model, model_ttl|
    model.plugin :caching, ::Legion::Cache, ttl: model_ttl
    Legion::Logging.debug("Caching enabled for #{model} (ttl: #{model_ttl})") if defined?(Legion::Logging)
  rescue StandardError => e
    Legion::Logging.warn("Caching failed for #{model}: #{e.message}") if defined?(Legion::Logging)
  end
  Legion::Logging.info 'Legion::Data external cache connected' if defined?(Legion::Logging)
end

.setup_static_cache ⇒ Object 



111
112
113
114
115
116
117
118
119
 
# File 'lib/legion/data.rb', line 111

def setup_static_cache
  [Model::Extension, Model::Runner, Model::Function].each do |model|
    model.plugin :static_cache
    Legion::Logging.debug("StaticCache enabled for #{model}") if defined?(Legion::Logging)
  rescue StandardError => e
    Legion::Logging.warn("StaticCache failed for #{model}: #{e.message}") if defined?(Legion::Logging)
  end
  Legion::Logging.info 'Legion::Data static cache loaded' if defined?(Legion::Logging)
end

.shutdown ⇒ Object 



142
143
144
145
146
 
# File 'lib/legion/data.rb', line 142

def shutdown
  Legion::Data::Local.shutdown if defined?(Legion::Data::Local) && Legion::Data::Local.connected?
  Legion::Data::Connection.shutdown
  Legion::Logging.info 'Legion::Data shutdown complete' if defined?(Legion::Logging)
end

.stats ⇒ Object 



53
54
55
56
57
58
 
# File 'lib/legion/data.rb', line 53

def stats
  {
    shared: Legion::Data::Connection.stats,
    local:  Legion::Data::Local.stats
  }
end