beenull/moby
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

pkg/fileutils: GetTotalUsedFds: reduce allocations

Browse source 
Use File.Readdirnames instead of os.ReadDir, as we're only interested in
the number of files, and results don't have to be sorted.

Before:

    BenchmarkGetTotalUsedFds-5   	  149272	      7896 ns/op	     945 B/op	      20 allocs/op

After:

    BenchmarkGetTotalUsedFds-5   	  153517	      7644 ns/op	     408 B/op	      10 allocs/op

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
This commit is contained in:
Sebastiaan van Stijn 2023-06-29 15:50:02 +02:00
parent 03390be5fa
commit eaa9494b71
No known key found for this signature in database
GPG key ID: 76698F39D527CE8C
1 changed files with 20 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
pkg/fileutils/fileutils_linux.go
View file

@ -3,6 +3,7 @@ package fileutils
import (
"context"
"fmt"
"io"
"os"
"github.com/containerd/containerd/log"
@ -11,10 +12,24 @@ import (
// GetTotalUsedFds Returns the number of used File Descriptors by
// reading it via /proc filesystem.
func GetTotalUsedFds() int {
if fds, err := os.ReadDir(fmt.Sprintf("/proc/%d/fd", os.Getpid())); err != nil {
log.G(context.TODO()).Errorf("Error opening /proc/%d/fd: %s", os.Getpid(), err)
} else {
return len(fds)
}
name := fmt.Sprintf("/proc/%d/fd", os.Getpid())
f, err := os.Open(name)
if err != nil {
log.G(context.TODO()).WithError(err).Error("Error listing file descriptors")
return -1
}
defer f.Close()
var fdCount int
for {
names, err := f.Readdirnames(100)
fdCount += len(names)
if err == io.EOF {
break
} else if err != nil {
log.G(context.TODO()).WithError(err).Error("Error listing file descriptors")
return -1
}
}
return fdCount
}