Fix NewSpannerLoaderFromDDL to check if table exists (#70)

cloudspannerecosystem · Jul 22, 2021 · 072917f · 072917f
1 parent 41f9497
commit 072917f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/loaders/parser.go b/loaders/parser.go
@@ -51,7 +51,10 @@ func NewSpannerLoaderFromDDL(fpath string) (*SpannerLoaderFromDDL, error) {
 			v.createTable = val
 			tables[val.Name.Name] = v
 		case *ast.CreateIndex:
-			v := tables[val.TableName.Name]
+			v, ok := tables[val.TableName.Name]
+			if !ok {
+				return nil, fmt.Errorf("table '%s' is undefined, but got '%s'", val.TableName.Name, ddl.SQL())
+			}
 			v.createIndexes = append(v.createIndexes, val)
 			tables[val.TableName.Name] = v
 		case *ast.AlterTable: