Merge pull request #7 from jilio/feat/simpler-writer

feat: Simpler writer and proper close

Author: jilio

README.md

@@ -14,7 +14,7 @@ SQLiteFS is a Go package that implements the `fs.FS` interface using SQLite as a
 
 To use SQLiteFS in your Go project, run the following command:
 
-```
+```sh
 go get github.com/jilio/sqlitefs
 ```
 
@@ -51,7 +51,7 @@ func main() {
  defer sqliteFS.Close()
 
  // Write a file
- writer := sqlitefs.NewSQLiteWriter(sqliteFS, "example.txt")
+ writer := sqliteFS.NewWriter("example.txt")
  _, err = writer.Write([]byte("Hello, SQLiteFS!"))
  if err != nil {
   log.Fatal(err)

sqlitefs.go

@@ -42,6 +42,11 @@ func NewSQLiteFS(db *sql.DB) (*SQLiteFS, error) {
 	return fs, nil
 }
 
+// NewWriter creates a new writer for the specified path.
+func (fs *SQLiteFS) NewWriter(path string) *SQLiteWriter {
+	return NewSQLiteWriter(fs, path)
+}
+
 // Open открывает файл по указанному пути.
 func (fs *SQLiteFS) Open(name string) (fs.File, error) {
 	// Проверка существования файла в базе данных

writer.go

@@ -1,6 +1,7 @@
 package sqlitefs
 
 import (
+	"errors"
 	"mime"
 	"path/filepath"
 )
@@ -14,8 +15,11 @@ type SQLiteWriter struct {
 	fragmentSize  int
 	fragmentIndex int
 	fileCreated   bool
+	closed        bool
 }
 
+// NewSQLiteWriter creates a new SQLiteWriter for the specified path.
+// Deprecated: Use SQLiteFS.NewWriter instead.
 func NewSQLiteWriter(fs *SQLiteFS, path string) *SQLiteWriter {
 	return &SQLiteWriter{
 		fs:           fs,
@@ -26,6 +30,10 @@ func NewSQLiteWriter(fs *SQLiteFS, path string) *SQLiteWriter {
 }
 
 func (w *SQLiteWriter) Write(p []byte) (n int, err error) {
+	if w.closed {
+		return 0, errors.New("sqlitefs: write to closed writer")
+	}
+
 	n = len(p)
 	w.buffer = append(w.buffer, p...)
 
@@ -84,11 +92,17 @@ func (w *SQLiteWriter) createFileRecord() error {
 }
 
 func (w *SQLiteWriter) Close() error {
+	if w.closed {
+		return nil
+	}
+
 	for len(w.buffer) > 0 || !w.fileCreated {
 		err := w.writeFragment()
 		if err != nil {
 			return err
 		}
 	}
+
+	w.closed = true
 	return nil
 }