feat: support export UE data on task scheduled
This commit is contained in:
@@ -9,6 +9,7 @@ import (
|
||||
"be.ems/src/modules/network_data/service"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// 模块路由注册
|
||||
@@ -323,6 +324,34 @@ func Setup(router *gin.Engine) {
|
||||
}
|
||||
}
|
||||
|
||||
// ResettableService 接口定义
|
||||
type ResettableService interface {
|
||||
ResetData(neID string) int64
|
||||
}
|
||||
|
||||
// 服务注册表
|
||||
var serviceRegistry = make(map[string]ResettableService)
|
||||
func RegisterService(name string, service ResettableService) {
|
||||
serviceRegistry[name] = service
|
||||
}
|
||||
|
||||
// 获取服务
|
||||
func GetService(name string) (ResettableService, error) {
|
||||
service, exists := serviceRegistry[name]
|
||||
if !exists {
|
||||
return nil, fmt.Errorf("service %s not found", name)
|
||||
}
|
||||
return service, nil
|
||||
}
|
||||
|
||||
// 初始化注册表
|
||||
func init() {
|
||||
RegisterService("UDMAuthData", service.NewUDMAuthUser)
|
||||
RegisterService("UDMSubUser", service.NewUDMSubUser)
|
||||
|
||||
// 这里注册更多服务
|
||||
}
|
||||
|
||||
// InitLoad 初始参数
|
||||
func InitLoad() {
|
||||
// 启动时,加载UPF上下行流量
|
||||
|
||||
Reference in New Issue
Block a user