zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-22 07:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: dataset/document/file service (#17071)

Browse Source 
### Summary

Refactor dataset.go document.do file.go file2document.go in
internal/service.
This commit is contained in:
Jack
2026-07-20 09:48:24 +08:00
committed by GitHub
parent 99f13172ad
commit 965590ccbe
88 changed files with 10776 additions and 9009 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
cmd/ragflow_server.go
View File

@@ -34,6 +34,9 @@ import (
"ragflow/internal/server/local"
"ragflow/internal/service"
"ragflow/internal/service/chunk"
dataset "ragflow/internal/service/dataset"
"ragflow/internal/service/document"
"ragflow/internal/service/file"
"ragflow/internal/service/nlp"
"ragflow/internal/storage"
"ragflow/internal/syncer"
@@ -681,8 +684,8 @@ func startServer(config *server.Config) {
// Initialize service layer
userService := service.NewUserService()
documentService := service.NewDocumentService()
datasetsService := service.NewDatasetService()
documentService := document.NewDocumentService()
datasetsService := dataset.NewDatasetService()
metadataService := service.NewMetadataService()
chunkService := chunk.NewChunkService()
llmService := service.NewLLMService()
@@ -696,7 +699,7 @@ func startServer(config *server.Config) {
connectorService := service.NewConnectorService()
searchService := service.NewSearchService()
searchService.SetTenantService(tenantService)
fileService := service.NewFileService()
fileService := file.NewFileService(service.CheckFileTeamPermission, documentService)
memoryService := service.NewMemoryService()
mcpService := service.NewMCPService()
modelProviderService := service.NewModelProviderService()
@@ -711,7 +714,7 @@ func startServer(config *server.Config) {
authHandler := handler.NewAuthHandler()
userHandler := handler.NewUserHandler(userService)
tenantHandler := handler.NewTenantHandler(tenantService, userService, datasetsService)
documentHandler := handler.NewDocumentHandler(documentService, datasetsService)
documentHandler := handler.NewDocumentHandler(documentService, datasetsService, fileService)
datasetsHandler := handler.NewDatasetsHandler(datasetsService, metadataService)
systemHandler := handler.NewSystemHandler(systemService)
chunkHandler := handler.NewChunkHandler(chunkService, userService)
@@ -741,7 +744,7 @@ func startServer(config *server.Config) {
return handler.MCPRetrieval(datasetsService, userID, req)
},
)
skillSearchHandler := handler.NewSkillSearchHandler(docEngine)
skillSearchHandler := handler.NewSkillSearchHandler(docEngine, documentService)
providerHandler := handler.NewProviderHandler(userService, modelProviderService)
// Install the agent service's Redis-backed run infrastructure
// (CheckPointStore / StateSerializer / RunTracker). When Redis
@@ -789,7 +792,7 @@ func startServer(config *server.Config) {
searchHandler.SetCompletionDependencies(modelProviderService, askService)
pluginHandler := handler.NewPluginHandler(service.NewPluginService())
modelHandler := handler.NewModelHandler(service.NewModelProviderService())
fileCommitHandler := handler.NewFileCommitHandler(service.NewFileCommitService())
fileCommitHandler := handler.NewFileCommitHandler(file.NewFileCommitService())
// Dify retrieval handler
docDAO := documentDAO