LukeParke/libopenapi
1
0
Fork 0
mirror of https://github.com/LukeHagar/libopenapi.git synced 2025-12-09 20:47:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added GetFullLineCount() to the Rolodex

Browse Source
This commit is contained in:
Calvin Lobo
2024-08-15 10:35:31 -04:00
committed by quobix
parent 49a292152b
commit 43144dec72
2 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
index/rolodex.go
View File

@@ -626,3 +626,23 @@ func (r *Rolodex) RolodexFileSize() int64 {
}
return size
}
// GetFullLineCount returns the total number of lines from all files in the Rolodex
func (r *Rolodex) GetFullLineCount() int64 {
var lineCount int64
for _, v := range r.localFS {
if lfs, ok := v.(RolodexFS); ok {
for _, f := range lfs.GetFiles() {
lineCount += int64(strings.Count(f.GetContent(), "\n")) + 1
}
}
}
for _, v := range r.remoteFS {
if lfs, ok := v.(RolodexFS); ok {
for _, f := range lfs.GetFiles() {
lineCount += int64(strings.Count(f.GetContent(), "\n")) + 1
}
}
}
return lineCount
}

3
index/rolodex_test.go
View File

@@ -1529,6 +1529,9 @@ func TestRolodex_SimpleTest_OneDoc(t *testing.T) {
assert.Nil(t, rolo.GetRootIndex())
assert.Len(t, rolo.GetIndexes(), 10)
lineCount := rolo.GetFullLineCount()
assert.Equal(t, int64(158), lineCount, "total line count in the rolodex is wrong")
assert.NoError(t, err)
assert.Len(t, rolo.indexes, 10)