f
This commit is contained in:
@@ -427,13 +427,26 @@ func registerAllProcessors(combService *comb.CombService) {
|
||||
"PDFG01GZ": pdfg.ProcessPDFG01GZRequest,
|
||||
}
|
||||
|
||||
// 批量注册到组合包服务
|
||||
// 批量注册到组合包服务(包装白名单:读 ctx 中命中的 api_code,嵌套子调用按子 api_code 判断)
|
||||
for apiCode, processor := range processorMap {
|
||||
combService.RegisterProcessor(apiCode, processor)
|
||||
wrapped := wrapProcessorWithWhitelist(apiCode, processor)
|
||||
combService.RegisterProcessor(apiCode, wrapped)
|
||||
}
|
||||
|
||||
// 同时设置全局处理器映射
|
||||
RequestProcessors = processorMap
|
||||
RequestProcessors = make(map[string]processors.ProcessorFunc, len(processorMap))
|
||||
for apiCode, processor := range processorMap {
|
||||
RequestProcessors[apiCode] = wrapProcessorWithWhitelist(apiCode, processor)
|
||||
}
|
||||
}
|
||||
|
||||
func wrapProcessorWithWhitelist(apiCode string, processor processors.ProcessorFunc) processors.ProcessorFunc {
|
||||
return func(ctx context.Context, params []byte, deps *processors.ProcessorDependencies) ([]byte, error) {
|
||||
if processors.WhitelistShouldReturnEmpty(ctx, apiCode) {
|
||||
return nil, processors.ErrNotFound
|
||||
}
|
||||
return processor(ctx, params, deps)
|
||||
}
|
||||
}
|
||||
|
||||
// 注册API处理器 - 现在通过registerAllProcessors统一管理
|
||||
|
||||
Reference in New Issue
Block a user