add: anowflake email kafka, refa: redis connectg

app/email/api/email.go

@@ -0,0 +1,34 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package main
+
+import (
+	"flag"
+	"fmt"
+
+	"juwan-backend/app/email/api/internal/config"
+	"juwan-backend/app/email/api/internal/handler"
+	"juwan-backend/app/email/api/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/conf"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+var configFile = flag.String("f", "etc/email-api.yaml", "the config file")
+
+func main() {
+	flag.Parse()
+
+	var c config.Config
+	conf.MustLoad(*configFile, &c)
+
+	server := rest.MustNewServer(c.RestConf)
+	defer server.Stop()
+
+	ctx := svc.NewServiceContext(c)
+	handler.RegisterHandlers(server, ctx)
+
+	fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
+	server.Start()
+}

app/email/api/etc/email-api.yaml

@@ -0,0 +1,19 @@
+Name: email-api
+Host: 0.0.0.0
+Port: 8888
+
+CacheConf:
+	- Host: "${REDIS_M_HOST}"
+		Type: node
+		Pass: "${REDIS_PASSWORD}"
+		User: "default"
+	- Host: "${REDIS_S_HOST}"
+		Type: node
+		Pass: "${REDIS_PASSWORD}"
+		User: "default"
+
+Kmq:
+	Name: email-api
+	Brokers:
+		- "${KAFKA_BROKER}"
+	Topic: "email-task"

app/email/api/internal/config/config.go

@@ -0,0 +1,16 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package config
+
+import (
+	"github.com/zeromicro/go-queue/kq"
+	"github.com/zeromicro/go-zero/core/stores/cache"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+type Config struct {
+	rest.RestConf
+	CacheConf cache.CacheConf
+	Kmq       kq.KqConf
+}

app/email/api/internal/handler/email/sendVerificationCodeHandler.go

@@ -0,0 +1,32 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package email
+
+import (
+	"net/http"
+
+	"github.com/zeromicro/go-zero/rest/httpx"
+	"juwan-backend/app/email/api/internal/logic/email"
+	"juwan-backend/app/email/api/internal/svc"
+	"juwan-backend/app/email/api/internal/types"
+)
+
+// 发送邮箱验证码
+func SendVerificationCodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.SendVerificationCodeReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := email.NewSendVerificationCodeLogic(r.Context(), svcCtx)
+		resp, err := l.SendVerificationCode(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

app/email/api/internal/handler/routes.go

@@ -0,0 +1,30 @@
+// Code generated by goctl. DO NOT EDIT.
+// goctl 1.9.2
+
+package handler
+
+import (
+	"net/http"
+
+	email "juwan-backend/app/email/api/internal/handler/email"
+	"juwan-backend/app/email/api/internal/svc"
+
+	"github.com/zeromicro/go-zero/rest"
+)
+
+func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
+	server.AddRoutes(
+		rest.WithMiddlewares(
+			[]rest.Middleware{serverCtx.Logger},
+			[]rest.Route{
+				{
+					// 发送邮箱验证码
+					Method:  http.MethodPost,
+					Path:    "/verification-code/send",
+					Handler: email.SendVerificationCodeHandler(serverCtx),
+				},
+			}...,
+		),
+		rest.WithPrefix("/api/email"),
+	)
+}

app/email/api/internal/logic/email/sendVerificationCodeLogic.go

@@ -0,0 +1,79 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package email
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"juwan-backend/app/email/api/internal/svc"
+	"juwan-backend/app/email/api/internal/types"
+	"juwan-backend/app/email/api/internal/utils"
+
+	"github.com/google/uuid"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type SendVerificationCodeLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+// 发送邮箱验证码
+func NewSendVerificationCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendVerificationCodeLogic {
+	return &SendVerificationCodeLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *SendVerificationCodeLogic) SendVerificationCode(req *types.SendVerificationCodeReq) (resp *types.SendVerificationCodeResp, err error) {
+	if l.svcCtx.RedisCluster == nil {
+		return nil, fmt.Errorf("redis not configured")
+	}
+
+	if l.svcCtx.EmailPusher == nil {
+		return nil, fmt.Errorf("kafka pusher not configured")
+	}
+
+	code := utils.GenCode()
+	requestID := uuid.NewString()
+
+	redisKey := fmt.Sprintf("%s:%s:%s", req.Email, code, req.Email)
+	if exists, getErr := l.svcCtx.RedisCluster.Get(l.ctx, redisKey).Result(); getErr == nil && exists != "" {
+		return nil, fmt.Errorf("verification code already sent, please wait before requesting a new one")
+	}
+	if setErr := l.svcCtx.RedisCluster.Set(l.ctx, redisKey, req.Scene, 60*time.Second).Err(); setErr != nil {
+		return nil, setErr
+	}
+
+	payload := map[string]any{
+		"type":      "verification_code",
+		"requestId": requestID,
+		"email":     req.Email,
+		"scene":     req.Scene,
+		"code":      code,
+		"expireIn":  60,
+	}
+	messageBytes, marshalErr := json.Marshal(payload)
+	if marshalErr != nil {
+		return nil, marshalErr
+	}
+
+	if pushErr := l.svcCtx.EmailPusher.PushWithKey(l.ctx, req.Email, string(messageBytes)); pushErr != nil {
+		return nil, pushErr
+	}
+
+	resp = &types.SendVerificationCodeResp{
+		RequestId:   requestID,
+		ExpireInSec: 60,
+		Message:     "verification code send task submitted",
+	}
+
+	return
+}

app/email/api/internal/middleware/loggerMiddleware.go

@@ -0,0 +1,22 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package middleware
+
+import "net/http"
+
+type LoggerMiddleware struct {
+}
+
+func NewLoggerMiddleware() *LoggerMiddleware {
+	return &LoggerMiddleware{}
+}
+
+func (m *LoggerMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		// TODO generate middleware implement function, delete after code implementation
+
+		// Passthrough to next handler if need
+		next(w, r)
+	}
+}

app/email/api/internal/svc/serviceContext.go

@@ -0,0 +1,43 @@
+// Code scaffolded by goctl. Safe to edit.
+// goctl 1.9.2
+
+package svc
+
+import (
+	"time"
+
+	"juwan-backend/app/email/api/internal/config"
+	"juwan-backend/app/email/api/internal/middleware"
+	"juwan-backend/common/redisx"
+
+	"github.com/redis/go-redis/v9"
+	"github.com/zeromicro/go-queue/kq"
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+type ServiceContext struct {
+	Config       config.Config
+	Logger       rest.Middleware
+	RedisCluster *redis.ClusterClient
+	EmailPusher  *kq.Pusher
+}
+
+func NewServiceContext(c config.Config) *ServiceContext {
+	redisConn, err := redisx.ConnectMasterSlaveCluster(c.CacheConf, 5*time.Second)
+	if err != nil {
+		logx.Errorf("failed to connect redis for email-api: %v", err)
+	}
+
+	var emailPusher *kq.Pusher
+	if len(c.Kmq.Brokers) > 0 && c.Kmq.Topic != "" {
+		emailPusher = kq.NewPusher(c.Kmq.Brokers, c.Kmq.Topic)
+	}
+
+	return &ServiceContext{
+		Config:       c,
+		Logger:       middleware.NewLoggerMiddleware().Handle,
+		RedisCluster: redisConn.Client,
+		EmailPusher:  emailPusher,
+	}
+}

app/email/api/internal/types/types.go

@@ -0,0 +1,15 @@
+// Code generated by goctl. DO NOT EDIT.
+// goctl 1.9.2
+
+package types
+
+type SendVerificationCodeReq struct {
+	Email string `json:"email" binding:"required,email"`
+	Scene string `json:"scene" binding:"required,oneof=register login reset_password bind_email"`
+}
+
+type SendVerificationCodeResp struct {
+	RequestId   string `json:"requestId"`
+	ExpireInSec int64  `json:"expireInSec"`
+	Message     string `json:"message"`
+}

app/email/api/internal/utils/genCode.go

@@ -0,0 +1,16 @@
+package utils
+
+import (
+	"crypto/rand"
+	"fmt"
+	"math/big"
+)
+
+func GenCode() string {
+	n, err := rand.Int(rand.Reader, big.NewInt(1000000))
+	if err != nil {
+		return "000000"
+	}
+
+	return fmt.Sprintf("%06d", n.Int64())
+}