ilios 0.1.0

data/ext/ilios/statement.c

@@ -0,0 +1,211 @@
+#include "ilios.h"
+
+static void statement_mark(void *ptr);
+static void statement_destroy(void *ptr);
+static size_t statement_memsize(const void *ptr);
+static void statement_compact(void *ptr);
+
+const rb_data_type_t cassandra_statement_data_type = {
+    "Ilios::Cassandra::Statement",
+    {
+        statement_mark,
+        statement_destroy,
+        statement_memsize,
+        statement_compact,
+    },
+    0, 0,
+    RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED | RUBY_TYPED_FROZEN_SHAREABLE,
+};
+
+void statement_default_config(CassandraStatement *cassandra_statement)
+{
+    VALUE config = rb_cvar_get(mCassandra, id_cvar_config);
+
+    cass_statement_set_paging_size(cassandra_statement->statement, NUM2INT(rb_hash_aref(config, sym_page_size)));
+}
+
+static int hash_cb(VALUE key, VALUE value, VALUE statement)
+{
+    CassandraStatement *cassandra_statement = (CassandraStatement *)statement;
+    const CassDataType* data_type;
+    CassValueType value_type;
+    CassError result;
+    const char *name;
+
+    if (SYMBOL_P(key)) {
+        key = rb_sym2str(key);
+    }
+    name = StringValueCStr(key);
+
+    data_type = cass_prepared_parameter_data_type_by_name(cassandra_statement->prepared, name);
+    if (data_type == NULL) {
+        rb_raise(eStatementError, "Invalid name %s was given.", name);
+    }
+    value_type = cass_data_type_type(data_type);
+
+    if (NIL_P(value)) {
+        result = cass_statement_bind_null_by_name(cassandra_statement->statement, name);
+        goto result_check;
+    }
+
+    switch (value_type) {
+    case CASS_VALUE_TYPE_TINY_INT:
+        {
+            long v = NUM2LONG(value);
+
+            if (v < INT8_MIN || v > INT8_MAX) {
+                rb_raise(rb_eRangeError, "Invalid value: %ld", v);
+            }
+            result = cass_statement_bind_int8_by_name(cassandra_statement->statement, name, (cass_int8_t)v);
+        }
+        break;
+
+    case CASS_VALUE_TYPE_SMALL_INT:
+        {
+            long v = NUM2LONG(value);
+
+            if (v < INT16_MIN || v > INT16_MAX) {
+                rb_raise(rb_eRangeError, "Invalid value: %ld", v);
+            }
+
+            result = cass_statement_bind_int16_by_name(cassandra_statement->statement, name, (cass_int16_t)v);
+        }
+        break;
+
+    case CASS_VALUE_TYPE_INT:
+        {
+            long v = NUM2LONG(value);
+
+            if (v < INT32_MIN || v > INT32_MAX) {
+                rb_raise(rb_eRangeError, "Invalid value: %ld", v);
+            }
+
+            result = cass_statement_bind_int32_by_name(cassandra_statement->statement, name, (cass_int32_t)v);
+        }
+        break;
+
+    case CASS_VALUE_TYPE_BIGINT:
+        result = cass_statement_bind_int64_by_name(cassandra_statement->statement, name, NUM2LONG(value));
+        break;
+
+    case CASS_VALUE_TYPE_FLOAT:
+        {
+            double v = NUM2DBL(value);
+
+            if (!isnan(v) && !isinf(v) && (v < -FLT_MAX || v > FLT_MAX)) {
+                rb_raise(rb_eRangeError, "Invalid value: %lf", v);
+            }
+
+            result = cass_statement_bind_float_by_name(cassandra_statement->statement, name, v);
+        }
+        break;
+
+    case CASS_VALUE_TYPE_DOUBLE:
+        result = cass_statement_bind_double_by_name(cassandra_statement->statement, name, NUM2DBL(value));
+        break;
+
+    case CASS_VALUE_TYPE_BOOLEAN:
+        {
+            cass_bool_t v = RTEST(value) ? cass_true : cass_false;
+            result = cass_statement_bind_bool_by_name(cassandra_statement->statement, name, v);
+        }
+        break;
+
+    case CASS_VALUE_TYPE_TEXT:
+    case CASS_VALUE_TYPE_ASCII:
+    case CASS_VALUE_TYPE_VARCHAR:
+        result = cass_statement_bind_string_by_name(cassandra_statement->statement, name, StringValueCStr(value));
+        break;
+
+    case CASS_VALUE_TYPE_TIMESTAMP:
+        if (rb_obj_class(value) != rb_cTime) {
+            if (rb_respond_to(value, id_to_time)) {
+                value = rb_funcall(value, id_to_time, 0);
+            } else {
+                rb_raise(rb_eTypeError, "no implicit conversion of %"PRIsVALUE" to Time", rb_obj_class(value));
+            }
+        }
+        result = cass_statement_bind_int64_by_name(cassandra_statement->statement, name, (cass_int64_t)(NUM2DBL(rb_Float(value)) * 1000));
+        break;
+
+    case CASS_VALUE_TYPE_UUID:
+        {
+            CassUuid uuid;
+            const char *uuid_string = StringValueCStr(value);
+
+            cass_uuid_from_string(uuid_string, &uuid);
+            result = cass_statement_bind_uuid_by_name(cassandra_statement->statement, name, uuid);
+        }
+        break;
+
+    default:
+        rb_raise(rb_eTypeError, "Unsupported %"PRIsVALUE" type: %s=%"PRIsVALUE"", rb_obj_class(value), name, value);
+    }
+
+result_check:
+    if (result != CASS_OK) {
+        rb_raise(eStatementError, "Failed to bind value: %s", cass_error_desc(result));
+    }
+
+    return ST_CONTINUE;
+}
+
+static VALUE statement_bind(VALUE self, VALUE hash)
+{
+    CassandraStatement *cassandra_statement;
+
+    Check_Type(hash, T_HASH);
+    TypedData_Get_Struct(self, CassandraStatement, &cassandra_statement_data_type, cassandra_statement);
+
+    rb_hash_foreach(hash, hash_cb, (VALUE)cassandra_statement);
+    return self;
+}
+
+static VALUE statement_page_size(VALUE self, VALUE page_size)
+{
+    CassandraStatement *cassandra_statement;
+
+    GET_STATEMENT(self, cassandra_statement);
+    cass_statement_set_paging_size(cassandra_statement->statement, NUM2INT(page_size));
+    return self;
+}
+
+static void statement_mark(void *ptr)
+{
+    CassandraStatement *cassandra_statement = (CassandraStatement *)ptr;
+    rb_gc_mark_movable(cassandra_statement->session_obj);
+}
+
+static void statement_destroy(void *ptr)
+{
+    CassandraStatement *cassandra_statement = (CassandraStatement *)ptr;
+
+    if (cassandra_statement->prepared) {
+        cass_prepared_free(cassandra_statement->prepared);
+    }
+    if (cassandra_statement->statement) {
+        cass_statement_free(cassandra_statement->statement);
+    }
+    xfree(cassandra_statement);
+}
+
+static size_t statement_memsize(const void *ptr)
+{
+    return sizeof(CassandraStatement);
+}
+
+static void statement_compact(void *ptr)
+{
+    CassandraStatement *cassandra_statement = (CassandraStatement *)ptr;
+
+    cassandra_statement->session_obj = rb_gc_location(cassandra_statement->session_obj);
+}
+
+
+void Init_statement(void)
+{
+    rb_undef_alloc_func(cStatement);
+
+    rb_define_method(cStatement, "bind", statement_bind, 1);
+    rb_define_method(cStatement, "page_size=", statement_page_size, 1);
+}

