status-go/vendor/github.com/anacrolix/missinggo/pproffd/pproffd.go

98 lines
1.3 KiB
Go

// Package pproffd is for detecting resource leaks due to unclosed handles.
package pproffd
import (
"io"
"net"
"os"
"runtime/pprof"
)
const enabled = false
var p *pprof.Profile
func init() {
if enabled {
p = pprof.NewProfile("fds")
}
}
type fd int
func (me *fd) Closed() {
p.Remove(me)
}
func add(skip int) (ret *fd) {
ret = new(fd)
p.Add(ret, skip+2)
return
}
type closeWrapper struct {
fd *fd
c io.Closer
}
func (me closeWrapper) Close() error {
me.fd.Closed()
return me.c.Close()
}
func newCloseWrapper(c io.Closer) closeWrapper {
return closeWrapper{
fd: add(2),
c: c,
}
}
type wrappedNetConn struct {
net.Conn
closeWrapper
}
func (me wrappedNetConn) Close() error {
return me.closeWrapper.Close()
}
// Tracks a net.Conn until Close() is explicitly called.
func WrapNetConn(nc net.Conn) net.Conn {
if !enabled {
return nc
}
if nc == nil {
return nil
}
return wrappedNetConn{
nc,
newCloseWrapper(nc),
}
}
type OSFile interface {
io.Reader
io.Seeker
io.Closer
io.Writer
Stat() (os.FileInfo, error)
io.ReaderAt
io.WriterAt
}
type wrappedOSFile struct {
*os.File
closeWrapper
}
func (me wrappedOSFile) Close() error {
return me.closeWrapper.Close()
}
func WrapOSFile(f *os.File) OSFile {
if !enabled {
return f
}
return &wrappedOSFile{f, newCloseWrapper(f)}
}