75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gis/internal/app"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var reprocessAll bool
|
|
|
|
var reprocessCmd = &cobra.Command{
|
|
Use: "reprocess [dataset-id]",
|
|
Short: "Re-enqueue the processing job for uploaded datasets",
|
|
Long: "Re-publish the RabbitMQ message that drives an uploaded dataset's\n" +
|
|
"asynchronous processing, selecting the right step from its file type\n" +
|
|
"(vector_with_kato -> parse, vector -> properties, raster -> cog).\n\n" +
|
|
"Pass a dataset id to reprocess a single dataset, or --all to reprocess\n" +
|
|
"every dataset.\n\n" +
|
|
"Examples:\n" +
|
|
" gis reprocess 06818b2b-1fc5-47d9-a764-db2d4cb3df75\n" +
|
|
" gis reprocess --all",
|
|
Args: cobra.MaximumNArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if reprocessAll == (len(args) == 1) {
|
|
return fmt.Errorf("provide exactly one of <dataset-id> or --all")
|
|
}
|
|
|
|
ctx, cancel := signalContext()
|
|
defer cancel()
|
|
|
|
application, err := app.New(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer application.Close()
|
|
|
|
if reprocessAll {
|
|
enqueued, failures, err := application.ReprocessAllDatasets(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for id, ferr := range failures {
|
|
application.Log.Error("failed to re-enqueue dataset processing",
|
|
"dataset_id", id, "error", ferr)
|
|
}
|
|
application.Log.Info("re-enqueued all dataset processing",
|
|
"enqueued", enqueued, "failed", len(failures))
|
|
if len(failures) > 0 {
|
|
return fmt.Errorf("%d of %d datasets failed to re-enqueue", len(failures), enqueued+len(failures))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
id, err := uuid.Parse(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("invalid dataset id %q: %w", args[0], err)
|
|
}
|
|
|
|
dataset, err := application.ReprocessDataset(ctx, id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
application.Log.Info("re-enqueued dataset processing",
|
|
"dataset_id", dataset.ID, "file_type", dataset.FileType, "status", dataset.Status)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
reprocessCmd.Flags().BoolVar(&reprocessAll, "all", false, "reprocess every dataset")
|
|
}
|