Упрощенный способ превратить байт [] в "виртуальный" объект File в golang?
2 ответа
В стандартной библиотеке нет готового решения для этого, но это не так сложно сделать самостоятельно.
Что нам нужно это http.File
интерфейс:
type File interface {
io.Closer
io.Reader
io.Seeker
Readdir(count int) ([]os.FileInfo, error)
Stat() (os.FileInfo, error)
}
Обратите внимание, что мы можем использовать bytes.Reader
сделать тяжелую задачу, как это одно только реализует io.Reader
а также io.Seeker
, io.Closer
может быть просто Readdir()
может вернуться nil, nil
когда мы дразним файл, а не каталог, его Readdir()
даже не позвонит.
Самое сложное - это издеваться Stat()
вернуть значение, которое реализует os.FileInfo
,
Вот простое издевательство FileInfo
:
type myFileInfo struct {
name string
data []byte
}
func (mif myFileInfo) Name() string { return mif.name }
func (mif myFileInfo) Size() int64 { return int64(len(mif.data)) }
func (mif myFileInfo) Mode() os.FileMode { return 0444 } // Read for all
func (mif myFileInfo) ModTime() time.Time { return time.Time{} } // Return anything
func (mif myFileInfo) IsDir() bool { return false }
func (mif myFileInfo) Sys() interface{} { return nil }
И с этим у нас есть все, чтобы создать нашу издеваться http.File
:
type MyFile struct {
*bytes.Reader
mif myFileInfo
}
func (mf *MyFile) Close() error { return nil } // Noop, nothing to do
func (mf *MyFile) Readdir(count int) ([]os.FileInfo, error) {
return nil, nil // We are not a directory but a single file
}
func (mf *MyFile) Stat() (os.FileInfo, error) {
return mf.mif, nil
}
Пример его использования (попробуйте на Go Playground):
data := []byte{0, 1, 2, 3}
mf := &MyFile{
Reader: bytes.NewReader(data),
mif: myFileInfo{
name: "somename.txt",
data: data,
},
}
var f http.File = mf
_ = f
Если вы ищете этот ответ на gzip http2 push, вы можете сделать это с помощью только параметров:
options := &http.PushOptions{
Header: http.Header{
"Accept-Encoding": r.Header["Accept-Encoding"],
},
}
if err := pusher.Push(filePush, options); err != nil {
log.Printf("Failed to push: %v", err)
return
}
Кажется, достаточно просто издеваться над собой.
type MockFile struct {
data []byte
isOpen bool
offset int64
}
type MockFileInfo struct {
mockFile *MockFile
}
func (mfi *MockFileInfo) Name() string { return "MockFile" }
func (mfi *MockFileInfo) Size() int64 { return len(mfi.data) }
func (mfi *MockFileInfo) Mode() os.FileMode { return os.ModeIrregular }
func (mfi *MockFileInfo) ModTime() time.Time { return time.Now() }
func (mfi *MockFileInfo) IsDir() bool { return false }
func (mfi *MockFileInfo) Sys() interface { return nil }
func (mf *MockFile) Read(p []byte) (n int, err error) {
if mf.isOpen {
n = copy(p, mf.data[mf.offset:])
mf.offset += n
} else {
err = errors.New("Cannot read from closed MockFile")
}
return
}
func (mf *MockFile) Close() error {
if !mf.isOpen {
return errors.New("Cannot close an already closed MockFile")
mf.isOpen = false
return nil
}
func (mf *MockFile) Seek(offset int64, whence int) (ret int64, err error) {
var relativeTo int64
switch whence {
case 0:
relativeTo = 0
case 1:
relativeTo = mf.offset
case 2:
relativeTo = len(mf.data)
}
ret := relativeTo + offset
if ret < 0 || ret > len(mf.data) {
return -1, errors.New("New offset would fall outside of the MockFile")
}
mf.offset = ret
return
}
func (mf *MockFile) Readdir(count int) ([]os.FileInfo, error) {
if count <= 0 {
return []os.FileInfo{}, nil
}
return []os.FileInfo{}, errors.New("MockFiles have no associated directory")
}
func (mf *MockFile) Stat() (os.FileInfo, error) {
return MockFileInfo{mf}
}
func OpenMockFile(data []byte) *MockFile {
mf := MockFile{data, true, 0}
}