add dataset existence validation for uploadContentRequest (#5)

myhau · Jul 12, 2024 · ea0d66a · ea0d66a
1 parent ca5c018
commit ea0d66a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/server/handler.go b/server/handler.go
@@ -403,10 +403,13 @@ func (h *uploadContentHandler) Handle(ctx context.Context, r *uploadContentReque
 	load := r.job.Content().Configuration.Load
 	tableRef := load.DestinationTable
 	dataset := r.project.Dataset(tableRef.DatasetId)
+	if dataset == nil {
+		return fmt.Errorf("dataset `%s` is not found", tableRef.DatasetId)
+	}
 	table := dataset.Table(tableRef.TableId)
 	if table == nil {
 		if load.CreateDisposition == "CREATE_NEVER" {
-			return fmt.Errorf("`%s` is not found", tableRef.TableId)
+			return fmt.Errorf("table `%s` is not found", tableRef.TableId)
 		}
 		if _, err := (&tablesInsertHandler{}).Handle(ctx, &tablesInsertRequest{
 			server:  r.server,