S.Solodyagin
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package filestore
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"git.company.lan/gopkg/filestore/remote"
|
|
)
|
|
|
|
// Убеждаемся в том, что мы всегда реализуем интерфейс http.FileSystem.
|
|
var _ http.FileSystem = (*HttpFS)(nil)
|
|
|
|
type HttpFS struct {
|
|
localStorage *LocalStorage
|
|
remoteStorage remote.Storage
|
|
}
|
|
|
|
type HttpFSOption func(*HttpFS)
|
|
|
|
// WithRemoteStorage
|
|
func WithRemoteStorage(storage remote.Storage) HttpFSOption {
|
|
return func(f *HttpFS) {
|
|
f.remoteStorage = storage
|
|
}
|
|
}
|
|
|
|
// NewHttpFS создаёт новый экземпляр файловой системы.
|
|
func NewHttpFS(rootDir string, opts ...HttpFSOption) (*HttpFS, error) {
|
|
localStorage, err := NewLocalStorage(rootDir)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
f := &HttpFS{}
|
|
f.localStorage = localStorage
|
|
|
|
for _, opt := range opts {
|
|
if opt != nil {
|
|
opt(f)
|
|
}
|
|
}
|
|
|
|
return f, nil
|
|
}
|
|
|
|
// Open реализует метод http.FileSystem.
|
|
func (f *HttpFS) Open(name string) (http.File, error) {
|
|
name = strings.TrimPrefix(name, "/")
|
|
if f.remoteStorage != nil {
|
|
return f.remoteStorage.Open(name)
|
|
}
|
|
return f.localStorage.Open(name)
|
|
}
|
|
|
|
// Remove удаляет файл.
|
|
func (f *HttpFS) Remove(name string) error {
|
|
name = strings.TrimPrefix(name, "/")
|
|
if f.remoteStorage != nil {
|
|
return f.remoteStorage.Remove(name)
|
|
}
|
|
return f.localStorage.Remove(name)
|
|
}
|
|
|
|
// LocalStorage возвращает указатель на локальное хранилище.
|
|
func (f *HttpFS) LocalStorage() *LocalStorage { return f.localStorage }
|
|
|
|
// RemoteStorage возвращает указатель на удалённое хранилище.
|
|
func (f *HttpFS) RemoteStorage() remote.Storage { return f.remoteStorage }
|