Fix panic when using load with zq (#5162)

Closes #5157
brimdata · Jul 1, 2024 · 3822b66 · 3822b66
1 parent ab603fa
commit 3822b66
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/compiler/semantic/op.go b/compiler/semantic/op.go
@@ -771,6 +771,10 @@ func (a *analyzer) semOp(o ast.Op, seq dag.Seq) dag.Seq {
 			Exprs: exprs,
 		})
 	case *ast.Load:
+		if !a.source.IsLake() {
+			a.error(o, errors.New("load operator cannot be used without a lake"))
+			return []dag.Op{badOp()}
+		}
 		poolID, err := lakeparse.ParseID(o.Pool)
 		if err != nil {
 			poolID, err = a.source.PoolID(a.ctx, o.Pool)

diff --git a/compiler/ztests/load-no-lake.yaml b/compiler/ztests/load-no-lake.yaml
@@ -0,0 +1,3 @@
+zed: "load test"
+
+errorRE: "load operator cannot be used without a lake"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		zed: "load test"

		errorRE: "load operator cannot be used without a lake"