wip! tvp columns

Author: Simon Thörnqvist

lib/tds/column.ex

@@ -0,0 +1,9 @@
+defmodule Tds.Column do
+  @type t :: %__MODULE__{
+    name: String.t | nil,
+    type: Atom | nil,
+    opts: Keyword.t
+  }
+
+  defstruct [name: "", type: nil, opts: []]
+end

lib/tds/types.ex

@@ -4,6 +4,7 @@ defmodule Tds.Types do
   use Bitwise
 
   alias Tds.Parameter
+  alias Tds.Column
   alias Tds.DateTime
   alias Tds.DateTime2
 
@@ -987,8 +988,8 @@ defmodule Tds.Types do
 
   def encode_data(@tds_data_type_tvp, %{columns: columns, rows: rows}, _attrs) do
     column_length = <<length(columns) :: little-unsigned-16>>
-    {column_attrs, column_meta} = Enum.reduce(columns, {[], <<>>}, fn (%Parameter{} = param, {attrs, acc_bin}) ->
-      {bin_type, data, attr} = encode_data_type(param)
+    {column_attrs, column_meta} = Enum.reduce(columns, {[], <<>>}, fn (%Column{} = param, {attrs, acc_bin}) ->
+      {bin_type, data, attr} = encode_column_type(param)
       bin = acc_bin <> <<0x00 :: little-unsigned-32, 0x00 :: little-unsigned-16 >> <> data <> <<0x00>>
 
       {[{bin_type, attr} | attrs], bin}
@@ -1008,9 +1009,66 @@ defmodule Tds.Types do
     column_length <> column_meta <> <<0x00>> <> row_data <> <<0x00>>
   end
 
+  def encode_column_type(%Column{type: type} = col) when type != nil do
+    case type do
+      :varchar -> encode_bigvarchar_col_type(col)
+      :boolean -> encode_binary_type(%Parameter{type: :boolean, value: nil})
+      :varbinary -> encode_varbinary_col_type(col)
+      :int -> encode_integer_type(%Parameter{type: :integer, value: nil})
+      :decimal -> encode_decimal_type(%Parameter{type: :decimal, value: nil})
+      :float -> encode_float_type(%Parameter{type: :float, value: nil})
+      :datetime -> encode_datetime_type(%Parameter{type: :datetime, value: nil})
+      :smalldatetime -> encode_smalldatetime_type(%Parameter{type: :smalldatetime, value: nil})
+      :datetime2 -> encode_datetime2_type(%Parameter{type: :datetime2, value: nil})
+      :datetimeoffset -> encode_datetimeoffset_type(%Parameter{type: :datetimeoffset, value: nil})
+      :date -> encode_date_type(%Parameter{type: :datetimeoffset, value: nil})
+      :time -> encode_time_type(%Parameter{type: :time, value: nil})
+      :uuid -> encode_uuid_type(%Parameter{type: :uuid, value: nil})
+    end
+  end
+
+  def encode_bigvarchar_col_type(%Column{opts: opts} = col) do
+    type = @tds_data_type_bigvarchar
+    length = Keyword.get(opts, :length, 0)
+    collation = <<0x00, 0x00, 0x00, 0x00, 0x00>>
+    bin_length = if length <= 8000, do: <<8000::little-unsigned-16>>, else: <<0xFF, 0xFF>>
+    data = <<type>> <> bin_length <> collation
+
+    {type, data, length: length}
+  end
+
+  def encode_varbinary_col_type(%Column{opts: opts} = col) do
+    type = @tds_data_type_bigvarbinary
+    length = Keyword.get(opts, :length, 0)
+    bin_length = if length <= 8000, do: <<8000::little-unsigned-16>>, else: <<0xFF, 0xFF>>
+    data = <<type>> <> bin_length
+
+    {type, data, length: length}
+  end
+
   @doc """
   Data Encoding String Types
   """
+
+  def encode_data(@tds_data_type_bigvarchar, nil, opts) do
+    length = Keyword.get(opts, :length, 0)
+    if length <= 8000,
+      do: <<65535::little-unsigned-16>>,
+      else: <<@tds_plp_null::little-unsigned-64>>
+  end
+
+  def encode_data(@tds_data_type_bigvarchar, value, opts) do
+    value_size = byte_size(value)
+    cond do
+      value_size <= 0 ->
+        <<0x00::unsigned-64, 0x00::unsigned-32>>
+      value_size > 8000 ->
+        encode_plp(value)
+      true ->
+        <<value_size::little-size(2)-unit(8)>> <> value
+    end
+  end
+
   def encode_data(@tds_data_type_nvarchar, nil, _),
     do: <<@tds_plp_null::little-unsigned-64>>
   def encode_data(@tds_data_type_nvarchar, value, _) do

test/tvp_test.exs

@@ -3,6 +3,7 @@ defmodule TvpTest do
   require Logger
   use ExUnit.Case, async: true
   alias Tds.Parameter
+  alias Tds.Column
 
   @tag timeout: 50000
 
@@ -17,20 +18,38 @@ defmodule TvpTest do
     assert :ok = query("BEGIN TRY DROP PROCEDURE __tvpTest DROP TYPE TvpTestType END TRY BEGIN CATCH END CATCH", [])
     assert :ok = query("""
       CREATE TYPE TvpTestType AS TABLE (
-        d int
+        a int,
+        b uniqueidentifier,
+        c varchar(100),
+        d varbinary(max)
       );
     """, [])
+
     assert :ok = query("""
     CREATE PROCEDURE __tvpTest (@tvp TvpTestType readonly)
       AS BEGIN
         select * from @tvp
       END
     """, [])
 
+    rows = [1, <<158, 3, 157, 56, 133, 56, 73, 67, 128, 121, 126, 204, 115, 227, 162, 157>>, "foo", "{\"foo\":\"bar\",\"baz\":\"biz\"}"]
     params = [
-      %Parameter{name: "@tvp", value: %{name: "TvpTestType", columns: [%Parameter{name: "d", type: :integer, value: nil}], rows: [[1]]}, type: :tvp}
+      %Parameter{
+        name: "@tvp",
+        value: %{
+          name: "TvpTestType",
+          columns: [
+            %Column{name: "a", type: :int},
+            %Column{name: "b", type: :uuid},
+            %Column{name: "c", type: :varchar},
+            %Column{name: "d", type: :varbinary},
+          ],
+          rows: [rows]
+        },
+        type: :tvp
+      }
     ]
 
-    assert [[1]] = proc("__tvpTest", params)
+    assert [^rows] = proc("__tvpTest", params)
   end
 end