fix: missing imports (sqlc-dev#1637)

Author: jlisthood

internal/codegen/golang/imports.go

@@ -310,7 +310,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
 				}
 			}
@@ -323,7 +324,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
 				}
 			}
@@ -418,7 +420,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
 				}
 			}
@@ -431,7 +434,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
 				}
 			}