Additional catalog client functionality for cache eviction

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,17 @@ 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
+	// ReleaseReservationByArtifactTag releases an acquired reservation for the given dataset ID, artifact tag and
+	// owner ID.
+	ReleaseReservationByArtifactTag(ctx context.Context, datasetID *datacatalog.DatasetID, artifactTag string, 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
+	// DeleteByArtifactID removes the artifact associated with the given dataset and artifact ID and deletes its
+	// underlying data from blob storage.
+	DeleteByArtifactID(ctx context.Context, datasetID *datacatalog.DatasetID, artifactID string) error
 }
 
 func IsNotFound(err error) bool {