Initial commit: Open sourcing all of the Maple Open Technologies code.
This commit is contained in:
commit
755d54a99d
2010 changed files with 448675 additions and 0 deletions
247
cloud/maplefile-backend/internal/repo/filemetadata/update.go
Normal file
247
cloud/maplefile-backend/internal/repo/filemetadata/update.go
Normal file
|
|
@ -0,0 +1,247 @@
|
|||
// monorepo/cloud/maplefile-backend/internal/maplefile/repo/filemetadata/update.go
|
||||
package filemetadata
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gocql/gocql"
|
||||
"go.uber.org/zap"
|
||||
|
||||
dom_file "codeberg.org/mapleopentech/monorepo/cloud/maplefile-backend/internal/domain/file"
|
||||
)
|
||||
|
||||
func (impl *fileMetadataRepositoryImpl) Update(file *dom_file.File) error {
|
||||
if file == nil {
|
||||
return fmt.Errorf("file cannot be nil")
|
||||
}
|
||||
|
||||
if !impl.isValidUUID(file.ID) {
|
||||
return fmt.Errorf("file ID is required")
|
||||
}
|
||||
|
||||
// Get existing file to compare changes
|
||||
existing, err := impl.Get(file.ID)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get existing file: %w", err)
|
||||
}
|
||||
|
||||
if existing == nil {
|
||||
return fmt.Errorf("file not found")
|
||||
}
|
||||
|
||||
// Update modified timestamp
|
||||
file.ModifiedAt = time.Now()
|
||||
|
||||
// Serialize encrypted file key
|
||||
encryptedKeyJSON, err := impl.serializeEncryptedFileKey(file.EncryptedFileKey)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to serialize encrypted file key: %w", err)
|
||||
}
|
||||
|
||||
// Serialize tags
|
||||
tagsJSON, err := impl.serializeTags(file.Tags)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to serialize tags: %w", err)
|
||||
}
|
||||
|
||||
batch := impl.Session.NewBatch(gocql.LoggedBatch)
|
||||
|
||||
// 1. Update main table
|
||||
batch.Query(`UPDATE maplefile.files_by_id SET
|
||||
collection_id = ?, owner_id = ?, encrypted_metadata = ?, encrypted_file_key = ?,
|
||||
encryption_version = ?, encrypted_hash = ?, encrypted_file_object_key = ?,
|
||||
encrypted_file_size_in_bytes = ?, encrypted_thumbnail_object_key = ?,
|
||||
encrypted_thumbnail_size_in_bytes = ?, tags = ?, created_at = ?, created_by_user_id = ?,
|
||||
modified_at = ?, modified_by_user_id = ?, version = ?, state = ?,
|
||||
tombstone_version = ?, tombstone_expiry = ?
|
||||
WHERE id = ?`,
|
||||
file.CollectionID, file.OwnerID, file.EncryptedMetadata, encryptedKeyJSON,
|
||||
file.EncryptionVersion, file.EncryptedHash, file.EncryptedFileObjectKey,
|
||||
file.EncryptedFileSizeInBytes, file.EncryptedThumbnailObjectKey,
|
||||
file.EncryptedThumbnailSizeInBytes, tagsJSON, file.CreatedAt, file.CreatedByUserID,
|
||||
file.ModifiedAt, file.ModifiedByUserID, file.Version, file.State,
|
||||
file.TombstoneVersion, file.TombstoneExpiry, file.ID)
|
||||
|
||||
// 2. Update collection table - delete old entry and insert new one
|
||||
if existing.CollectionID != file.CollectionID || existing.ModifiedAt != file.ModifiedAt {
|
||||
batch.Query(`DELETE FROM maplefile.files_by_collection
|
||||
WHERE collection_id = ? AND modified_at = ? AND id = ?`,
|
||||
existing.CollectionID, existing.ModifiedAt, file.ID)
|
||||
|
||||
batch.Query(`INSERT INTO maplefile.files_by_collection
|
||||
(collection_id, modified_at, id, owner_id, encrypted_metadata, encrypted_file_key,
|
||||
encryption_version, encrypted_hash, encrypted_file_object_key, encrypted_file_size_in_bytes,
|
||||
encrypted_thumbnail_object_key, encrypted_thumbnail_size_in_bytes,
|
||||
created_at, created_by_user_id, modified_by_user_id, version,
|
||||
state, tombstone_version, tombstone_expiry)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
file.CollectionID, file.ModifiedAt, file.ID, file.OwnerID, file.EncryptedMetadata,
|
||||
encryptedKeyJSON, file.EncryptionVersion, file.EncryptedHash, file.EncryptedFileObjectKey,
|
||||
file.EncryptedFileSizeInBytes, file.EncryptedThumbnailObjectKey,
|
||||
file.EncryptedThumbnailSizeInBytes, file.CreatedAt, file.CreatedByUserID,
|
||||
file.ModifiedByUserID, file.Version, file.State, file.TombstoneVersion, file.TombstoneExpiry)
|
||||
}
|
||||
|
||||
// 3. Update owner table - delete old entry and insert new one
|
||||
if existing.OwnerID != file.OwnerID || existing.ModifiedAt != file.ModifiedAt {
|
||||
batch.Query(`DELETE FROM maplefile.files_by_owner
|
||||
WHERE owner_id = ? AND modified_at = ? AND id = ?`,
|
||||
existing.OwnerID, existing.ModifiedAt, file.ID)
|
||||
|
||||
batch.Query(`INSERT INTO maplefile.files_by_owner
|
||||
(owner_id, modified_at, id, collection_id, encrypted_metadata, encrypted_file_key,
|
||||
encryption_version, encrypted_hash, encrypted_file_object_key, encrypted_file_size_in_bytes,
|
||||
encrypted_thumbnail_object_key, encrypted_thumbnail_size_in_bytes,
|
||||
created_at, created_by_user_id, modified_by_user_id, version,
|
||||
state, tombstone_version, tombstone_expiry)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
file.OwnerID, file.ModifiedAt, file.ID, file.CollectionID, file.EncryptedMetadata,
|
||||
encryptedKeyJSON, file.EncryptionVersion, file.EncryptedHash, file.EncryptedFileObjectKey,
|
||||
file.EncryptedFileSizeInBytes, file.EncryptedThumbnailObjectKey,
|
||||
file.EncryptedThumbnailSizeInBytes, file.CreatedAt, file.CreatedByUserID,
|
||||
file.ModifiedByUserID, file.Version, file.State, file.TombstoneVersion, file.TombstoneExpiry)
|
||||
}
|
||||
|
||||
// 4. Update created_by table - only if creator changed (rare) or created date changed
|
||||
if existing.CreatedByUserID != file.CreatedByUserID || existing.CreatedAt != file.CreatedAt {
|
||||
batch.Query(`DELETE FROM maplefile.files_by_creator
|
||||
WHERE created_by_user_id = ? AND created_at = ? AND id = ?`,
|
||||
existing.CreatedByUserID, existing.CreatedAt, file.ID)
|
||||
|
||||
batch.Query(`INSERT INTO maplefile.files_by_creator
|
||||
(created_by_user_id, created_at, id, collection_id, owner_id, encrypted_metadata,
|
||||
encrypted_file_key, encryption_version, encrypted_hash, encrypted_file_object_key,
|
||||
encrypted_file_size_in_bytes, encrypted_thumbnail_object_key, encrypted_thumbnail_size_in_bytes,
|
||||
modified_at, modified_by_user_id, version, state, tombstone_version, tombstone_expiry)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
file.CreatedByUserID, file.CreatedAt, file.ID, file.CollectionID, file.OwnerID,
|
||||
file.EncryptedMetadata, encryptedKeyJSON, file.EncryptionVersion, file.EncryptedHash,
|
||||
file.EncryptedFileObjectKey, file.EncryptedFileSizeInBytes, file.EncryptedThumbnailObjectKey,
|
||||
file.EncryptedThumbnailSizeInBytes, file.ModifiedAt, file.ModifiedByUserID, file.Version,
|
||||
file.State, file.TombstoneVersion, file.TombstoneExpiry)
|
||||
}
|
||||
|
||||
// 5. Update user sync table - delete old entry and insert new one for owner
|
||||
batch.Query(`DELETE FROM maplefile.files_by_user
|
||||
WHERE user_id = ? AND modified_at = ? AND id = ?`,
|
||||
existing.OwnerID, existing.ModifiedAt, file.ID)
|
||||
|
||||
batch.Query(`INSERT INTO maplefile.files_by_user
|
||||
(user_id, modified_at, id, collection_id, owner_id, encrypted_metadata,
|
||||
encrypted_file_key, encryption_version, encrypted_hash, encrypted_file_object_key,
|
||||
encrypted_file_size_in_bytes, encrypted_thumbnail_object_key, encrypted_thumbnail_size_in_bytes,
|
||||
tags, created_at, created_by_user_id, modified_by_user_id, version,
|
||||
state, tombstone_version, tombstone_expiry)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
file.OwnerID, file.ModifiedAt, file.ID, file.CollectionID, file.OwnerID,
|
||||
file.EncryptedMetadata, encryptedKeyJSON, file.EncryptionVersion, file.EncryptedHash,
|
||||
file.EncryptedFileObjectKey, file.EncryptedFileSizeInBytes, file.EncryptedThumbnailObjectKey,
|
||||
file.EncryptedThumbnailSizeInBytes, tagsJSON, file.CreatedAt, file.CreatedByUserID,
|
||||
file.ModifiedByUserID, file.Version, file.State, file.TombstoneVersion, file.TombstoneExpiry)
|
||||
|
||||
// 6. Update denormalized files_by_tag_id table
|
||||
// Calculate tag changes
|
||||
oldTagsMap := make(map[gocql.UUID]bool)
|
||||
for _, tag := range existing.Tags {
|
||||
oldTagsMap[tag.ID] = true
|
||||
}
|
||||
|
||||
newTagsMap := make(map[gocql.UUID]bool)
|
||||
for _, tag := range file.Tags {
|
||||
newTagsMap[tag.ID] = true
|
||||
}
|
||||
|
||||
// Delete entries for removed tags
|
||||
for tagID := range oldTagsMap {
|
||||
if !newTagsMap[tagID] {
|
||||
impl.Logger.Debug("removing file from tag denormalized table",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("tag_id", tagID.String()))
|
||||
batch.Query(`DELETE FROM maplefile.files_by_tag_id
|
||||
WHERE tag_id = ? AND file_id = ?`,
|
||||
tagID, file.ID)
|
||||
}
|
||||
}
|
||||
|
||||
// Insert/Update entries for current tags
|
||||
for _, tag := range file.Tags {
|
||||
impl.Logger.Debug("updating file in tag denormalized table",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("tag_id", tag.ID.String()))
|
||||
|
||||
batch.Query(`INSERT INTO maplefile.files_by_tag_id
|
||||
(tag_id, file_id, collection_id, owner_id, encrypted_metadata, encrypted_file_key,
|
||||
encryption_version, encrypted_hash, encrypted_file_object_key,
|
||||
encrypted_file_size_in_bytes, encrypted_thumbnail_object_key,
|
||||
encrypted_thumbnail_size_in_bytes, tag_ids, created_at, created_by_user_id,
|
||||
modified_at, modified_by_user_id, version, state, tombstone_version, tombstone_expiry,
|
||||
created_from_ip_address, modified_from_ip_address, ip_anonymized_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||
tag.ID, file.ID, file.CollectionID, file.OwnerID, file.EncryptedMetadata,
|
||||
encryptedKeyJSON, file.EncryptionVersion, file.EncryptedHash,
|
||||
file.EncryptedFileObjectKey, file.EncryptedFileSizeInBytes,
|
||||
file.EncryptedThumbnailObjectKey, file.EncryptedThumbnailSizeInBytes,
|
||||
tagsJSON, file.CreatedAt, file.CreatedByUserID, file.ModifiedAt,
|
||||
file.ModifiedByUserID, file.Version, file.State, file.TombstoneVersion,
|
||||
file.TombstoneExpiry,
|
||||
nil, nil, nil) // IP tracking fields not yet in domain model
|
||||
}
|
||||
|
||||
// Execute batch
|
||||
if err := impl.Session.ExecuteBatch(batch); err != nil {
|
||||
impl.Logger.Error("failed to update file",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.Error(err))
|
||||
return fmt.Errorf("failed to update file: %w", err)
|
||||
}
|
||||
|
||||
// Handle file count updates based on state changes
|
||||
wasActive := existing.State == dom_file.FileStateActive
|
||||
isActive := file.State == dom_file.FileStateActive
|
||||
|
||||
// Handle collection change for active files
|
||||
if existing.CollectionID != file.CollectionID && wasActive && isActive {
|
||||
// File moved from one collection to another while remaining active
|
||||
// Decrement old collection count
|
||||
if err := impl.CollectionRepo.DecrementFileCount(context.Background(), existing.CollectionID); err != nil {
|
||||
impl.Logger.Error("failed to decrement old collection file count",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("collection_id", existing.CollectionID.String()),
|
||||
zap.Error(err))
|
||||
// Don't fail the entire operation if count update fails
|
||||
}
|
||||
// Increment new collection count
|
||||
if err := impl.CollectionRepo.IncrementFileCount(context.Background(), file.CollectionID); err != nil {
|
||||
impl.Logger.Error("failed to increment new collection file count",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("collection_id", file.CollectionID.String()),
|
||||
zap.Error(err))
|
||||
// Don't fail the entire operation if count update fails
|
||||
}
|
||||
} else if wasActive && !isActive {
|
||||
// File transitioned from active to non-active (e.g., deleted)
|
||||
if err := impl.CollectionRepo.DecrementFileCount(context.Background(), existing.CollectionID); err != nil {
|
||||
impl.Logger.Error("failed to decrement collection file count",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("collection_id", existing.CollectionID.String()),
|
||||
zap.Error(err))
|
||||
// Don't fail the entire operation if count update fails
|
||||
}
|
||||
} else if !wasActive && isActive {
|
||||
// File transitioned from non-active to active (e.g., restored)
|
||||
if err := impl.CollectionRepo.IncrementFileCount(context.Background(), file.CollectionID); err != nil {
|
||||
impl.Logger.Error("failed to increment collection file count",
|
||||
zap.String("file_id", file.ID.String()),
|
||||
zap.String("collection_id", file.CollectionID.String()),
|
||||
zap.Error(err))
|
||||
// Don't fail the entire operation if count update fails
|
||||
}
|
||||
}
|
||||
|
||||
impl.Logger.Info("file updated successfully",
|
||||
zap.String("file_id", file.ID.String()))
|
||||
|
||||
return nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue