github-mirrors/VictoriaMetrics
1
0
Fork 0
mirror of https://github.com/VictoriaMetrics/VictoriaMetrics.git synced 2024-11-21 14:44:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

lib/fs: add ReaderAt.Path() function

Browse source 
This function is going to be used in VictoriaLogs
This commit is contained in:
Aliaksandr Valialkin 2023-06-19 22:42:25 -07:00
parent 497ec3f3e6
commit 371182f299
No known key found for this signature in database
GPG key ID: A72BEC6CD3D0DED1
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/fs/reader_at.go
View file

@ -19,6 +19,9 @@ const is32BitPtr = (^uintptr(0) >> 32) == 0
// MustReadAtCloser is rand-access read interface. // MustReadAtCloser is rand-access read interface.
type MustReadAtCloser interface { type MustReadAtCloser interface {
// Path must return path for the reader (e.g. file path, url or in-memory reference)
Path() string
// MustReadAt must read len(p) bytes from offset off to p. // MustReadAt must read len(p) bytes from offset off to p.
MustReadAt(p []byte, off int64) MustReadAt(p []byte, off int64)
@ -37,6 +40,11 @@ type ReaderAt struct {
useLocalStats bool useLocalStats bool
} }
// Path returns path to r.
func (r *ReaderAt) Path() string {
return r.f.Name()
}
// MustReadAt reads len(p) bytes at off from r. // MustReadAt reads len(p) bytes at off from r.
func (r *ReaderAt) MustReadAt(p []byte, off int64) { func (r *ReaderAt) MustReadAt(p []byte, off int64) {
if len(p) == 0 { if len(p) == 0 {
@ -48,10 +56,10 @@ func (r *ReaderAt) MustReadAt(p []byte, off int64) {
if len(r.mmapData) == 0 { if len(r.mmapData) == 0 {
n, err := r.f.ReadAt(p, off) n, err := r.f.ReadAt(p, off)
if err != nil { if err != nil {
logger.Panicf("FATAL: cannot read %d bytes at offset %d of file %q: %s", len(p), off, r.f.Name(), err) logger.Panicf("FATAL: cannot read %d bytes at offset %d of file %q: %s", len(p), off, r.Path(), err)
} }
if n != len(p) { if n != len(p) {
logger.Panicf("FATAL: unexpected number of bytes read from file %q; got %d; want %d", r.f.Name(), n, len(p)) logger.Panicf("FATAL: unexpected number of bytes read from file %q; got %d; want %d", r.Path(), n, len(p))
} }
} else { } else {
if off > int64(len(r.mmapData)-len(p)) { if off > int64(len(r.mmapData)-len(p)) {
@ -73,7 +81,7 @@ func (r *ReaderAt) MustReadAt(p []byte, off int64) {
// MustClose closes r. // MustClose closes r.
func (r *ReaderAt) MustClose() { func (r *ReaderAt) MustClose() {
fname := r.f.Name() fname := r.Path()
if len(r.mmapData) > 0 { if len(r.mmapData) > 0 {
if err := mUnmap(r.mmapData[:cap(r.mmapData)]); err != nil { if err := mUnmap(r.mmapData[:cap(r.mmapData)]); err != nil {
logger.Panicf("FATAL: cannot unmap data for file %q: %s", fname, err) logger.Panicf("FATAL: cannot unmap data for file %q: %s", fname, err)
@ -109,7 +117,7 @@ func (r *ReaderAt) SetUseLocalStats() {
// if prefetch is set, then the OS is hinted to prefetch f data. // if prefetch is set, then the OS is hinted to prefetch f data.
func (r *ReaderAt) MustFadviseSequentialRead(prefetch bool) { func (r *ReaderAt) MustFadviseSequentialRead(prefetch bool) {
if err := fadviseSequentialRead(r.f, prefetch); err != nil { if err := fadviseSequentialRead(r.f, prefetch); err != nil {
logger.Panicf("FATAL: error in fadviseSequentialRead(%q, %v): %s", r.f.Name(), prefetch, err) logger.Panicf("FATAL: error in fadviseSequentialRead(%q, %v): %s", r.Path(), prefetch, err)
} }
} }