first commit

apps/sentinel/internal/logic/secret/createsecretlogic.go

@@ -0,0 +1,46 @@
+package secretlogic
+
+import (
+	"context"
+	"tianyuan-api/apps/sentinel/internal/model"
+	"tianyuan-api/apps/sentinel/internal/svc"
+	"tianyuan-api/apps/sentinel/sentinel"
+	"tianyuan-api/pkg/crypto"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type CreateSecretLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewCreateSecretLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateSecretLogic {
+	return &CreateSecretLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// Secret methods
+func (l *CreateSecretLogic) CreateSecret(in *sentinel.CreateSecretRequest) (*sentinel.Secret, error) {
+	secretId, err := crypto.GenerateSecretId()
+	if err != nil {
+		return nil, err
+	}
+	aesKey, err := crypto.GenerateSecretKey()
+	if err != nil {
+		return nil, err
+	}
+	_, err = l.svcCtx.SecretsModel.Insert(l.ctx, &model.Secrets{
+		UserId:   in.UserId,
+		SecretId: secretId,
+		AesKey:   aesKey,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &sentinel.Secret{}, nil
+}

apps/sentinel/internal/logic/secret/getsecretbysecretidlogic.go

@@ -0,0 +1,43 @@
+package secretlogic
+
+import (
+	"context"
+	"errors"
+	"tianyuan-api/apps/sentinel/internal/model"
+	"tianyuan-api/apps/sentinel/internal/svc"
+	"tianyuan-api/apps/sentinel/sentinel"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetSecretBySecretIdLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewGetSecretBySecretIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSecretBySecretIdLogic {
+	return &GetSecretBySecretIdLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *GetSecretBySecretIdLogic) GetSecretBySecretId(in *sentinel.GetSecretBySecretIdRequest) (*sentinel.Secret, error) {
+	secret, err := l.svcCtx.SecretsModel.FindOneBySecretId(l.ctx, in.SecretId)
+	if err != nil {
+		if errors.Is(err, model.ErrNotFound) {
+			return nil, nil
+		} else {
+			return nil, err
+
+		}
+	}
+	return &sentinel.Secret{
+		Id:       secret.Id,
+		UserId:   secret.UserId,
+		SecretId: secret.SecretId,
+		AesKey:   secret.AesKey,
+	}, nil
+}

apps/sentinel/internal/logic/secret/getsecretbyuseridlogic.go

@@ -0,0 +1,40 @@
+package secretlogic
+
+import (
+	"context"
+	"errors"
+	"tianyuan-api/apps/sentinel/internal/model"
+	"tianyuan-api/apps/sentinel/internal/svc"
+	"tianyuan-api/apps/sentinel/sentinel"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetSecretByUserIdLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewGetSecretByUserIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSecretByUserIdLogic {
+	return &GetSecretByUserIdLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *GetSecretByUserIdLogic) GetSecretByUserId(in *sentinel.GetRecordByIdRequest) (*sentinel.Secret, error) {
+	secretModel, err := l.svcCtx.SecretsModel.FindOneByUserId(l.ctx, in.Id)
+	if err != nil {
+		if err == model.ErrNotFound {
+			return nil, errors.New("请先进行企业认证")
+		}
+		return nil, err
+	}
+
+	return &sentinel.Secret{
+		SecretId: secretModel.SecretId,
+		AesKey:   secretModel.AesKey,
+	}, nil
+}