package mount import ( "fmt" "github.com/chrislusf/seaweedfs/weed/glog" "github.com/chrislusf/seaweedfs/weed/mount/page_writer" "github.com/chrislusf/seaweedfs/weed/pb/filer_pb" "io" "sync" "time" ) type ChunkedDirtyPages struct { fh *FileHandle writeWaitGroup sync.WaitGroup lastErr error collection string replication string uploadPipeline *page_writer.UploadPipeline hasWrites bool } var ( _ = page_writer.DirtyPages(&ChunkedDirtyPages{}) ) func newMemoryChunkPages(fh *FileHandle, chunkSize int64) *ChunkedDirtyPages { dirtyPages := &ChunkedDirtyPages{ fh: fh, } swapFileDir := fh.wfs.option.getTempFilePageDir() dirtyPages.uploadPipeline = page_writer.NewUploadPipeline(fh.wfs.concurrentWriters, chunkSize, dirtyPages.saveChunkedFileIntevalToStorage, fh.wfs.option.ConcurrentWriters, swapFileDir) return dirtyPages } func (pages *ChunkedDirtyPages) AddPage(offset int64, data []byte, isSequential bool) { pages.hasWrites = true glog.V(4).Infof("%v memory AddPage [%d, %d)", pages.fh.fh, offset, offset+int64(len(data))) pages.uploadPipeline.SaveDataAt(data, offset, isSequential) return } func (pages *ChunkedDirtyPages) FlushData() error { if !pages.hasWrites { return nil } pages.uploadPipeline.FlushAll() if pages.lastErr != nil { return fmt.Errorf("flush data: %v", pages.lastErr) } return nil } func (pages *ChunkedDirtyPages) ReadDirtyDataAt(data []byte, startOffset int64) (maxStop int64) { if !pages.hasWrites { return } return pages.uploadPipeline.MaybeReadDataAt(data, startOffset) } func (pages *ChunkedDirtyPages) GetStorageOptions() (collection, replication string) { return pages.collection, pages.replication } func (pages *ChunkedDirtyPages) saveChunkedFileIntevalToStorage(reader io.Reader, offset int64, size int64, cleanupFn func()) { mtime := time.Now().UnixNano() defer cleanupFn() fileFullPath := pages.fh.FullPath() fileName := fileFullPath.Name() chunk, collection, replication, err := pages.fh.wfs.saveDataAsChunk(fileFullPath)(reader, fileName, offset) if err != nil { glog.V(0).Infof("%v saveToStorage [%d,%d): %v", fileFullPath, offset, offset+size, err) pages.lastErr = err return } chunk.Mtime = mtime pages.collection, pages.replication = collection, replication pages.fh.addChunks([]*filer_pb.FileChunk{chunk}) pages.fh.entryViewCache = nil glog.V(3).Infof("%v saveToStorage %s [%d,%d)", fileFullPath, chunk.FileId, offset, offset+size) } func (pages ChunkedDirtyPages) Destroy() { pages.uploadPipeline.Shutdown() } func (pages *ChunkedDirtyPages) LockForRead(startOffset, stopOffset int64) { pages.uploadPipeline.LockForRead(startOffset, stopOffset) } func (pages *ChunkedDirtyPages) UnlockForRead(startOffset, stopOffset int64) { pages.uploadPipeline.UnlockForRead(startOffset, stopOffset) }