97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
package impl
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/infraboard/mcube/v2/ioc/config/cache"
|
|
"github.com/infraboard/mcube/v2/types"
|
|
"gitlab.com/go-course-project/go17/devcloud-mini/cmdb/apps/secret"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
// CreateSecret implements secret.Service.
|
|
func (s *SecretServiceImpl) CreateSecret(ctx context.Context, in *secret.CreateSecretRequest) (*secret.Secret, error) {
|
|
ins := secret.NewSecret(in)
|
|
|
|
// 需要加密
|
|
if err := ins.EncryptedApiSecret(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// upsert, gorm save
|
|
_, err := s.col.UpdateOne(ctx, bson.M{"_id": ins.Id}, bson.M{"$set": ins}, options.Update().SetUpsert(true))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ins, nil
|
|
}
|
|
|
|
// DescribeSecret implements secret.Service.
|
|
// 缓存怎么做
|
|
// 1. 从缓存中去(内存， 公共的内存服务 Redis)
|
|
// 2. 能获取到，直接返回
|
|
// 3. 不能获取, 选好从本地获取，返回，再把他设置到缓存中去
|
|
// 4. 怎么实现: redis redis get(key)/set(key), obj -> JSON
|
|
// 5. https://github.com/redis/go-redis get, set
|
|
// CacheGetter --> go-redis --> ObjectFinder
|
|
func (s *SecretServiceImpl) DescribeSecret(ctx context.Context, in *secret.DescribeSecretRequeset) (*secret.Secret, error) {
|
|
// 封装过后的改良版
|
|
ins := secret.NewSecret(secret.NewCreateSecretRequest())
|
|
err := cache.NewGetter(ctx, func(ctx context.Context, objectId string) (any, error) {
|
|
return s.describeSecret(ctx, in)
|
|
}).Get(in.Id, ins)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return ins, nil
|
|
|
|
}
|
|
|
|
// @cached(ttl=30s)
|
|
// h = cached(ttl=30s) -> h
|
|
func (s *SecretServiceImpl) describeSecret(ctx context.Context, in *secret.DescribeSecretRequeset) (*secret.Secret, error) {
|
|
// 取出后，需要解密
|
|
e := secret.NewSecret(&secret.CreateSecretRequest{})
|
|
// gorm take
|
|
if err := s.col.FindOne(ctx, bson.M{"_id": in.Id}).Decode(e); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
e.SetIsEncrypted(true)
|
|
if err := e.DecryptedApiSecret(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 解密过后的数据
|
|
return e, nil
|
|
}
|
|
|
|
// QuerySecret implements secret.Service.
|
|
func (s *SecretServiceImpl) QuerySecret(ctx context.Context, in *secret.QuerySecretRequest) (*types.Set[*secret.Secret], error) {
|
|
set := secret.NewSecretSet()
|
|
|
|
filter := bson.M{}
|
|
cursor, err := s.col.Find(ctx, filter, options.Find().SetLimit(int64(in.PageSize)).SetSkip(in.ComputeOffset()))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for cursor.Next(ctx) {
|
|
e := secret.NewSecret(&secret.CreateSecretRequest{})
|
|
if err := cursor.Decode(e); err != nil {
|
|
return nil, err
|
|
}
|
|
set.Add(e)
|
|
}
|
|
|
|
return set, nil
|
|
|
|
}
|
|
|
|
// SyncResource implements secret.Service.
|
|
func (s *SecretServiceImpl) SyncResource(context.Context, *secret.SyncResourceRequest, secret.SyncResourceHandleFunc) error {
|
|
panic("unimplemented")
|
|
}
|