timescaledb 0.2.8 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 240bc1a55d3955c734d79946868b5e878cacfa354f7b0a843b72f78c26a85c83
-  data.tar.gz: 23ccc7e91c0da1ea522e5c8a71cd4522478aa1c73ff0e9ee6f61e42b26f9955c
+  metadata.gz: 89553264709636b53e3d56174f3e244561c10f2c4801b922ba25adb8bf32aca7
+  data.tar.gz: f254f76133d13d6854f8f7dee865a0f3bb4adc57617c200ee276d1408541475a
 SHA512:
-  metadata.gz: 3ae2cda69cc099ad24a47aa4125ee10db96eb8186e261125f1e56ad6f4f2d805888d02d9eee92d48d18eab49d628918fef4e33234edc48ddb4ecb402d4518207
-  data.tar.gz: 65ee70a4b17944880979e71d1abee756a5f398f48cb048b575afbe4a0412f91d993ec97723c2b10d472ceafd405d7d013480824702c2a5db11e23f6eacb7f69a
+  metadata.gz: b30bbba9b0da08b3cad42fbc80e73c7aed2e37763d01cfd2b93d569e4bd65768a507fc6e25a757c7b83bd4f9ad3e7e18d98090b7a2f2e8bebcb5d16fb3a63d7c
+  data.tar.gz: bb0ba7379e86631d71256cf66cc506eb42511145f17e343e749bc10e28b7c394a482e9d4a3380ff9ee2af511be51d44529556566fe08421aa92b924755d492c8

data/lib/timescaledb/connection.rb

@@ -1,6 +1,11 @@
 require 'singleton'
 
 module Timescaledb
+  # Minimal connection setup for Timescaledb directly with the PG.
+  # The concept is use a singleton component that can query
+  # independently of the ActiveRecord::Base connections.
+  # This is useful for the extension and hypertable metadata.
+  # It can also #use_connection from active record if needed.
   class Connection
     include Singleton
 
@@ -34,10 +39,16 @@ module Timescaledb
       !@config.nil?
     end
 
+    # Override the connection with a raw PG connection.
+    # @param [PG::Connection] connection The raw PG connection.
+    def use_connection connection
+      @connection = connection
+    end
+
     private
 
     def connection
       @connection ||= PG.connect(@config)
     end
   end
-end
+end

data/lib/timescaledb/connection_handling.rb

@@ -1,16 +1,21 @@
 module Timescaledb
   class ConnectionNotEstablishedError < StandardError; end
 
-  # @param [String] config The postgres connection string.
+  module_function
+  
+  # @param [String] config with the postgres connection string.
   def establish_connection(config)
     Connection.instance.config = config
   end
-  module_function :establish_connection
+
+  # @param [PG::Connection] to use it directly from a raw connection
+  def use_connection conn
+    Connection.instance.use_connection conn
+  end
 
   def connection
     raise ConnectionNotEstablishedError.new unless Connection.instance.connected?
 
     Connection.instance
   end
-  module_function :connection
-end
+end

data/lib/timescaledb/extension.rb

@@ -0,0 +1,23 @@
+module Timescaledb
+
+  # Provides metadata around the extension in the database
+  module Extension
+    module_function
+    # @return String version of the timescaledb extension
+    def version
+      @version ||= Timescaledb.connection.query_first(<<~SQL)&.version
+        SELECT extversion as version
+        FROM pg_extension
+        WHERE extname = 'timescaledb'
+      SQL
+    end
+
+    def installed?
+      version.present?
+    end
+
+    def update!
+      Timescaledb.connection.execute('ALTER EXTENSION timescaledb UPDATE')
+    end
+  end
+end

data/lib/timescaledb/scenic/extension.rb

@@ -25,7 +25,6 @@ module Timescaledb
 
       # @override Scenic::Adapters::Postgres#create_view
       # to add the `with: ` keyword that can be used for such option.
-      #
       def create_view(name, version: nil, with: nil, sql_definition: nil, materialized: false, no_data: false)
         if version.present? && sql_definition.present?
           raise(
@@ -69,4 +68,4 @@ end
 
 
 Scenic::Adapters::Postgres.include(Timescaledb::Scenic::Extension)
-ActiveRecord::ConnectionAdapters::AbstractAdapter.include(Timescaledb::Scenic::MigrationHelpers)
+ActiveRecord::ConnectionAdapters::AbstractAdapter.prepend(Timescaledb::Scenic::MigrationHelpers)

data/lib/timescaledb/schema_dumper.rb

@@ -7,14 +7,29 @@ module Timescaledb
   # * retention policies
   # * continuous aggregates
   # * compression settings
+  # It also ignores Timescale related schemas when dumping the schema.
+  # It also ignores dumping options as extension is not installed or no hypertables are available.
   module SchemaDumper
     def tables(stream)
       super # This will call #table for each table in the database
-      return unless Timescaledb::Hypertable.table_exists?
 
-      timescale_hypertables(stream)
-      timescale_retention_policies(stream)
-      timescale_continuous_aggregates(stream) # Define these before any Scenic views that might use them
+      if exports_timescaledb_metadata?
+        timescale_hypertables(stream)
+        timescale_retention_policies(stream)
+        timescale_continuous_aggregates(stream) # Define these before any Scenic views that might use them
+      end
+    end
+
+    # Ignore dumps in case DB is not eligible for TimescaleDB metadata.
+    # @return [Boolean] true if the extension is installed and hypertables are available, otherwise false.
+    private def exports_timescaledb_metadata?
+      # Note it's safe to use the raw connection here because we're only reading from the database
+      # and not modifying it. We're also on the same connection pool as ActiveRecord::Base.
+      # The dump process also runs standalone, so we don't need to worry about the connection being
+      # used elsewhere.
+      Timescaledb.use_connection @connection.raw_connection
+
+      Timescaledb.extension.installed? && Timescaledb.hypertables.any?
     end
 
     # Ignores Timescale related schemas when dumping the schema

data/lib/timescaledb/version.rb

@@ -1,3 +1,3 @@
 module Timescaledb
-  VERSION = '0.2.8'
+  VERSION = '0.2.9'
 end

data/lib/timescaledb.rb

@@ -17,11 +17,20 @@ require_relative 'timescaledb/schema_dumper'
 require_relative 'timescaledb/stats'
 require_relative 'timescaledb/stats_report'
 require_relative 'timescaledb/migration_helpers'
+require_relative 'timescaledb/extension'
 require_relative 'timescaledb/version'
 
 module Timescaledb
   module_function
 
+  def connection
+    Connection.instance
+  end
+
+  def extension
+    Extension
+  end
+
   def chunks
     Chunk.all
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: timescaledb
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.2.9
 platform: ruby
 authors:
 - Jônatas Davi Paganini
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-04-30 00:00:00.000000000 Z
+date: 2024-06-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg
@@ -163,6 +163,7 @@ files:
 - lib/timescaledb/database/schema_statements.rb
 - lib/timescaledb/database/types.rb
 - lib/timescaledb/dimensions.rb
+- lib/timescaledb/extension.rb
 - lib/timescaledb/hypertable.rb
 - lib/timescaledb/job.rb
 - lib/timescaledb/job_stats.rb