Repository content update

Signed-off-by: Valery Piashchynski <[email protected]>

1 files changed, 21 insertions, 0 deletions

diff --git a/internal/container/container.go b/internal/container/container.go
new file mode 100644
index 00000000..aa767b2e
--- /dev/null
+++ b/internal/container/container.go
@@ -0,0 +1,21 @@
+package container
+
+import (
+	endure "github.com/spiral/endure/pkg/container"
+)
+
+// NewContainer creates endure container with all required options (based on container Config). Logger is nil by
+// default.
+func NewContainer(cfg Config) (*endure.Endure, error) {
+	endureOptions := []endure.Options{
+		endure.SetLogLevel(cfg.LogLevel),
+		endure.RetryOnFail(cfg.RetryOnFail),
+		endure.GracefulShutdownTimeout(cfg.GracePeriod),
+	}
+
+	if cfg.PrintGraph {
+		endureOptions = append(endureOptions, endure.Visualize(endure.StdOut, ""))
+	}
+
+	return endure.NewContainer(nil, endureOptions...)
+}