data/ilios.gemspec

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require_relative 'lib/ilios/version'
+
+Gem::Specification.new do |spec|
+  spec.name = 'ilios'
+  spec.version = Ilios::VERSION
+  spec.authors = ['Watson']
+  spec.email = ['watson1978@gmail.com']
+
+  spec.summary = 'Cassandra driver written by C language'
+  spec.description = 'Cassandra driver written by C language'
+  spec.homepage = 'https://github.com/Watson1978/ilios'
+  spec.required_ruby_version = '>= 3.0.0'
+
+  spec.requirements << 'cmake'
+
+  spec.metadata['homepage_uri'] = spec.homepage
+  spec.metadata['source_code_uri'] = 'https://github.com/Watson1978/ilios'
+  # spec.metadata["changelog_uri"] = "TODO: Put your gem's CHANGELOG.md URL here."
+  spec.metadata['rubygems_mfa_required'] = 'true'
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files =
+    Dir.chdir(__dir__) do
+      `git ls-files -z`.split("\x0").reject do |f|
+        (f == __FILE__) || f.match(%r{\A(?:(?:test)/|\.(?:git|editorconfig|rubocop.*))})
+      end
+    end
+  spec.require_paths = ['lib']
+  spec.extensions << 'ext/ilios/extconf.rb'
+
+  spec.add_runtime_dependency('mini_portile2', '~> 2.8')
+end

data/lib/ilios/version.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Ilios
+  VERSION = '0.1.0'
+  public_constant :VERSION
+
+  CASSANDRA_CPP_DRIVER_VERSION = '2.17.0'
+  public_constant :CASSANDRA_CPP_DRIVER_VERSION
+
+  LIBUV_VERSION = '1.46.0'
+  public_constant :LIBUV_VERSION
+end

data/lib/ilios.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require 'ilios.so'
+require 'ilios/version'
+
+module Ilios
+  module Cassandra
+    @@config = {
+      keyspace: 'ilios',
+      hosts: ['127.0.0.1'],
+      timeout_ms: 5_000,
+      constant_delay_ms: 15_000,
+      max_speculative_executions: 2,
+      page_size: 10_000
+    }
+
+    def self.config=(config)
+      @@config = @@config.merge(config)
+    end
+
+    def self.session
+      Thread.current[:ilios_cassandra_session] ||= connect
+    end
+  end
+end

data/sig/ilios.rbs

@@ -0,0 +1,4 @@
+module Ilios
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: ilios
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Watson
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-11-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mini_portile2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+description: Cassandra driver written by C language
+email:
+- watson1978@gmail.com
+executables: []
+extensions:
+- ext/ilios/extconf.rb
+extra_rdoc_files: []
+files:
+- Dockerfile
+- Gemfile
+- README.md
+- Rakefile
+- docker-compose.yml
+- ext/ilios/cassandra.c
+- ext/ilios/extconf.rb
+- ext/ilios/future.c
+- ext/ilios/ilios.c
+- ext/ilios/ilios.h
+- ext/ilios/nogvl.c
+- ext/ilios/result.c
+- ext/ilios/session.c
+- ext/ilios/statement.c
+- ilios.gemspec
+- lib/ilios.rb
+- lib/ilios/version.rb
+- sig/ilios.rbs
+homepage: https://github.com/Watson1978/ilios
+licenses: []
+metadata:
+  homepage_uri: https://github.com/Watson1978/ilios
+  source_code_uri: https://github.com/Watson1978/ilios
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements:
+- cmake
+rubygems_version: 3.4.10
+signing_key:
+specification_version: 4
+summary: Cassandra driver written by C language
+test_files: []