sequel-xtdb 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1b33179262aa907817cbbf65cb0980a19bea740bc3bda9ab569a10fd49544e25
-  data.tar.gz: 25b43d46d3e1f2c9ceb846f6d09590d600a2fea2a9bad02cae930cc457494105
+  metadata.gz: feef2fdeeefa8b1adfd9d214aab772c02250814aa8e6ef36a79bc7eddc4a8386
+  data.tar.gz: 417e16888185230ad9f7714b7d0d322594457f6f0d8645c033a9203463103293
 SHA512:
-  metadata.gz: 5603a63bd7d20916755c3e4375ed4a5162832925c3615f0ac72459e1f37d46f3bd32a96b686cc1e09075f48c4cc766582750eb6e7b4c8e0961bb8c621c10eaed
-  data.tar.gz: 418b300f8ab58dbeda2b786953911eed8ddece9885132833f0b328e5f551ac79ca81e18751b5ab6668b9ca5e40b483b51ab5237a4630f581f858d5c325c111f6
+  metadata.gz: 28456e08dfc22572d63a5de8dbfba0b8f849a78478f1ec43336e2495212a4217bacf665d53faf7f205d54c75ef8df931064288dd079bf496dd85389bcb3d1911
+  data.tar.gz: cb4e306b7641e07812e146843d982104e143935c83d58ffc7cd480246f1dec99fc0f5946fa2ebd964eed4c50e338051785fa772119fe497d156cf88e82e2925c

data/.envrc

@@ -0,0 +1,7 @@
+# env-var using [direnv](https://github.com/direnv/direnv)
+PATH_add bin
+
+export XTDB_URL=xtdb://localhost:5432/xtdb
+
+# put overrides in .envrc.local
+source_env_if_exists .envrc.local

data/README.md

@@ -24,7 +24,7 @@ Shortcut to happiness:
 $ sequel 'xtdb://localhost:5432/xtdb'
 
 # ..or from repl/your project
-DB = Sequel.connect("xtdb://localhost:54321/xtdb")
+DB = Sequel.connect("xtdb://localhost:5432/xtdb")
 
 # then
 irb(main)> DB << "insert into products(_id, name, price) values(1, 'Spam', 1000), (2, 'Ham', 1200)"
@@ -32,6 +32,35 @@ irb(main)> DB["select * from products"].all
 => [{:_id=>2, :name=>"Ham", :price=>1200}, {:_id=>1, :name=>"Spam", :price=>1100}]
 ```
 
+### time-travel
+
+_these examples use the [activesupport time helpers](https://api.rubyonrails.org/classes/ActiveSupport/Duration.html)_
+
+```ruby
+DB = Sequel.connect("xtdb://localhost:5432/xtdb")
+
+# get a dataset (ie query)
+users = DB[:users]
+past, future = 2.days.ago, 2.days.from_now
+ds1, ds2 = users.as_of(valid: past), users.as_of(valid: future)
+
+# expect empty
+ds1.all
+ds1.insert(_id: 1, name: "James")
+
+# expect a user
+ds1.as_of(valid: 2.days.ago).all
+
+# add to future
+ds2.insert(_id: 2, name: "Jeremy")
+
+# expect only James
+users.all
+# expect both James and Jeremy
+ds2.as_of(valid: 2.days.from_now).all
+```
+
+
 ## Status
 
 Very early days :)  

data/lib/sequel/adapters/shared/xtdb.rb

@@ -4,9 +4,101 @@ module Sequel
       def database_type
         :xtdb
       end
+
+      def primary_key(_table)
+        # eg used for RETURNING on insert (prevents crash)
+        :_id
+      end
+
+      # Get a dataset with `current`, `valid` and `system` set.
+      #
+      # For selects this creates the SETTING pre-amble, e.g. 'SETTING DEFAULT VALID_TIME ...':
+      # ```
+      # DB.as_of(current: 2.weeks.ago).select(Sequel.lit('current_timestamp')).single_value
+      # ```
+      #
+      # A block can be provided as a convenience to stay in SQL-land (selects only):
+      # ```
+      # DB.as_of(current: 2.hours.ago) do
+      #   DB["select current_timestamp"]
+      # end.sql
+      # =>
+      # SETTING
+      #  CURRENT_TIME TO TIMESTAMP '2024-12-17T12:59:48+01:00'
+      # select current_timestamp
+      # ```
+      #
+      # When doing inserts, the `_valid_from` will be added (if not provided):
+      # ```
+      # DB[:products].as_of(valid: 2.weeks.ago).insert(_id: 1, name: 'Spam')
+      # ```
+      def as_of(...)
+        ds = @default_dataset.as_of(...)
+        return ds unless block_given?
+
+        yield.clone(append_sql: ds.select_setting_sql(""))
+      end
     end
 
     module DatasetMethods
