added support for table specific column exclusion. #38

marcominerva · Jan 24, 2024 · 6d6339e · 6d6339e
1 parent aba1053
commit 6d6339e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/DatabaseGpt.Npgsql/NpgsqlDatabaseGptProvider.cs b/src/DatabaseGpt.Npgsql/NpgsqlDatabaseGptProvider.cs
@@ -61,10 +61,13 @@ public async Task<string> GetCreateTablesScriptAsync(IEnumerable<string> tables,
                     CASE WHEN CHARACTER_MAXIMUM_LENGTH = -1 THEN 'MAX' ELSE CAST(CHARACTER_MAXIMUM_LENGTH AS VARCHAR(10)) END || ')','') || ' ' || 
                     CASE WHEN IS_NULLABLE = 'YES' THEN 'NULL' ELSE 'NOT NULL' END
                 FROM INFORMATION_SCHEMA.COLUMNS
-                WHERE TABLE_SCHEMA = @schema AND TABLE_NAME = @table AND COLUMN_NAME NOT IN (SELECT UNNEST(@excludedColumns));
+                WHERE TABLE_SCHEMA = @schema
+                    AND TABLE_NAME = @table
+                    AND COLUMN_NAME NOT IN (SELECT UNNEST(@excludedColumns))
+                    AND TABLE_SCHEMA || '.' || TABLE_NAME || '.' || COLUMN_NAME NOT IN (SELECT UNNEST(@excludedColumns));
                 """;
 
-            var columns = await connection.QueryAsync<string>(query, new { schema = table.Schema, table = table.Name, ExcludedColumns = excludedColumns });
+            var columns = await connection.QueryAsync<string>(query, new { schema = table.Schema, table = table.Name, excludedColumns });
             result.AppendLine($"CREATE TABLE [{table.Schema}].[{table.Name}] ({string.Join(", ", columns)});");
         }