snowflaked 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7b821e9cbccec942c789a9d8b0afb547c1cba389283f46169623c4f185753bbd
-  data.tar.gz: 763c5b9c22393698b115742b50f51fbbbc122810e8c6a2942ee14032171ecb5b
+  metadata.gz: 2d467289c785dd03c1de80d82d2ad25657b43677affd2c3112090f42109f5ef5
+  data.tar.gz: 701ea640497cbf5555570ca8eb03c3d47cb88742d4b80b0a331fd72cdb345168
 SHA512:
-  metadata.gz: dc619f463e94136072da4ecc99532e33a21616b86bf32828fbe91ee24206e9bdcb64b384b425d631b6c386317d48f8edb490ca7a5c21de8cdd55fa65586ae486
-  data.tar.gz: 932623948b2a6cffa8cced8864e4fdd7efc837fb3f1f06051e4634ccad74840cfc49720377cf4ac20cf6dce19fd1c83895138d57d8af28a5ecf5eabb40b8886c
+  metadata.gz: f46bb795c7a598091deb5983e0ef3cec686dc515e129712258d67e4d7858a6e0ac07127275b60477a5d01170ebd11ff6c5fd9ebac195934c44935f675e03ff70
+  data.tar.gz: f3ac6a668615e9c4bbc0657c3bd360e1e5da63e982880bbdce91d2d97fa562d01547c7e8af3b3ba732aeb88a84e230ba09affd61f6e95244718161f4fe6cf999

data/README.md

@@ -1,5 +1,10 @@
 # Snowflaked
 
+[![CI](https://github.com/luizkowalski/snowflaked/actions/workflows/ci.yml/badge.svg)](https://github.com/luizkowalski/snowflaked/actions/workflows/ci.yml)
+[![Gem Version](https://badge.fury.io/rb/snowflaked.svg)](https://badge.fury.io/rb/snowflaked)
+[![Downloads](https://img.shields.io/gem/dt/snowflaked.svg)](https://rubygems.org/gems/snowflaked)
+[![License](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE.txt)
+
 A high-performance, thread-safe Snowflake ID generator for Ruby, powered by Rust.
 
 Snowflake IDs are 64-bit unique identifiers that encode a timestamp, machine ID, and sequence number. They're time-sortable (IDs created later are always larger), making them ideal for distributed systems where you need unique IDs without coordination between machines. Unlike UUIDs, Snowflake IDs are smaller, sortable, and index-friendly for databases.
@@ -38,16 +43,22 @@ class CreateUsers < ActiveRecord::Migration[8.1]
   def change
     create_table :users do |t|
       t.snowflake :external_id
+      t.bigint    :uid
     end
   end
 end
 ```
 
+Columns created with `t.snowflake` are automatically detected and will have Snowflake IDs generated for them.
+
+> [!WARNING]
+> SQLite does not support column comments, which Snowflaked uses to auto-detect snowflake columns other than `:id`. When using SQLite, you must explicitly declare snowflake columns using the `snowflake_id` helper in your model.
+
 If you want to generate Snowflake IDs for additional columns, you can do so by using the `snowflake_id` method, without having to migrate the table:
 
 ```ruby
 class User < ApplicationRecord
-  snowflake_id :external_id
+  snowflake_id :uid
 end
 ```
 
@@ -72,7 +83,7 @@ end
 ```ruby
 Snowflaked.configure do |config|
   config.machine_id = 42
-  config.epoch = Time.utc(2020, 1, 1)
+  config.epoch = Time.utc(1989, 1, 3) # When not configured, the epoch is set to the Unix epoch (January 1, 1970)
 end
 ```
 

data/ext/snowflaked/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snowflaked"
-version = "0.1.0"
+version = "0.1.2"
 edition = "2021"
 publish = false
 

data/ext/snowflaked/src/lib.rs

@@ -1,7 +1,8 @@
 use magnus::{function, prelude::*, Error, RHash, Ruby};
 use snowflaked::sync::Generator;
-use snowflaked::Snowflake;
+use snowflaked::{Builder, Snowflake};
 use std::sync::OnceLock;
+use std::time::UNIX_EPOCH;
 
 struct GeneratorState {
     generator: Generator,
@@ -13,11 +14,18 @@ static STATE: OnceLock<GeneratorState> = OnceLock::new();
 
 fn init_generator(machine_id: u16, epoch_ms: Option<u64>) -> bool {
     let was_empty = STATE.get().is_none();
+    let epoch_offset = epoch_ms.unwrap_or(0);
 
-    STATE.get_or_init(|| GeneratorState {
-        generator: Generator::new(machine_id),
-        epoch_offset: epoch_ms.unwrap_or(0),
-        machine_id,
+    STATE.get_or_init(|| {
+        let epoch = UNIX_EPOCH + std::time::Duration::from_millis(epoch_offset);
+
+        let generator = Builder::new().instance(machine_id).epoch(epoch).build();
+
+        GeneratorState {
+            generator,
+            epoch_offset,
+            machine_id,
+        }
     });
 
     was_empty

data/lib/snowflaked/model_extensions.rb

@@ -19,7 +19,7 @@ module Snowflaked
       def _snowflake_columns_from_comments
         return [] unless table_exists?
 
-        columns.select { |c| c.comment == Snowflaked::SchemaDefinitions::COMMENT }.map { |c| c.name.to_sym }
+        columns.filter_map { |col| col.name.to_sym if col.comment == Snowflaked::SchemaDefinitions::COMMENT }
       end
     end
 

data/lib/snowflaked/schema_definitions.rb

@@ -4,18 +4,13 @@ module Snowflaked
   module SchemaDefinitions
     COMMENT = "snowflaked"
 
-    module TableDefinition
-      def snowflake(name, **options)
-        options[:comment] = Snowflaked::SchemaDefinitions::COMMENT
-        column(name, :snowflake, **options)
+    module SnowflakeColumn
+      def snowflake(name, **)
+        column(name, :snowflake, comment: COMMENT, **)
       end
     end
 
-    module Table
-      def snowflake(name, **options)
-        options[:comment] = Snowflaked::SchemaDefinitions::COMMENT
-        column(name, :snowflake, **options)
-      end
-    end
+    TableDefinition = SnowflakeColumn
+    Table           = SnowflakeColumn
   end
 end

data/lib/snowflaked/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Snowflaked
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: snowflaked
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Luiz Eduardo Kowalski