EdgeNode/internal/utils/bfs/file_reader.go

// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .

package bfs

import (
	"errors"
	"github.com/iwind/TeaGo/types"
	"io"
	"os"
)

type FileReader struct {
	bFile      *BlocksFile
	fp         *os.File
	fileHeader *FileHeader

	pos int64
}

func NewFileReader(bFile *BlocksFile, fp *os.File, fileHeader *FileHeader) *FileReader {
	return &FileReader{
		bFile:      bFile,
		fp:         fp,
		fileHeader: fileHeader,
	}
}

func (this *FileReader) FileHeader() *FileHeader {
	return this.fileHeader
}

func (this *FileReader) Read(b []byte) (n int, err error) {
	n, err = this.ReadAt(b, this.pos)
	this.pos += int64(n)

	return
}

func (this *FileReader) ReadAt(b []byte, offset int64) (n int, err error) {
	if offset >= this.fileHeader.MaxOffset() {
		err = io.EOF
		return
	}

	blockInfo, ok := this.fileHeader.BlockAt(offset)
	if !ok {
		err = errors.New("could not find block at '" + types.String(offset) + "'")
		return
	}

	var delta = offset - blockInfo.OriginOffsetFrom
	var bFrom = blockInfo.BFileOffsetFrom + delta
	var bTo = blockInfo.BFileOffsetTo
	if bFrom > bTo {
		err = errors.New("invalid block information")
		return
	}

	var bufLen = len(b)
	if int64(bufLen) > bTo-bFrom {
		bufLen = int(bTo - bFrom)
	}

	n, err = this.fp.ReadAt(b[:bufLen], bFrom)

	return
}

func (this *FileReader) Close() error {
	return this.fp.Close()
}