fix: allow passing a asyncpg.Pool as a ConnectionLike

davfsa · davfsa · commit e1649896e6e5 · 2025-06-17T11:37:50.000+02:00
Signed-off-by: davfsa &lt;davfsa@gmail.com&gt;
diff --git a/internal/codegen/drivers/asyncpg.go b/internal/codegen/drivers/asyncpg.go
@@ -16,7 +16,7 @@ const AsyncpgResult = "asyncpg.Record"
 func AsyncpgTypeCheckingHook() []string {
 	return []string{
 		fmt.Sprintf(
-			"ConnectionLike: typing.TypeAlias = asyncpg.Connection[%[1]s] | asyncpg.pool.PoolConnectionProxy[%[1]s]",
+			"ConnectionLike: typing.TypeAlias = asyncpg.Connection[%[1]s] | asyncpg.Pool[%[1]s] | asyncpg.pool.PoolConnectionProxy[%[1]s]",
 			AsyncpgResult,
 		),
 	}

Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ const AsyncpgResult = "asyncpg.Record"`
`16`	`16`	`func AsyncpgTypeCheckingHook() []string {`
`17`	`17`	`return []string{`
`18`	`18`	`fmt.Sprintf(`
`19`		`- "ConnectionLike: typing.TypeAlias = asyncpg.Connection[%[1]s] \| asyncpg.pool.PoolConnectionProxy[%[1]s]",`
	`19`	`+ "ConnectionLike: typing.TypeAlias = asyncpg.Connection[%[1]s] \| asyncpg.Pool[%[1]s] \| asyncpg.pool.PoolConnectionProxy[%[1]s]",`
`20`	`20`	`AsyncpgResult,`
`21`	`21`	`),`
`22`	`22`	`}`