feat: wire ResourcesRepo and ResourceManagementService into main

This commit is contained in:
tian 2026-05-06 12:43:37 +08:00
parent d4796a4ddb
commit b4a59c91aa

View File

@ -51,10 +51,17 @@ func main() {
if err := modelSvc.SyncStandardModelsFromDirectory(standardModelsDir); err != nil {
log.Fatalf("sync standard models: %v", err)
}
resourcesRepo := storage.NewResourcesRepo(store.DB())
resourceSvc := service.NewResourceManagementService(resourcesRepo)
standardResourcesDir := filepath.Join("resources", "standard_resources")
if err := resourceSvc.SyncStandardResourcesFromDirectory(standardResourcesDir); err != nil {
log.Printf("sync standard resources: %v", err)
}
stateRepo := storage.NewDeviceConfigStateRepo(store.DB())
auditRepo := storage.NewAuditLogsRepo(store.DB())
taskSvc := service.NewTaskService(cfg, agentClient, regSvc, taskRepo)
taskSvc.SetStandardModels(modelsRepo, standardModelsDir)
taskSvc.SetStandardResources(resourcesRepo)
taskSvc.SetDeviceConfigStateRepo(stateRepo)
taskSvc.SetAuditLogRepo(auditRepo)
if err := taskSvc.LoadPersistedTasks(); err != nil {
@ -88,6 +95,7 @@ func main() {
ui.SetStateRepo(stateRepo)
ui.SetAuditRepo(auditRepo)
ui.SetDBPath(cfg.DBPathOrDefault())
ui.SetResourcesRepo(resourcesRepo)
uiRouter, err := ui.Routes()
if err != nil {
log.Fatalf("failed to init ui routes: %v", err)