ruby-kuzu 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d9ef51b4e013cad8b7f0434db1a6fac89068681273f0d53981761c56a570628d
-  data.tar.gz: db7e93a8c4537f33cfa3bf77a6315ccb334f74d437869b2f053f1d479be7ee6b
+  metadata.gz: 992fc29928cf3ac28d7ef615302fafdbb65c18311bcac23ad7178ff56c2949f9
+  data.tar.gz: 8ccd0253875842c9db1ef0b9de49d5a2120dd93193bcc19a840b020c8b3a8940
 SHA512:
-  metadata.gz: 2b70df49dbb9038ea0d02c0d6edb1e00c19ccfce943c464fd05efc55f6e848f110732aee64ce2910fc11107f1942fcfe5d8f19efd868bbffa5ceaeab8aaa7eae
-  data.tar.gz: 6633c48dd7c4cfad5b13de88b6123b7c7226dabcc2323a59831519db29b992f5efff9b48c946311934b96a3fffe8fe0ef17ca38afe3ba8f79094cc195ba2d880
+  metadata.gz: adce4adb10a022eb7ac6c079a66faaf91fb260e29a55f78b9ea235f3ecac3a6c2b068342e7274ac2bf3a8f6ba297ae494f19ba8e404c67d423f34976f0ce22d8
+  data.tar.gz: 36b289d90d94cf0273745e89590be11ee62d005beac8f2b9fb7024e08414683bb510f1f158a81ced28ece50f082ba21d9efbb338795738a472c1397fc8045586

data/History.md

@@ -1,6 +1,14 @@
 # Release History for ruby-kuzu
 
 ---
+## v0.1.0 [2025-06-17] Michael Granger <ged@FaerieMUD.org>
+
+Enhancements:
+
+- Add Connection#database
+- Squelch some warnings
+- Add Result#tuples and Result#[].
+
 
 ## v0.0.3 [2025-05-22] Michael Granger <ged@FaerieMUD.org>
 

data/ext/kuzu_ext/connection.c

@@ -170,7 +170,10 @@ rkuzu_connection_do_query( VALUE self, VALUE query )
 		rkuzu_connection_do_query_without_gvl, (void *)&qcall,
 		rkuzu_connection_cancel_query, (void *)&conn->conn );
 
+	_Pragma("GCC diagnostic push")
+	_Pragma("GCC diagnostic ignored \"-Wvoid-pointer-to-enum-cast\"")
 	query_state = (kuzu_state)result_ptr;
+	_Pragma("GCC diagnostic pop")
 
 	if ( query_state != KuzuSuccess ) {
 		char *err_detail = kuzu_query_result_get_error_message( &result );
@@ -278,6 +281,21 @@ rkuzu_connection_query_timeout_eq( VALUE self, VALUE timeout )
 }
 
 
