refactor(build/registry): add support for context with options

Signed-off-by: Massimiliano Giovagnoli <[email protected]>
falcosecurity · Jun 23, 2023 · 48c224f · 48c224f
1 parent 3738986
commit 48c224f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/build/registry/cmd/registry/registry.go b/build/registry/cmd/registry/registry.go
@@ -43,6 +43,7 @@ func main() {
  defer out.Flush()
 
  opts := options.NewCommonOptions(
+ options.WithContext(context.Background()),
  options.WithOutput(out),
  )
 
@@ -86,7 +87,7 @@ func main() {
  Args: cobra.ExactArgs(1),
  DisableFlagsInUseLine: true,
  RunE: func(c *cobra.Command, args []string) error {
- status, err := oci.DoUpdateOCIRegistry(context.Background(), args[0])
+ status, err := oci.DoUpdateOCIRegistry(opts.Context, args[0])
  if err != nil {
  return err
  }

diff --git a/build/registry/internal/options/options.go b/build/registry/internal/options/options.go
@@ -17,11 +17,13 @@ limitations under the License.
 package options
 
 import (
+ "context"
  "io"
 )
 
 type CommonOptions struct {
- Output io.Writer
+ Output io.Writer
+ Context context.Context
 }
 
 type CommonOption func(opts *CommonOptions)
@@ -41,3 +43,9 @@ func WithOutput(out io.Writer) CommonOption {
  opts.Output = out
  }
 }
+
+func WithContext(ctx context.Context) CommonOption {
+ return func(opts *CommonOptions) {
+ opts.Context = ctx
+ }
+}