kirei 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 86e3ce57ba0d801b5279632d302837a42195565c3daa66a2b5fbd6adb31f0176
-  data.tar.gz: a3ba957501c2603cc8cfba22282ab0c281ccf8227f99fedad78af0c980f3edd3
+  metadata.gz: c1adc30a0abdfee8e565869c726e247ab07f3ab1cd535c2d7874182f501962a9
+  data.tar.gz: 533abff887e8aca8eda259ad66679b76b7d25d4236966a80fcf32dfafe18dcd2
 SHA512:
-  metadata.gz: '078c2a848f346f8e25c0b9fcacc72d0d2e3b8f82620df2cffe4fee683e9d44aba43d3de635dd918449369617aa97362d2f66adba26d52c625565ff614eafe5d9'
-  data.tar.gz: a57e28a79ff8fe6bf47060550d11521b14112f22e15088a365bf7965ef0caa1a9f56a80e55571477cea8807c63775513fdd04eb63bda355f677ce5aba95d032d
+  metadata.gz: ed22516b55aa7677b85dbf09e1f6f88822646574897e55831318eead12759f275e36394076be292be54d02f08223f78e827da3990d82871bcef6bd8d98a6f26c
+  data.tar.gz: 6aeb8fcc2cdf887da4824e1075c935f186f67d4639e01d6b099341a60e6c9ae7b6ce7ed30ad81ab1a40583152cf7006f4205c78d10b6aa775f45c345a53ff3f0

data/README.md

@@ -13,7 +13,7 @@ TL;DR:
 * **zero magic**
 * **strict typing**
 * **very few low level dependencies**
-* ultra low memory footprint
+* low memory footprint
 * high performance
 * simple to understand
 

data/lib/kirei/model/class_methods.rb

@@ -85,12 +85,40 @@ module Kirei
         return unless App.config.db_extensions.include?(:pg_json)
 
         attributes.each_pair do |key, value|
-          next unless value.is_a?(Hash) || value.is_a?(Array)
-
-          attributes[key] = T.unsafe(Sequel).pg_jsonb_wrap(value)
+          if vector_column?(key.to_s)
+            attributes[key] = cast_to_vector(value)
+          elsif value.is_a?(Hash) || value.is_a?(Array)
+            attributes[key] = T.unsafe(Sequel).pg_jsonb_wrap(value)
+          end
         end
       end
 
+      #
+      # install the gem "pgvector" if you need to use vector columns
+      # also add `:pgvector` to the `App.config.db_extensions` array
+      # and enable the vector extension on the database.
+      #
+      sig { params(column_name: String).returns(T::Boolean) }
+      def vector_column?(column_name)
+        _col_name, col_info = T.let(
+          db.schema(table_name.to_sym).find { _1[0] == column_name.to_sym },
+          [Symbol, T::Hash[Symbol, T.untyped]],
+        )
+        col_info.fetch(:db_type).match?(/vector\(\d+\)/)
+      end
+
+      # New method to cast an array to a vector
+      sig { params(value: T.any(T::Array[Numeric], Sequel::SQL::Expression)).returns(Sequel::SQL::Expression) }
+      def cast_to_vector(value)
+        return value if value.is_a?(Sequel::SQL::Expression) || value.is_a?(Sequel::SQL::PlaceholderLiteralString)
+
+        Kernel.raise("'pg_array' extension is not enabled") unless db.extension(:pg_array)
+
+        pg_array = T.unsafe(Sequel).pg_array(value)
+
+        Sequel.lit("?::vector", pg_array)
+      end
+
       sig do
         override.params(
           hash: T::Hash[Symbol, T.untyped],

data/lib/kirei/routing/rack_env_type.rb

@@ -13,9 +13,9 @@ module Kirei
           String,
           Numeric,
           TCPSocket,
-          Puma::Client,
+          ::Puma::Client,
           StringIO,
-          Puma::Configuration,
+          ::Puma::Configuration,
         )
       ]
     end

data/lib/kirei/services/runner.rb

@@ -10,7 +10,7 @@ module Kirei
       sig do
         type_parameters(:T)
           .params(
-            class_name: T.untyped,
+            class_name: T.untyped, # @TODO(lud, 17.08.2024): replace all non-word characters with `.`?
             log_tags: T::Hash[String, T.untyped],
             block: T.proc.returns(T.type_parameter(:T)),
           ).returns(T.type_parameter(:T))
@@ -24,12 +24,7 @@ module Kirei
         stop = Process.clock_gettime(Process::CLOCK_MONOTONIC, :float_millisecond)
         latency_in_ms = stop - T.must(start)
 
-        result = case service
-                 when Services::Result
-                   service.success? ? "success" : "failure"
-                 else
-                   "unknown"
-        end
+        result = service_result(service)
 
         metric_tags = Logging::Metric.inject_defaults({ "service.result" => result })
         ::StatsD.measure(class_name, latency_in_ms, tags: metric_tags)
@@ -42,13 +37,31 @@ module Kirei
         }
         logtags.merge!(log_tags)
 
-        Logging::Logger.call(level: Logging::Level::INFO, label: "Service Finished", meta: logtags)
+        Logging::Logger.call(level: log_level(result), label: "Service Finished", meta: logtags)
       end
 
       sig { params(proc: T.proc.returns(T.untyped)).returns(String) }
       private_class_method def self.source_location(proc)
         proc.source_location.join(":").gsub(App.root.to_s, "")
       end
+
+      sig { params(service: T.untyped).returns(String) }
+      def self.service_result(service)
+        case service
+        when Services::Result
+          service.success? ? "success" : "failure"
+        else
+          "unknown"
+        end
+      end
+
+      sig { params(result: String).returns(Logging::Level) }
+      private_class_method def self.log_level(result)
+        return Logging::Level::INFO if result == "success"
+        return Logging::Level::WARN if result == "failure"
+
+        Logging::Level::UNKNOWN
+      end
     end
   end
 end

data/lib/kirei/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module Kirei
-  VERSION = "0.4.1"
+  VERSION = "0.5.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kirei
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.0
 platform: ruby
 authors:
 - Ludwig Reinmiedl
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-06-10 00:00:00.000000000 Z
+date: 2024-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: oj