Use dependency injection such as dig for commands

[YuriyChmil]

Hello! I am using the dig package as a dependency injection tool for my project.
I have one command with has some dependency:

package cmd

import (
	"github.com/test/event-analytic/db"
	"github.com/test/event-analytic/db/commands"
	"github.com/test/event-analytic/di"
	"github.com/spf13/cobra"
)
var serveCmd = &cobra.Command{
	Use:   "migrate",
	Short: "Run migrations for clickhouse database",
	Run: func(cmd *cobra.Command, args []string) {
		container := di.BuildContainer()
		container.Invoke(func(client *db.ClickHouseClient) {
			commands.Migrate(client)
		})
	},
}

func init() {
	rootCmd.AddCommand(serveCmd)
}

All works fine, but in case if I need create more commands I need to repeat this stuff:

container := di.BuildContainer()
		container.Invoke(func(client *db.ClickHouseClient) {
			newCommand
		})

every time.
Are there any possibility to pass container once in main file, something like this:

package main

import (
	"github.com/joho/godotenv"
	"log"
	"os"
)

func main() {
	err := godotenv.Load(os.Getenv("ENV_FILE"))
	if err != nil {
		log.Fatal("Error loading .env file")
	}
        container := di.BuildContainer()
	cmd.Execute(container)
}

And then in custom command use something like this:

var serveCmd = &cobra.Command{
	Use:   "migrate",
	Short: "Run migrations for clickhouse database",
	Run: func(cmd *cobra.Command, args []string, container ) {
		container.Invoke(func(client *db.ClickHouseClient) {
			commands.Migrate(client)
		})
	},
}

[github-actions]

This issue is being marked as stale due to a long period of inactivity

[johnSchnake]

I think the pattern you're looking for is something like below. You don't define the commands directly but create functions that will create closures around that variable:

package main

import (
	"github.com/joho/godotenv"
	"log"
	"os"
)

func main() {
	err := godotenv.Load(os.Getenv("ENV_FILE"))
	if err != nil {
		log.Fatal("Error loading .env file")
	}
        container := di.BuildContainer()
        ... 
        rootCmd.AddCommand(newServeCmd(container))
        ...
}

func newServeCmd(c Container) *cobra.Command{
  return &cobra.Command{
	Use:   "migrate",
	Short: "Run migrations for clickhouse database",
	Run: func(cmd *cobra.Command, args []string, container ) {
		c.Invoke(func(client *db.ClickHouseClient) {
			commands.Migrate(client)
		})
	},
}

Closing as part of #1600 but feel free to reopen or discuss if you still need any assistance.