feat(sqlite): support sqlc.embed function

Author: hakobera

internal/codegen/golang/result.go

@@ -114,13 +114,13 @@ type goEmbed struct {
 
 // look through all the structs and attempt to find a matching one to embed
 // We need the name of the struct and its field names.
-func newGoEmbed(embed *plugin.Identifier, structs []Struct) *goEmbed {
+func newGoEmbed(embed *plugin.Identifier, structs []Struct, defaultSchema string) *goEmbed {
 	if embed == nil {
 		return nil
 	}
 
 	for _, s := range structs {
-		embedSchema := "public"
+		embedSchema := defaultSchema
 		if embed.Schema != "" {
 			embedSchema = embed.Schema
 		}
@@ -280,7 +280,7 @@ func buildQueries(req *plugin.CodeGenRequest, structs []Struct) ([]Query, error)
 					columns = append(columns, goColumn{
 						id:     i,
 						Column: c,
-						embed:  newGoEmbed(c.EmbedTable, structs),
+						embed:  newGoEmbed(c.EmbedTable, structs, req.Catalog.DefaultSchema),
 					})
 				}
 				var err error

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

@@ -1,3 +1,5 @@
+ATTACH 'baz.db' AS baz;
+
 CREATE TABLE users (
     id integer PRIMARY KEY,
     name text NOT NULL,
@@ -9,6 +11,12 @@ CREATE TABLE posts (
     user_id integer NOT NULL
 );
 
+CREATE TABLE baz.users (
+    id integer PRIMARY KEY,
+    name text NOT NULL
+);
+
+
 -- name: Only :one
 SELECT sqlc.embed(users) FROM users;
 
@@ -27,3 +35,10 @@ SELECT sqlc.embed(users), sqlc.embed(users) FROM users;
 -- name: Join :one
 SELECT sqlc.embed(u), sqlc.embed(p) FROM posts AS p
 INNER JOIN users AS u ON p.user_id = u.users.id;
+
+-- name: WithSchema :one
+SELECT sqlc.embed(bu) FROM baz.users AS bu;
+
+-- name: WithCrossSchema :many
+SELECT sqlc.embed(u), sqlc.embed(bu) FROM users AS u
+INNER JOIN baz.users bu ON u.id = bu.id;