support join select

main.go

@@ -40,9 +40,7 @@ func (s *SQLVet) reportError(format string, a ...interface{}) {
 // Vet performs static analysis
 func (s *SQLVet) Vet() {
 	queries, err := vet.CheckDir(
-		vet.VetContext{
-			Schema: s.Schema,
-		},
+		vet.NewContext(s.Schema.Tables),
 		s.ProjectRoot,
 		s.Cfg.BuildFlags,
 		s.Cfg.SqlFuncMatchers,

pkg/schema/postgres.go

@@ -91,51 +91,50 @@ func parsePostgresSchema(schemaInput string) (map[string]Table, error) {
 					continue
 				}
 
-				if resTarget.Name != "" {
-					table.Columns[resTarget.Name] = Column{
-						Name: resTarget.Name,
-					}
-					continue
+				if col, ok := GetResTargetColumn(resTarget); ok {
+					table.Columns[col.Name] = col
 				}
+			}
 
-				if resTarget.Val == nil {
-					continue
-				}
+			tables[tableName] = table
+		}
+	}
 
-				colRef := resTarget.Val.GetColumnRef()
-				if colRef == nil {
-					// parse only column references when no alias is provided
-					continue
-				}
+	return tables, nil
+}
 
-				var colField *pg_query.Node
-				if len(colRef.Fields) > 0 {
-					colField = colRef.Fields[len(colRef.Fields)-1]
-				}
+func GetResTargetColumn(resTarget *pg_query.ResTarget) (col Column, ok bool) {
+	if resTarget.Name != "" {
+		return Column{Name: resTarget.Name}, true
+	}
 
-				if colField == nil {
-					continue
-				}
+	if resTarget.Val == nil {
+		return
+	}
 
-				if colField.GetAStar() != nil {
-					// SELECT * - force parsing explicit columns for simplicity
-					continue
-				}
+	colRef := resTarget.Val.GetColumnRef()
+	if colRef == nil {
+		// parse only column references when no alias is provided
+		return
+	}
 
-				if colField.GetString_() == nil {
-					continue
-				}
+	var colField *pg_query.Node
+	if len(colRef.Fields) > 0 {
+		colField = colRef.Fields[len(colRef.Fields)-1]
+	}
 
-				colName := colField.GetString_().GetStr()
-				table.Columns[colName] = Column{
-					Name: colName,
-					Type: "", // type not set, never used for validation
-				}
-			}
+	if colField == nil {
+		return
+	}
 
-			tables[tableName] = table
-		}
+	if colField.GetAStar() != nil {
+		// SELECT * - force parsing explicit columns for simplicity
+		return
 	}
 
-	return tables, nil
+	if colField.GetString_() == nil {
+		return
+	}
+
+	return Column{Name: colField.GetString_().GetStr()}, true
 }

pkg/vet/gosource.go

@@ -142,7 +142,7 @@ func handleQuery(ctx VetContext, qs *QuerySite) {
 	}
 
 	var queryParams []QueryParam
-	queryParams, qs.Err = ValidateSqlQuery(ctx, qs.Query)
+	queryParams, qs.Err = ValidateSqlQuery(NewContext(ctx.Schema.Tables), qs.Query)
 
 	if qs.Err != nil {
 		return