diff --git a/app/cli/cmd/artifact_download.go b/app/cli/cmd/artifact_download.go index c1392e305..8f487c32c 100644 --- a/app/cli/cmd/artifact_download.go +++ b/app/cli/cmd/artifact_download.go @@ -49,6 +49,13 @@ func newArtifactDownloadCmd() *cobra.Command { return action.NewArtifactDownload(opts).Run(downloadPath, digest) }, + PostRunE: func(cmd *cobra.Command, args []string) error { + if artifactCASConn != nil { + return artifactCASConn.Close() + } + + return nil + }, } cmd.Flags().StringVarP(&digest, "digest", "d", "", "digest of the file to download") diff --git a/app/cli/cmd/artifact_upload.go b/app/cli/cmd/artifact_upload.go index b87ba5475..26181b00d 100644 --- a/app/cli/cmd/artifact_upload.go +++ b/app/cli/cmd/artifact_upload.go @@ -49,6 +49,13 @@ func newArtifactUploadCmd() *cobra.Command { _, err := action.NewArtifactUpload(opts).Run(filePath) return err }, + PostRunE: func(cmd *cobra.Command, args []string) error { + if artifactCASConn != nil { + return artifactCASConn.Close() + } + + return nil + }, } cmd.Flags().StringVarP(&filePath, "file", "f", "", "path to file to upload") diff --git a/app/cli/cmd/attestation_add.go b/app/cli/cmd/attestation_add.go index 3ddd4a8d3..1b6288386 100644 --- a/app/cli/cmd/attestation_add.go +++ b/app/cli/cmd/attestation_add.go @@ -72,6 +72,13 @@ func newAttestationAddCmd() *cobra.Command { logger.Info().Msg("material added to attestation") + return nil + }, + PostRunE: func(cmd *cobra.Command, args []string) error { + if artifactCASConn != nil { + return artifactCASConn.Close() + } + return nil }, }