Merge pull request #43 from tinh-tinh/feat/ren/42-add-for-root-module

feat: add for roor module to store common config

Author: ren0503

module.go

@@ -8,6 +8,20 @@ import (
 
 const QUEUE core.Provide = "QUEUE"
 
+func ForRoot(opt *Options) core.Modules {
+	return func(module core.Module) core.Module {
+		queueModule := module.New(core.NewModuleOptions{})
+
+		queueModule.NewProvider(core.ProviderOptions{
+			Name:  QUEUE,
+			Value: opt,
+		})
+		queueModule.Export(QUEUE)
+
+		return queueModule
+	}
+}
+
 // getQueueName generates a unique name for a queue provider.
 //
 // The name is in the form "<name>Queue".
@@ -18,13 +32,24 @@ func getQueueName(name string) core.Provide {
 // Register registers a new queue module with the given name and options. The
 // registered module creates a new queue with the given name and options, and
 // exports the queue under the name "<name>Queue".
-func Register(name string, opt *Options) core.Modules {
+func Register(name string, opts ...*Options) core.Modules {
+	var option *Options
+	if len(opts) > 0 {
+		option = opts[0]
+	}
 	return func(module core.Module) core.Module {
+		if option == nil {
+			defaultOptions, ok := module.Ref(QUEUE).(*Options)
+			if !ok || defaultOptions == nil {
+				panic("not config option for queue")
+			}
+			option = defaultOptions
+		}
 		queueModule := module.New(core.NewModuleOptions{})
 
 		queueModule.NewProvider(core.ProviderOptions{
 			Name:  getQueueName(name),
-			Value: New(name, opt),
+			Value: New(name, option),
 		})
 		queueModule.Export(getQueueName(name))
 

module_test.go

@@ -28,3 +28,46 @@ func Test_Module(t *testing.T) {
 	queue := queue.Inject(module, "jobs")
 	require.NotNil(t, queue)
 }
+
+func Test_Panic(t *testing.T) {
+	require.Panics(t, func() {
+		module := core.NewModule(core.NewModuleOptions{
+			Imports: []core.Modules{
+				queue.Register("jobs"),
+			},
+		})
+
+		require.NotNil(t, module)
+	})
+}
+
+func Test_DefaultOptions(t *testing.T) {
+	module := core.NewModule(core.NewModuleOptions{
+		Imports: []core.Modules{
+			queue.ForRoot(&queue.Options{
+				Connect: &redis.Options{
+					Addr:     "localhost:6379",
+					DB:       0,
+					Password: "",
+				},
+				Workers:       6,
+				RetryFailures: 3,
+			}),
+			queue.Register("video"),
+			queue.Register("media", &queue.Options{
+				Connect: &redis.Options{
+					Addr:     "localhost:6379",
+					DB:       0,
+					Password: "",
+				},
+				Workers: 3,
+			}),
+		},
+	})
+
+	videoQueue := queue.Inject(module, "video")
+	require.NotNil(t, videoQueue)
+
+	mediaQueue := queue.Inject(module, "media")
+	require.NotNil(t, mediaQueue)
+}