feat(redis): extend redis client using ioc

Author: lvan100

example/example.go

@@ -26,20 +26,36 @@ import (
 	"github.com/go-spring/spring-core/gs"
 	"github.com/redis/go-redis/v9"
 
-	_ "github.com/go-spring/starter-go-redis"
+	StarterGoRedis "github.com/go-spring/starter-go-redis"
 )
 
 type Service struct {
 	Redis *redis.Client `autowire:""`
 }
 
+// AnotherRedisFactory is a custom implementation of the Factory interface.
+type AnotherRedisFactory struct{}
+
+func (AnotherRedisFactory) CreateClient(c StarterGoRedis.Config) (*redis.Client, error) {
+	return redis.NewClient(&redis.Options{
+		Addr:     c.Addr,
+		Password: c.Password,
+	}), nil
+}
+
 func main() {
 
+	// Register a custom Factory bean to replace the default one.
+	gs.Provide(func() StarterGoRedis.Factory {
+		return &AnotherRedisFactory{}
+	})
+
 	// Here `s` is not referenced by any other object,
 	// so we need to register it as a root object.
 	s := &Service{}
 	gs.Root(gs.Object(s))
 
+	// Define a handler to GET a Redis key value.
 	http.HandleFunc("/get", func(w http.ResponseWriter, r *http.Request) {
 		str, err := s.Redis.Get(r.Context(), "key").Result()
 		if err != nil {
@@ -49,6 +65,7 @@ func main() {
 		_, _ = w.Write([]byte(str))
 	})
 
+	// Define a handler to SET a Redis key value.
 	http.HandleFunc("/set", func(w http.ResponseWriter, r *http.Request) {
 		str, err := s.Redis.Set(r.Context(), "key", "value", 0).Result()
 		if err != nil {

starter.go

@@ -17,24 +17,61 @@
 package StarterGoRedis
 
 import (
+	"github.com/go-spring/spring-core/conf"
 	"github.com/go-spring/spring-core/gs"
 	"github.com/redis/go-redis/v9"
 )
 
+// Config defines Redis connection configuration.
 type Config struct {
 	Addr     string `value:"${addr}"`
 	Password string `value:"${password:=}"`
 }
 
+// Factory defines an interface for creating Redis clients.
+type Factory interface {
+	CreateClient(c Config) (*redis.Client, error)
+}
+
+type DefaultFactory struct{}
+
+// CreateClient creates a new Redis client based on the provided configuration.
+func (DefaultFactory) CreateClient(c Config) (*redis.Client, error) {
+	return redis.NewClient(&redis.Options{
+		Addr:     c.Addr,
+		Password: c.Password,
+	}), nil
+}
+
 func init() {
-	gs.Group("${spring.go-redis}",
-		func(c Config) (*redis.Client, error) { // init
-			return redis.NewClient(&redis.Options{
-				Addr:     c.Addr,
-				Password: c.Password,
-			}), nil
-		},
-		func(client *redis.Client) error { // destroy
-			return client.Close()
-		})
+	const key = "spring.go-redis"
+
+	// Register a module that initializes Redis clients
+	gs.Module([]gs.ConditionOnProperty{
+		gs.OnProperty(key),
+	}, func(p conf.Properties) error {
+
+		// Bind configuration into a map of name -> Config
+		var m map[string]Config
+		if err := p.Bind(&m, "${"+key+"}"); err != nil {
+			return err
+		}
+
+		// Register DefaultFactory as a bean implementing Factory,
+		// but only if no other Factory bean has been provided.
+		gs.Object(&DefaultFactory{}).
+			Condition(gs.OnMissingBean[Factory]()).
+			Export(gs.As[Factory]())
+
+		// For each Redis configuration entry,
+		// create and register a Redis client bean.
+		for name, c := range m {
+			gs.Provide(func(factory Factory) (*redis.Client, error) { // create
+				return factory.CreateClient(c)
+			}).Destroy(func(client *redis.Client) error { // destroy
+				return client.Close()
+			}).Name(name)
+		}
+		return nil
+	})
 }