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/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/relationship.rb,
lib/legion/data/extract/handlers/csv.rb,
lib/legion/data/extract/handlers/pdf.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, Connection, Encryption, EventStore, Extract, Helper, Local, Migration, Model, Models, PartitionManager, Retention, Rls, Settings, Spool, StorageTiers, Vector

Constant Summary collapse

VERSION = 
'1.6.7'

Class Method Summary collapse

Class Method Details

.can_read?(table_name) ⇒ Boolean

Returns:

  • (Boolean) 


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

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) 


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

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) 


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

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

.connection ⇒ Object 



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

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

.connection_setup ⇒ Object 



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

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

  Legion::Data::Connection.setup
end

.load_models ⇒ Object 



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

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

.local ⇒ Object 



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

def local
  Legion::Data::Local
end

.migrate ⇒ Object 



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

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

.reload_static_cache ⇒ Object 



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

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 



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

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

.setup ⇒ Object 



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

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 



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

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 



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

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 



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

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 



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

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 



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

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