chore: add redis infra

Author: sysulq

examples/redis/kod.toml

@@ -0,0 +1,2 @@
+["github.com/go-kod/kod/examples/redis/Component"]
+redisConfig = { addr = "localhost:6379" }

examples/redis/kod_gen.go

@@ -0,0 +1,29 @@
+package redis
+
+import (
+	"context"
+
+	"github.com/go-kod/kod"
+	"github.com/go-kod/kod/ext/client/kredis"
+	"github.com/redis/go-redis/v9"
+)
+
+type impl struct {
+	kod.Implements[Component]
+	kod.WithConfig[config]
+
+	cc *redis.Client
+}
+
+type config struct {
+	RedisConfig kredis.Config
+}
+
+func (ins *impl) Init(ctx context.Context) error {
+	ins.cc = ins.Config().RedisConfig.Build()
+	return nil
+}
+
+func (ins *impl) Client() *redis.Client {
+	return ins.cc
+}

examples/redis/kod_gen_interface.go

@@ -0,0 +1,2 @@
+["github.com/go-kod/kod/examples/redis/Component"]
+redisConfig = { addr = "localhost:6379" }

examples/redis/kod_gen_mock.go

@@ -5,20 +5,23 @@ import (
 	"time"
 
 	"github.com/go-kod/kod"
+	"github.com/go-kod/kod/examples/redis"
 	"github.com/sony/sonyflake"
 )
 
 type impl struct {
 	kod.Implements[Component]
+	redis kod.Ref[redis.Component]
+
 	snowflake *sonyflake.Sonyflake
 }
 
 func (ins *impl) Init(ctx context.Context) error {
 	ins.snowflake = sonyflake.NewSonyflake(sonyflake.Settings{
 		StartTime: time.Now(),
 		MachineID: func() (uint16, error) {
-			// TODO use redis to get machine id
-			return 1, nil
+			machineId, err := ins.redis.Get().Client().Incr(ctx, "snowflake:id").Uint64()
+			return uint16(machineId), err
 		},
 	})
 

examples/redis/redis.go

@@ -0,0 +1,17 @@
+package snowflake
+
+import (
+	"context"
+	"testing"
+
+	"github.com/go-kod/kod"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSnowflake(t *testing.T) {
+	kod.RunTest(t, func(ctx context.Context, c Component) {
+		nextIDRes, err := c.NextID(ctx, &NextIDRequest{})
+		assert.Nil(t, err)
+		assert.NotZero(t, nextIDRes.ID)
+	})
+}

examples/snowflake/kod.toml

@@ -1,2 +1,2 @@
-["github.com/go-kod/kod/examples/user/Component"]
+["github.com/go-kod/kod/examples/redis/Component"]
 redisConfig = { addr = "localhost:6379" }

examples/snowflake/kod_gen.go

@@ -7,6 +7,7 @@ import (
 	"time"
 
 	"github.com/go-kod/kod"
+	redisC "github.com/go-kod/kod/examples/redis"
 	"github.com/go-kod/kod/examples/snowflake"
 	"github.com/go-kod/kod/ext/client/kredis"
 	"github.com/go-kod/kod/interceptor/kmetric"
@@ -21,6 +22,7 @@ type impl struct {
 	kod.Implements[Component]
 	kod.WithConfig[config]
 	snowflake kod.Ref[snowflake.Component]
+	redisComp kod.Ref[redisC.Component]
 
 	redis *redis.Client
 }
@@ -37,7 +39,7 @@ type claims struct {
 }
 
 func (ins *impl) Init(ctx context.Context) error {
-	ins.redis = ins.Config().Redis.Build()
+	ins.redis = ins.redisComp.Get().Client()
 
 	if ins.Config().SecretKey == "" {
 		ins.Config().SecretKey = "my-secret"