add DuckDB::LogicalType#set_alias

Author: otegami

ext/duckdb/logical_type.c

@@ -22,6 +22,7 @@ static VALUE duckdb_logical_type__internal_type(VALUE self);
 static VALUE duckdb_logical_type_dictionary_size(VALUE self);
 static VALUE duckdb_logical_type_dictionary_value_at(VALUE self, VALUE didx);
 static VALUE duckdb_logical_type__get_alias(VALUE self);
+static VALUE duckdb_logical_type__set_alias(VALUE self, VALUE aname);
 
 static const rb_data_type_t logical_type_data_type = {
     "DuckDB/LogicalType",
@@ -380,6 +381,27 @@ static VALUE duckdb_logical_type__get_alias(VALUE self) {
     return alias;
 }
 
+/*
+ *  call-seq:
+ *    col.logical_type.alias(alias) -> String
+ *
+ *  Return the set alias of the logical type.
+ *
+ */
+static VALUE duckdb_logical_type__set_alias(VALUE self, VALUE aname) {
+    rubyDuckDBLogicalType *ctx;
+    VALUE alias = Qnil;
+    const char *_alias = StringValuePtr(aname);
+
+    TypedData_Get_Struct(self, rubyDuckDBLogicalType, &logical_type_data_type, ctx);
+    duckdb_logical_type_set_alias(ctx->logical_type, _alias);
+    if (_alias != NULL) {
+        alias = rb_utf8_str_new_cstr(_alias);
+    }
+
+    return alias;
+}
+
 VALUE rbduckdb_create_logical_type(duckdb_logical_type logical_type) {
     VALUE obj;
     rubyDuckDBLogicalType *ctx;
@@ -415,6 +437,6 @@ void rbduckdb_init_duckdb_logical_type(void) {
     rb_define_method(cDuckDBLogicalType, "_internal_type", duckdb_logical_type__internal_type, 0);
     rb_define_method(cDuckDBLogicalType, "dictionary_size", duckdb_logical_type_dictionary_size, 0);
     rb_define_method(cDuckDBLogicalType, "dictionary_value_at", duckdb_logical_type_dictionary_value_at, 1);
-    rb_define_method(cDuckDBLogicalType, "alias", duckdb_logical_type_alias, 0);
     rb_define_method(cDuckDBLogicalType, "get_alias", duckdb_logical_type__get_alias, 0);
+    rb_define_method(cDuckDBLogicalType, "set_alias", duckdb_logical_type__set_alias, 1);
 }

lib/duckdb/logical_type.rb

@@ -3,6 +3,7 @@
 module DuckDB
   class LogicalType
     alias :alias get_alias
+    alias :alias= set_alias
 
     # returns logical type's type symbol
     # `:unknown` means that the logical type's type is unknown/unsupported by ruby-duckdb.

test/duckdb_test/logical_type_test.rb

@@ -118,6 +118,13 @@ def test_type
       assert_equal(EXPECTED_TYPES, logical_types.map(&:type))
     end
 
+    def test_alias
+      enum_column = @columns.find { |column| column.type == :enum }
+      enum_logical_type = enum_column.logical_type
+      assert_equal("mood", enum_logical_type.alias=("mood"))
+      assert_equal("mood", enum_logical_type.alias)
+    end
+
     def test_decimal_internal_type
       decimal_column = @columns.find { |column| column.type == :decimal }
       assert_equal(:integer, decimal_column.logical_type.internal_type)