Visualize container

greeflas · Jan 14, 2024 · 68f80f7 · 68f80f7
1 parent 626a745
commit 68f80f7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/cmd/create_message_command/main.go b/cmd/create_message_command/main.go
@@ -5,6 +5,8 @@ import (
 	"log"
 	"os"
 
+	"go.uber.org/dig"
+
 	"github.com/greeflas/go_di_example/internal/model"
 	"github.com/greeflas/go_di_example/internal/repository"
 	"github.com/greeflas/go_di_example/pkg/di"
@@ -25,6 +27,10 @@ func runCommand(env di.Environment) error {
 		return logger
 	})
 
+	if err := dig.Visualize(c, os.Stdout); err != nil {
+		return err
+	}
+
 	return c.Invoke(func(messageRepository *repository.MessageRepository) error {
 		if len(os.Args) < 2 {
 			return errors.New("pattern arg is required")

diff --git a/cmd/server/main.go b/cmd/server/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"log"
+	"os"
 
 	"github.com/greeflas/go_di_example/internal/handler"
 	"github.com/greeflas/go_di_example/pkg/di"
@@ -19,9 +20,15 @@ func runApp(env di.Environment) error {
 	c := di.BuildContainer(env)
 	serverScope := addAppSpecificDependencies(c)
 
-	return serverScope.Invoke(func(apiServer *server.APIServer) error {
+	if err := dig.Visualize(c, os.Stdout); err != nil {
+		return err
+	}
+
+	err := serverScope.Invoke(func(apiServer *server.APIServer) error {
 		return apiServer.Start()
 	})
+
+	return err
 }
 
 func addAppSpecificDependencies(container *dig.Container) *dig.Scope {