Added function for deleting artifacts to catalog client interface

Extended reservation retrieval to allow querying via artifact tag in catalog client interface

Signed-off-by: Nick Müller <[email protected]>

go/tasks/pluginmachinery/catalog/client.go

@@ -131,6 +131,9 @@ type Client interface {
 	// GetOrExtendReservation tries to retrieve a (valid) reservation for the given key, creating a new one using the
 	// specified owner ID if none was found or updating an existing one if it has expired.
 	GetOrExtendReservation(ctx context.Context, key Key, ownerID string, heartbeatInterval time.Duration) (*datacatalog.Reservation, error)
+	// GetOrExtendReservationByArtifactTag tries to retrieve a (valid) reservation for the given dataset ID and artifact
+	// tag, creating a new one using the specified owner ID if none was found or updating an existing one if it has expired.
+	GetOrExtendReservationByArtifactTag(ctx context.Context, datasetID *datacatalog.DatasetID, artifactTag string, ownerID string, heartbeatInterval time.Duration) (*datacatalog.Reservation, error)
 	// Put stores the given data using the specified key, creating artifact entries as required.
 	// To update an existing artifact, use Update instead.
 	Put(ctx context.Context, key Key, reader io.OutputReader, metadata Metadata) (Status, error)
@@ -139,6 +142,11 @@ type Client interface {
 	Update(ctx context.Context, key Key, reader io.OutputReader, metadata Metadata) (Status, error)
 	// ReleaseReservation releases an acquired reservation for the given key and owner ID.
 	ReleaseReservation(ctx context.Context, key Key, ownerID string) error
+	// Delete removes the artifact associated with the given key and deletes its underlying data from blob storage.
+	Delete(ctx context.Context, key Key) error
+	// DeleteByArtifactTag removes the artifact associated with the given dataset ID and artifact tag and deletes its
+	// underlying data from blob storage.
+	DeleteByArtifactTag(ctx context.Context, datasetID *datacatalog.DatasetID, artifactTag string) error
 }
 
 func IsNotFound(err error) bool {