gorm store update

store/gorm.go

@@ -6,41 +6,41 @@ import (
 	"gorm.io/gorm"
 )
 
-// GormStorage реализует интерфейс VersionStorage через GORM (SQLite)
+// GormStorage реализует интерфейс VersionStorage через GORM
 type GormStorage struct {
-	db *gorm.DB
+	open func() *gorm.DB
 }
 
-func NewGormStorage(db *gorm.DB) *GormStorage {
-	return &GormStorage{db: db}
+func NewGormStorage(open func() *gorm.DB) *GormStorage {
+	return &GormStorage{open: open}
 }
 
 func (s *GormStorage) CreateDocument(doc *models.Document) error {
-	err := s.db.Create(&doc).Error
+	err := s.open().Create(&doc).Error
 	return err
 }
 
 func (s *GormStorage) GetDocument(id uuid.UUID) (*models.Document, error) {
 	var d models.Document
-	if err := s.db.First(&d, "id = ?", id).Error; err != nil {
+	if err := s.open().First(&d, "id = ?", id).Error; err != nil {
 		return nil, err
 	}
 	return &d, nil
 }
 
 func (s *GormStorage) UpdateLatestVersion(docID, versionID uuid.UUID) error {
-	return s.db.Model(&models.Document{}).
+	return s.open().Model(&models.Document{}).
 		Where("id = ?", docID).
 		Update("latest_version", versionID).Error
 }
 
 func (s *GormStorage) CreateVersion(v *models.Version) error {
-	return s.db.Create(v).Error
+	return s.open().Create(v).Error
 }
 
 func (s *GormStorage) GetLatestVersion(docID uuid.UUID) (*models.Version, error) {
 	var v models.Version
-	err := s.db.
+	err := s.open().
 		Where("document_id = ?", docID).
 		Order("created_at desc").
 		First(&v).Error
@@ -52,7 +52,7 @@ func (s *GormStorage) GetLatestVersion(docID uuid.UUID) (*models.Version, error)
 
 func (s *GormStorage) GetVersionByID(id uuid.UUID) (*models.Version, error) {
 	var v models.Version
-	if err := s.db.First(&v, "id = ?", id).Error; err != nil {
+	if err := s.open().First(&v, "id = ?", id).Error; err != nil {
 		return nil, err
 	}
 	return &v, nil
@@ -63,7 +63,7 @@ func (s *GormStorage) GetParentVersion(v *models.Version) (*models.Version, erro
 		return nil, nil
 	}
 	var parent models.Version
-	if err := s.db.First(&parent, "id = ?", v.ParentID).Error; err != nil {
+	if err := s.open().First(&parent, "id = ?", v.ParentID).Error; err != nil {
 		return nil, err
 	}
 	return &parent, nil