98 lines
1.3 KiB
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)}
|
|
}
|