From 0e6f69e23ba0d5e4c748fa05092a2705378ad163 Mon Sep 17 00:00:00 2001
From: ZacJW <zac@zacjw.com>
Date: Mon, 16 Jun 2025 07:54:53 +0100
Subject: [PATCH] update sql_datatype_to_rust_type

---
 rasql-core/src/rust/type_gen.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/rasql-core/src/rust/type_gen.rs b/rasql-core/src/rust/type_gen.rs
index 4d50fe2..9942c7b 100644
--- a/rasql-core/src/rust/type_gen.rs
+++ b/rasql-core/src/rust/type_gen.rs
@@ -88,7 +88,9 @@ impl TypeGenerator<rasql_traits::PostgresTypesTraits> for TokioPostgresGenerator
             {
                 syn::Type::Verbatim(quote::quote! {rust_decimal::Decimal})
             }
-            sqlparser::ast::DataType::Int2(_) => syn::Type::Verbatim(quote::quote! {i16}),
+            sqlparser::ast::DataType::Int2(_) | sqlparser::ast::DataType::SmallInt(_) => {
+                syn::Type::Verbatim(quote::quote! {i16})
+            }
             sqlparser::ast::DataType::UnsignedInt2(_) => syn::Type::Verbatim(quote::quote! {u16}),
             sqlparser::ast::DataType::Int16 => todo!(),
             sqlparser::ast::DataType::Int128 => todo!(),