Use DefaultDataset in Job

goccy · Dec 1, 2024 · f636484 · f636484
1 parent e411226
commit f636484
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/server/handler.go b/server/handler.go
@@ -1400,13 +1400,17 @@ func (h *jobsInsertHandler) Handle(ctx context.Context, r *jobsInsertRequest) (*
 		return nil, fmt.Errorf("failed to start transaction: %w", err)
 	}
 	defer tx.RollbackIfNotCommitted()
+	datasetId := ""
+	if job.Configuration.Query.DefaultDataset != nil {
+		datasetId = job.Configuration.Query.DefaultDataset.DatasetId
+	}
 	hasDestinationTable := job.Configuration.Query.DestinationTable != nil
 	startTime := time.Now()
 	response, jobErr := r.server.contentRepo.Query(
 		ctx,
 		tx,
 		r.project.ID,
-		"",
+		datasetId,
 		job.Configuration.Query.Query,
 		job.Configuration.Query.QueryParameters,
 	)