fix: missing imports (sqlc-dev#1637)

Author: jlisthood

internal/codegen/golang/imports.go

@@ -309,7 +309,8 @@ func (i *importer) queryImports(filename string) fileImports {
 						}
 					}
 				}
-				if strings.HasPrefix(q.Ret.Type(), name) {
+				retType := strings.TrimPrefix(q.Ret.Type(), "[]")
+				if strings.HasPrefix(retType, name) {
 					return true
 				}
 			}
@@ -322,7 +323,8 @@ func (i *importer) queryImports(filename string) fileImports {
 						}
 					}
 				}
-				if strings.HasPrefix(q.Arg.Type(), name) {
+				argType := strings.TrimPrefix(q.Arg.Type(), "[]")
+				if strings.HasPrefix(argType, name) {
 					return true
 				}
 			}
@@ -414,7 +416,8 @@ func (i *importer) batchImports(filename string) fileImports {
 						}
 					}
 				}
-				if strings.HasPrefix(q.Ret.Type(), name) {
+				retType := strings.TrimPrefix(q.Ret.Type(), "[]")
+				if strings.HasPrefix(retType, name) {
 					return true
 				}
 			}
@@ -427,7 +430,8 @@ func (i *importer) batchImports(filename string) fileImports {
 						}
 					}
 				}
-				if strings.HasPrefix(q.Arg.Type(), name) {
+				argType := strings.TrimPrefix(q.Arg.Type(), "[]")
+				if strings.HasPrefix(argType, name) {
 					return true
 				}
 			}