+      Dataset.def_sql_method(self, :select,
+        [["if opts[:values]",
+          %w[values compounds order limit]],
+          ["else",
+            %w[setting select distinct columns from join where group having compounds order limit lock]]])
+
+      def as_of(valid: nil, system: nil, current: nil)
+        {valid: valid, system: system, current: current}.reject { |_k, v| v.nil? }.then do |opts|
+          clone(opts)
+        end
+      end
+
+      def server_version
+        0
+      end
+
+      def insert_values_sql(sql)
+        if (from_ix = opts[:columns].index(:_valid_from))
+          opts[:values][from_ix] = Sequel.lit("TIMESTAMP ?", opts[:values][from_ix])
+        end
+
+        if (to_ix = opts[:columns].index(:_valid_to))
+          opts[:values][to_ix] = Sequel.lit("TIMESTAMP ?", opts[:values][to_ix])
+        end
+
+        super
+      end
+
+      def insert_columns_sql(sql)
+        if opts[:valid] && !opts[:columns].index(:_valid_from)
+          opts[:columns] << :_valid_from
+          opts[:values] << opts[:valid]
+        end
+        super
+      end
+
+      def select_setting_sql(sql)
+        setting = opts.slice(:current, :valid, :system)
+        return sql if setting.empty?
+
+        cast_value = ->(v) do
+          case v
+          when DateTime, Time
+            literal_append "TIMESTAMP ", v.iso8601
+          when Date
+            literal_append "DATE ", v.iso8601
+          end
+        end
+        sql << "SETTING "
+        sql << setting.map do |k, v|
+          if k == :current
+            literal_append "CURRENT_TIME TO TIMESTAMP ", v.iso8601
+          else
+            "DEFAULT #{k.upcase}_TIME AS OF #{cast_value[v]}"
+          end
+        end.join(", ")
+        sql << " "
+      end
     end
   end
 end

data/lib/sequel/adapters/xtdb.rb

@@ -13,6 +13,15 @@ module Sequel
       def adapter_initialize
         # XTDB can't handle this SET-command
         @opts[:force_standard_strings] = false
+
+        Sequel.database_timezone = :utc
+        Sequel.application_timezone = :local
+
+        if (app_tz = @opts[:application_timezone])
+          Sequel.extension(:named_timezones)
+          Sequel.application_timezone = app_tz
+        end
+
         super
       end
 
@@ -23,6 +32,12 @@ module Sequel
 
     class Dataset < Sequel::Postgres::Dataset
       include ::Sequel::XTDB::DatasetMethods
+
+      private
+
+      def default_timestamp_format
+        "'%Y-%m-%d %H:%M:%S'"
+      end
     end
   end
 end

data/lib/sequel/xtdb/version.rb

@@ -3,6 +3,6 @@
 module Sequel
   module XTDB
     # NOTE when releasing the git-tag will replace the version below
-    VERSION = "0.2.1"
+    VERSION = "0.3.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sequel-xtdb
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Gert Goet
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-12-13 00:00:00.000000000 Z
+date: 2024-12-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sequel
@@ -45,6 +45,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".envrc"
 - ".rspec"
 - ".standard.yml"
 - CHANGELOG.md