67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
//go:build !appengine && !plan9
|
|
// +build !appengine,!plan9
|
|
|
|
package maxminddb
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
)
|
|
|
|
// Open takes a string path to a MaxMind DB file and returns a Reader
|
|
// structure or an error. The database file is opened using a memory map,
|
|
// except on Google App Engine where mmap is not supported; there the database
|
|
// is loaded into memory. Use the Close method on the Reader object to return
|
|
// the resources to the system.
|
|
func Open(file string) (*Reader, error) {
|
|
mapFile, err := os.Open(file)
|
|
if err != nil {
|
|
_ = mapFile.Close()
|
|
return nil, err
|
|
}
|
|
|
|
stats, err := mapFile.Stat()
|
|
if err != nil {
|
|
_ = mapFile.Close()
|
|
return nil, err
|
|
}
|
|
|
|
fileSize := int(stats.Size())
|
|
mmap, err := mmap(int(mapFile.Fd()), fileSize)
|
|
if err != nil {
|
|
_ = mapFile.Close()
|
|
return nil, err
|
|
}
|
|
|
|
if err := mapFile.Close(); err != nil {
|
|
//nolint:errcheck // we prefer to return the original error
|
|
munmap(mmap)
|
|
return nil, err
|
|
}
|
|
|
|
reader, err := FromBytes(mmap)
|
|
if err != nil {
|
|
//nolint:errcheck // we prefer to return the original error
|
|
munmap(mmap)
|
|
return nil, err
|
|
}
|
|
|
|
reader.hasMappedFile = true
|
|
runtime.SetFinalizer(reader, (*Reader).Close)
|
|
return reader, nil
|
|
}
|
|
|
|
// Close unmaps the database file from virtual memory and returns the
|
|
// resources to the system. If called on a Reader opened using FromBytes
|
|
// or Open on Google App Engine, this method does nothing.
|
|
func (r *Reader) Close() error {
|
|
var err error
|
|
if r.hasMappedFile {
|
|
runtime.SetFinalizer(r, nil)
|
|
r.hasMappedFile = false
|
|
err = munmap(r.buffer)
|
|
}
|
|
r.buffer = nil
|
|
return err
|
|
}
|