sqlite: fixed sqlite column type override (#1986)

We should not assume that a table has a default schema name in the override config.
Because different sql databases can have different default schemas.

Fixes: #1985

Author: mcdoker18

internal/cmd/shim.go

@@ -19,7 +19,6 @@ func pluginOverride(o config.Override) *plugin.Override {
 		colParts := strings.Split(o.Column, ".")
 		switch len(colParts) {
 		case 2:
-			table.Schema = "public"
 			table.Name = colParts[0]
 			column = colParts[1]
 		case 3:

internal/codegen/sdk/sdk.go

@@ -32,7 +32,7 @@ func Matches(o *plugin.Override, n *plugin.Identifier, defaultSchema string) boo
 	if o.Table.Catalog != "" && !MatchString(o.Table.Catalog, n.Catalog) {
 		return false
 	}
-	if o.Table.Schema == "" && schema != "" {
+	if o.Table.Schema == "" && n.Schema != "" {
 		return false
 	}
 	if o.Table.Schema != "" && !MatchString(o.Table.Schema, schema) {

internal/endtoend/testdata/overrides/sqlite/go/db.go

@@ -0,0 +1 @@
+SELECT 1;

internal/endtoend/testdata/overrides/sqlite/go/models.go

@@ -0,0 +1,5 @@
+CREATE TABLE foo (
+    other     text NOT NULL,
+    total     bigint NOT NULL,
+    retyped   text NOT NULL
+);

internal/endtoend/testdata/overrides/sqlite/query.sql

@@ -0,0 +1,18 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "name": "override",
+      "engine": "sqlite",
+      "schema": "schema.sql",
+      "queries": "query.sql",
+      "overrides": [
+        {
+          "go_type": "github.com/kyleconroy/sqlc-testdata/pkg.CustomType",
+          "column": "foo.retyped"
+        }
+      ]
+    }
+  ]
+}

internal/endtoend/testdata/overrides/sqlite/schema.sql

@@ -0,0 +1 @@
+SELECT 1;

internal/endtoend/testdata/overrides/sqlite/sqlc.json

@@ -0,0 +1,14 @@
+CREATE TABLE foo (
+    other     text NOT NULL,
+    tagged    text NOT NULL
+);
+
+CREATE TABLE bar (
+    other          text NOT NULL,
+    also_tagged    text NOT NULL
+);
+
+CREATE TABLE baz (
+    other          text NOT NULL,
+    also_tagged    text NOT NULL
+);

internal/endtoend/testdata/overrides_go_struct_tags/sqlite/go/db.go

@@ -0,0 +1,22 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "name": "override",
+      "engine": "sqlite",
+      "schema": "schema.sql",
+      "queries": "query.sql",
+      "overrides": [
+        {
+          "go_struct_tag": "a:\"b\" x:\"y,z\"",
+          "column": "foo.tagged"
+        },
+        {
+          "go_struct_tag": "also:\"tagged\"",
+          "column": "*.also_tagged"
+        }
+      ]
+    }
+  ]
+}

internal/endtoend/testdata/overrides_go_struct_tags/sqlite/go/models.go

@@ -0,0 +1 @@
+SELECT 1;

internal/endtoend/testdata/overrides_go_struct_tags/sqlite/query.sql

@@ -0,0 +1,17 @@
+CREATE TABLE foo (
+    other     text NOT NULL,
+    total     bigint NOT NULL,
+    retyped   text NOT NULL
+);
+
+CREATE TABLE bar (
+    other          text NOT NULL,
+    total          bigint NOT NULL,
+    also_retyped   text NOT NULL
+);
+
+CREATE TABLE baz (
+    other          text NOT NULL,
+    total          bigint NOT NULL,
+    also_retyped   text NOT NULL
+);

internal/endtoend/testdata/overrides_go_struct_tags/sqlite/schema.sql

@@ -0,0 +1,22 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "name": "override",
+      "engine": "sqlite",
+      "schema": "schema.sql",
+      "queries": "query.sql",
+      "overrides": [
+        {
+          "go_type": "github.com/kyleconroy/sqlc-testdata/pkg.CustomType",
+          "column": "foo.retyped"
+        },
+        {
+          "go_type": "github.com/kyleconroy/sqlc-testdata/pkg.CustomType",
+          "column": "*.also_retyped"
+        }
+      ]
+    }
+  ]
+}