senlinai/agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(documents): rebuild project document workbench

Browse Source
This commit is contained in:
yanweidong
2026-07-24 12:16:02 +08:00
parent 366136cee9
commit 77beaebb30
73 changed files with 5867 additions and 1264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
backend/internal/logic/search/service_postgres_test.go
View File

@@ -21,7 +21,7 @@ func TestPostgresSearchMatchesChinesePartialKeywords(t *testing.T) {
project := models.SaProject{OwnerID: owner.ID, Name: "中文回调平台", Identifier: "PG-CALLBACK"}
require.NoError(t, database.Create(&project).Error)
require.NoError(t, database.Create(&models.SaTask{ProjectID: project.ID, CreatedBy: owner.ID, Title: "回调任务", Status: "open"}).Error)
require.NoError(t, database.Create(&models.SaNote{ProjectID: project.ID, CreatedBy: owner.ID, Title: "说明", Markdown: "回调验签流程"}).Error)
createPostgresSearchDocument(t, database, project, owner.ID, models.SaDocument{Name: "说明.md"}, "回调验签流程")
results, err := NewService(database).Search(owner.ID, "回调")
@@ -38,16 +38,15 @@ func TestPostgresSearchKeepsAssignmentAndExplicitShareBoundaries(t *testing.T) {
assigned := models.SaTask{ProjectID: project.ID, CreatedBy: owner.ID, AssigneeID: &viewer.ID, Title: "边界词已指派", Status: "open"}
require.NoError(t, database.Create(&assigned).Error)
require.NoError(t, database.Create(&models.SaTask{ProjectID: project.ID, CreatedBy: owner.ID, Title: "边界词私有任务", Status: "open"}).Error)
sharedNote := models.SaNote{ProjectID: project.ID, CreatedBy: owner.ID, Title: "边界词共享笔记"}
require.NoError(t, database.Create(&sharedNote).Error)
require.NoError(t, database.Create(&models.SaNote{ProjectID: project.ID, CreatedBy: owner.ID, Title: "边界词私有笔记"}).Error)
require.NoError(t, database.Create(&models.SaTaskShare{TaskID: assigned.ID, ObjectType: "note", ObjectID: sharedNote.ID}).Error)
sharedDocument := createPostgresSearchDocument(t, database, project, owner.ID, models.SaDocument{Name: "边界词共享笔记.md"}, "")
createPostgresSearchDocument(t, database, project, owner.ID, models.SaDocument{Name: "边界词私有笔记.md"}, "")
require.NoError(t, database.Create(&models.SaTaskShare{TaskID: assigned.ID, ObjectType: "document", ObjectID: sharedDocument.ID}).Error)
results, err := NewService(database).Search(viewer.ID, "边界词")
require.NoError(t, err)
require.Len(t, results, 2)
require.ElementsMatch(t, []string{assigned.Identity, sharedNote.Identity}, []string{results[0].ID, results[1].ID})
require.ElementsMatch(t, []string{assigned.Identity, sharedDocument.Identity}, []string{results[0].ID, results[1].ID})
}
func TestPostgresSearchUsesStableFairLimitAndRuneBoundedSnippets(t *testing.T) {
@@ -71,14 +70,11 @@ func TestPostgresSearchUsesStableFairLimitAndRuneBoundedSnippets(t *testing.T) {
Status: "open",
UpdatedAt: baseTime.Add(time.Duration(index) * time.Minute),
}).Error)
require.NoError(t, database.Create(&models.SaNote{
createPostgresSearchDocument(t, database, project, owner.ID, models.SaDocument{
Identity: fmt.Sprintf("00000000-0000-7003-8000-%012d", index),
ProjectID: project.ID,
CreatedBy: owner.ID,
Title: fmt.Sprintf("稳定排序笔记 %02d", index),
Markdown: strings.Repeat("森", 300) + "尾",
Name: fmt.Sprintf("稳定排序笔记 %02d.md", index),
UpdatedAt: baseTime.Add(time.Duration(index) * time.Minute),
}).Error)
}, strings.Repeat("森", 300)+"尾")
}
service := NewService(database)
@@ -86,7 +82,7 @@ func TestPostgresSearchUsesStableFairLimitAndRuneBoundedSnippets(t *testing.T) {
require.NoError(t, err)
require.Len(t, results, maxSearchResults)
require.Equal(t,
[]string{"project", "task", "note", "project", "task", "note", "project", "task", "note"},
[]string{"project", "task", "document", "project", "task", "document", "project", "task", "document"},
resultTypes(results[:9]),
)
require.Equal(t, "00000000-0000-7001-8000-000000000059", results[0].ID)
@@ -96,17 +92,39 @@ func TestPostgresSearchUsesStableFairLimitAndRuneBoundedSnippets(t *testing.T) {
for _, result := range results {
counts[result.Type]++
}
require.Equal(t, map[string]int{"project": 17, "task": 17, "note": 16}, counts)
require.Equal(t, map[string]int{"project": 17, "task": 17, "document": 16}, counts)
again, err := service.Search(owner.ID, "稳定排序")
require.NoError(t, err)
require.Equal(t, results, again)
noteResults, err := service.Search(owner.ID, "森")
documentResults, err := service.Search(owner.ID, "森")
require.NoError(t, err)
require.NotEmpty(t, noteResults)
require.Len(t, []rune(noteResults[0].Snippet), 240)
require.NotContains(t, noteResults[0].Snippet, "尾")
require.NotEmpty(t, documentResults)
require.Len(t, []rune(documentResults[0].Snippet), 240)
require.NotContains(t, documentResults[0].Snippet, "尾")
}
func createPostgresSearchDocument(
t *testing.T,
database *gorm.DB,
project models.SaProject,
ownerID uint,
document models.SaDocument,
markdown string,
) models.SaDocument {
t.Helper()
document.ProjectID = project.ID
document.ProjectIdentity = project.Identity
document.CreatedBy = ownerID
document.Kind = models.DocumentKindFile
document.Extension = ".md"
document.MimeType = "text/markdown; charset=utf-8"
document.NormalizedName = strings.ToLower(document.Name)
document.Revision = 1
require.NoError(t, database.Create(&document).Error)
require.NoError(t, database.Create(&models.SaDocumentContent{DocumentID: document.ID, Markdown: markdown}).Error)
return document
}
func newPostgresSearchTestDB(t *testing.T) *gorm.DB {