feat: support export UE data on task scheduled

This commit is contained in:
zhangsz
2025-04-01 11:44:39 +08:00
parent c12061ce36
commit cfcef54ea8
5 changed files with 256 additions and 157 deletions

View File

@@ -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上下行流量