mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package ingest
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
"github.com/Rain-kl/Wavelet/internal/repository"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Ingest stores or resolves an upload using the configured policy and side effects.
|
|
func Ingest(ctx context.Context, req Request) (Result, error) {
|
|
normalizeRequest(&req)
|
|
if req.Hash == "" {
|
|
return Result{}, errors.New("ingest hash is required")
|
|
}
|
|
if req.Reader == nil {
|
|
return Result{}, errors.New("ingest reader is required")
|
|
}
|
|
if req.Size < 0 {
|
|
return Result{}, errors.New("ingest size must be non-negative")
|
|
}
|
|
|
|
switch req.Policy {
|
|
case PolicyDedupNewRecord, PolicyResolveExisting:
|
|
return ingestWithHashPolicy(ctx, req)
|
|
case PolicyCreate:
|
|
return createNewUpload(ctx, req)
|
|
default:
|
|
return Result{}, errors.New("unsupported ingest policy")
|
|
}
|
|
}
|
|
|
|
// FindByHash returns a reusable active upload with the same hash and size.
|
|
func FindByHash(ctx context.Context, hash string, size int64) (model.Upload, error) {
|
|
return repository.FindReusableUploadByHash(ctx, hash, size)
|
|
}
|
|
|
|
func ingestWithHashPolicy(ctx context.Context, req Request) (Result, error) {
|
|
existing, err := repository.FindReusableUploadByHash(ctx, req.Hash, req.Size)
|
|
if err == nil {
|
|
switch req.Policy {
|
|
case PolicyResolveExisting:
|
|
return Result{
|
|
Upload: existing,
|
|
Resolved: true,
|
|
}, nil
|
|
case PolicyDedupNewRecord:
|
|
if uploadstorageReadOnly(ctx) {
|
|
return Result{}, ErrStorageReadOnly
|
|
}
|
|
return createDedupRecord(ctx, existing, req)
|
|
}
|
|
}
|
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return Result{}, err
|
|
}
|
|
|
|
return createNewUpload(ctx, req)
|
|
}
|