yvan/kra-new
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化结构

Browse Source
This commit is contained in:
Yvan
2026-08-28 14:54:06 +08:00
parent 56f9440a6b
commit ae5345de14
15 changed files with 188 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/service/system/export.go
+12 -2
View File
@@ -53,7 +53,13 @@ func (s *ExportService) IssueToken(ctx context.Context, templateID string, param
}
func (s *ExportService) ConsumeToken(ctx context.Context, token string, expectBlank bool) (*ExportToken, error) {
raw, ok, err := s.cache.Get(ctx, "export:"+token)
key := "export:" + token
raw, ok, err := s.cache.Get(ctx, key)
if atomic, supported := s.cache.(interface {
GetAndDelete(context.Context, string) (string, bool, error)
}); supported {
raw, ok, err = atomic.GetAndDelete(ctx, key)
}
if err != nil || !ok {
return nil, ErrExportTokenInvalid
}
@@ -64,7 +70,11 @@ func (s *ExportService) ConsumeToken(ctx context.Context, token string, expectBl
if value.Blank != expectBlank {
return nil, ErrExportTokenType
}
_ = s.cache.Delete(ctx, "export:"+token)
if _, supported := s.cache.(interface {
GetAndDelete(context.Context, string) (string, bool, error)
}); !supported {
_ = s.cache.Delete(ctx, key)
}
return &ExportToken{TemplateID: value.TemplateID, Params: value.Params, Blank: value.Blank}, nil
}
func exportTemplateDomain(value *dto.ExportTemplateRequest) *system.ExportTemplate {