+/*
+ * call-seq:
+ *    connection.database              -> database
+ *
+ * Return the database object the connection belongs to (a Kuzu::Database).
+ *
+ */
+static VALUE
+rkuzu_connection_database( VALUE self )
+{
+	rkuzu_connection *ptr = CHECK_CONNECTION( self );
+	return ptr->database;
+}
+
+
 
 /*
  * Document-class: Kuzu::Connection
@@ -306,5 +324,8 @@ rkuzu_init_connection( void )
 
 	rb_define_method( rkuzu_cKuzuConnection, "query_timeout=", rkuzu_connection_query_timeout_eq, 1 );
 
+	rb_define_method( rkuzu_cKuzuConnection, "database", rkuzu_connection_database, 0 );
+	rb_define_alias( rkuzu_cKuzuConnection, "db", "database" );
+
 	rb_require( "kuzu/connection" );
 }

data/ext/kuzu_ext/prepared_statement.c

@@ -181,7 +181,11 @@ rkuzu_prepared_statement_do_execute( VALUE self )
 	result_ptr = rb_thread_call_without_gvl(
 		rkuzu_connection_do_execute_without_gvl, (void *)&qcall,
 		rkuzu_connection_cancel_execute, (void *)&conn->conn );
+
+	_Pragma("GCC diagnostic push")
+	_Pragma("GCC diagnostic ignored \"-Wvoid-pointer-to-enum-cast\"")
 	execute_state = (kuzu_state)result_ptr;
+	_Pragma("GCC diagnostic pop")
 
 	if ( execute_state != KuzuSuccess ) {
 		char *err_detail = kuzu_query_result_get_error_message( &result );

data/lib/kuzu/prepared_statement.rb

@@ -18,6 +18,8 @@ class Kuzu::PreparedStatement
 	### then finished automatically, and the return value of the block returned
 	### instead.
 	def execute( **bound_variables, &block )
+		self.log.debug "Executing statement:\n%s\nwith variables:\n%p" %
+			[ self.query, bound_variables ]
 		self.bind( **bound_variables )
 		result = self._execute
 		return Kuzu::Result.wrap_block_result( result, &block )

data/lib/kuzu/result.rb

@@ -74,6 +74,19 @@ class Kuzu::Result
 	end
 
 
+	### Return the tuples from the current result set. This method is memoized
+	### for efficiency.
+	def tuples
+		return @_tuples ||= self.to_a
+	end
+
+
+	### Index operator: fetch the tuple at +index+ of the current result set.
+	def []( index )
+		return self.tuples[ index ]
+	end
+
+
 	### Return the next result set after this one as a Kuzu::Result, or `nil`if
 	### there is no next set.
 	def next_set
@@ -116,6 +129,7 @@ class Kuzu::Result
 
 	### Return an Enumerator that yields result tuples as Hashes.
 	def tuple_enum
+		self.log.debug "Fetching a tuple Enumerator"
 		return Enumerator.new do |yielder|
 			self.reset_iterator
 			while self.has_next?
@@ -126,7 +140,9 @@ class Kuzu::Result
 	end
 
 
+	### Return an Enumerator that yields a Result for each set.
 	def next_set_enum
+		self.log.debug "Fetching a result set Enumerator"
 		result = self
 		return Enumerator.new do |yielder|
 			while result

data/lib/kuzu.rb

@@ -11,7 +11,7 @@ module Kuzu
 
 
 	# Library version
-	VERSION = '0.0.3'
+	VERSION = '0.1.0'
 
 
 	# Set up a logger for Kuzu classes

data/spec/kuzu/connection_spec.rb

@@ -0,0 +1,36 @@
+# -*- ruby -*-
+
+require_relative '../spec_helper'
+
+require 'kuzu/connection'
+
+
+RSpec.describe( Kuzu::Connection ) do
+
+	let( :db ) { Kuzu.database }
+
+
+	it "can set the maximum number of threads for execution" do
+		connection = db.connect
+
+		expect {
+			connection.max_num_threads_for_exec += 1
+		}.to change { connection.max_num_threads_for_exec }.by( 1 )
+	end
+
+
+	it "shows the number of threads used for execution when inspected" do
+		connection = db.connect
+
+		expect( connection.inspect ).to match( /threads:\d+/i )
+	end
+
+
+	it "knows what database it's a connection for" do
+		connection = db.connect
+
+		expect( connection.database ).to eq( db )
+	end
+
+end
+

data/spec/kuzu/database_spec.rb

@@ -21,7 +21,7 @@ RSpec.describe( Kuzu::Database ) do
 
 
 	it "can be created read-only from an existing on-disk database" do
-		original = described_class.new( db_path.to_s )
+		_original = described_class.new( db_path.to_s )
 
 		ro = described_class.new( db_path.to_s, read_only: true )
 		expect( ro ).to be_read_only

data/spec/kuzu/result_spec.rb

@@ -78,7 +78,7 @@ RSpec.describe( Kuzu::Result ) do
 			expect( result ).to be_success
 			expect( result.num_columns ).to eq( 3 )
 			expect( result.column_names ).to eq([ "a.name", "b.name", "f.since" ])
-			expect( result.each.to_a ).to eq([
+			expect( result.to_a ).to eq([
 				{ "a.name" => "Adam", "b.name" => "Karissa", "f.since" => 2020 },
 				{ "a.name" => "Adam", "b.name" => "Zhang", "f.since" => 2020 },
 				{ "a.name" => "Karissa", "b.name" => "Zhang", "f.since" => 2021 },
@@ -89,6 +89,25 @@ RSpec.describe( Kuzu::Result ) do
 		end
 
 
+		it "can fetch individual result tuples via the index operator" do
+			setup_demo_db()
+
+			result = described_class.from_query( connection, <<~END_OF_QUERY )
+				MATCH ( a:User )-[ f:Follows ]->( b:User )
+			    RETURN a.name, b.name, f.since;
+			END_OF_QUERY
+
+			tuples = result.to_a
+
+			expect( result[0] ).to eq( tuples[0] )
+			expect( result[1] ).to eq( tuples[1] )
+			expect( result[2] ).to eq( tuples[2] )
+			expect( result[3] ).to eq( tuples[3] )
+
+			result.finish
+		end
+
+
 		it "can iterate over result sets" do
 			result = described_class.from_query( connection, <<~END_QUERY )
 				return 1;
@@ -97,7 +116,7 @@ RSpec.describe( Kuzu::Result ) do
 			END_QUERY
 
 			rval = result.each_set.flat_map do |subset|
-				subset.each.to_a
+				subset.to_a
 			end
 
 			expect( rval ).to eq([

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-kuzu
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Michael Granger
@@ -34,7 +34,7 @@ cert_chain:
   8qAqdfV+4u6Huu1KzAuDQCheyEyISsLST37sU/irV3czV6BiFipWag1XiJciRT3A
   wZqCfTNVHTdtsCbfdA1DsA3RdG2iEH3TOHzv1Rqzqh4=
   -----END CERTIFICATE-----
-date: 2025-05-22 00:00:00.000000000 Z
+date: 2025-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake-compiler
@@ -127,6 +127,7 @@ files:
 - lib/kuzu/rel.rb
 - lib/kuzu/result.rb
 - spec/kuzu/config_spec.rb
+- spec/kuzu/connection_spec.rb
 - spec/kuzu/database_spec.rb
 - spec/kuzu/prepared_statement_spec.rb
 - spec/kuzu/query_summary_spec.rb
@@ -157,7 +158,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 3.6.7
 specification_version: 4
 summary: A Ruby binding for the Kùzu embedded graph database.
 test_files: []

metadata.gz.sig

@@ -1,2 +1,2 @@
-zA���H_��8PT�%�9��T[:��l���mࡷ�a�U^��2��l�D��5�v�tv��O�h6f��E$"�ӡ���U�pWE�gCR7��.�鉐���蘩q���K��l��>#�m���~���C�*X+�0�R�mb��QĬ�	N���
o������T�{�J��%��eø6�v�3O��
�b>��§����i@���j����0>�)�}E�^�!$d�8y�h)�2�(~��"��4cO��wrB	�.m-��[��Ż��9;¹fa��3L{I���k���!-c�S͘L��A�^���� n��y��	��N�E��ٻ(��!Wl`�+Ǐ
-r��NS�Xt� 6P����U}ޅ�
�.dN
+�x02ڶ�X?�' ��]Ԛ;�h'���L�%�qo�q;��(C~���f
�US��RF-h�Ϛl�R�.L`�c�R�E��:�~nj�8c��Q�K�6�����r�𿿫��2ۍ_E�kȼm�#j�-H#]��t�����/#ȩ~L(K}�Y��8�T��<A���?���m�
+$��c;�T:5�f%b�TIn,۳`W~F~`j�H8Y��߉��#@qſ�,`�n���Ԅ�r�j.e�N&}!�f�� ��