Adding support for WriteAt

go-git · Dec 6, 2023 · 63a02cc · 63a02cc
1 parent d23351b
commit 63a02cc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/fs.go b/fs.go
@@ -164,6 +164,7 @@ type File interface {
 	// Name returns the name of the file as presented to Open.
 	Name() string
 	io.Writer
+	io.WriterAt
 	io.Reader
 	io.ReaderAt
 	io.Seeker

diff --git a/memfs/memory.go b/memfs/memory.go
@@ -272,6 +272,10 @@ func (f *file) Seek(offset int64, whence int) (int64, error) {
 }
 
 func (f *file) Write(p []byte) (int, error) {
+	return f.WriteAt(p, f.position)
+}
+
+func (f *file) WriteAt(p []byte, off int64) (int, error) {
 	if f.isClosed {
 		return 0, os.ErrClosed
 	}
@@ -280,8 +284,8 @@ func (f *file) Write(p []byte) (int, error) {
 		return 0, errors.New("write not supported")
 	}
 
-	n, err := f.content.WriteAt(p, f.position)
-	f.position += int64(n)
+	n, err := f.content.WriteAt(p, off)
+	f.position = off + int64(n)
 
 	return n, err
 }

diff --git a/test/mock.go b/test/mock.go
@@ -115,6 +115,10 @@ func (*FileMock) ReadAt(b []byte, off int64) (int, error) {
 	return 0, nil
 }
 
+func (*FileMock) WriteAt(b []byte, off int64) (int, error) {
+	return 0, nil
+}
+
 func (*FileMock) Seek(offset int64, whence int) (int64, error) {
 	return 0, nil
 }