notify-service/internal/data/greeter.go

package data

import (
	"context"
	"time"

	"bkb-notify/internal/biz"

	"errors"
	"github.com/go-kratos/kratos/v2/log"
)

type greeterRepo struct {
	data *Data
	log  *log.Helper
}

// NewGreeterRepo .
func NewGreeterRepo(data *Data, logger log.Logger) biz.GreeterRepo {
	return &greeterRepo{
		data: data,
		log:  log.NewHelper(logger),
	}
}

func (r *greeterRepo) Save(ctx context.Context, g *biz.Greeter) (*biz.Greeter, error) {
	return g, nil
}

func (r *greeterRepo) FindByID(context.Context, int64) (*biz.Greeter, error) {
	return nil, nil
}

func (r *greeterRepo) ListByName(context.Context, string) ([]*biz.Greeter, error) {
	return nil, nil
}

func (r *greeterRepo) ListAll(context.Context) ([]*biz.Greeter, error) {
	return nil, nil
}
// smsRepo
// (r *greeterRepo)set()
// r.Data.rdb.set 

func (r *greeterRepo) CreateSms(ctx context.Context, phone, code string) error {
	return r.data.rdb.Set("sms_code:"+phone, code, 5*time.Minute).Err()
}
func (r *greeterRepo) CheckSms(ctx context.Context, phone, code string) error {
	result, _ := r.data.rdb.Get("sms_code:" + phone).Result()
	if result == "" {
		return errors.New("验证码已过期，请重新获取")
	}
	if result != code {
		return errors.New("验证码错误，请重新输入")
	}
	return r.data.rdb.Del("sms_code:" + phone).Err()
}