mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package mirror
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
repo_model "github.com/hanzoai/git/models/repo"
|
|
"github.com/hanzoai/git/modules/log"
|
|
"github.com/hanzoai/git/modules/queue"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
)
|
|
|
|
// doMirrorSync causes this request to mirror itself
|
|
func doMirrorSync(ctx context.Context, req *SyncRequest) {
|
|
if req.ReferenceID == 0 {
|
|
log.Warn("Skipping mirror sync request, no mirror ID was specified")
|
|
return
|
|
}
|
|
switch req.Type {
|
|
case PushMirrorType:
|
|
_ = SyncPushMirror(ctx, req.ReferenceID)
|
|
case PullMirrorType:
|
|
_ = SyncPullMirror(ctx, req.ReferenceID)
|
|
default:
|
|
log.Error("Unknown Request type in queue: %v for MirrorID[%d]", req.Type, req.ReferenceID)
|
|
}
|
|
}
|
|
|
|
var errLimit = errors.New("reached limit")
|
|
|
|
// Update checks and updates mirror repositories.
|
|
func Update(ctx context.Context, pullLimit, pushLimit int) error {
|
|
if !setting.Mirror.Enabled {
|
|
log.Warn("Mirror feature disabled, but cron job enabled: skip update")
|
|
return nil
|
|
}
|
|
log.Trace("Doing: Update")
|
|
|
|
handler := func(bean any) error {
|
|
var repo *repo_model.Repository
|
|
var mirrorType SyncType
|
|
var referenceID int64
|
|
|
|
if m, ok := bean.(*repo_model.Mirror); ok {
|
|
if m.GetRepository(ctx) == nil {
|
|
log.Error("Disconnected mirror found: %d", m.ID)
|
|
return nil
|
|
}
|
|
repo = m.Repo
|
|
mirrorType = PullMirrorType
|
|
referenceID = m.RepoID
|
|
} else if m, ok := bean.(*repo_model.PushMirror); ok {
|
|
if m.GetRepository(ctx) == nil {
|
|
log.Error("Disconnected push-mirror found: %d", m.ID)
|
|
return nil
|
|
}
|
|
repo = m.Repo
|
|
mirrorType = PushMirrorType
|
|
referenceID = m.ID
|
|
} else {
|
|
log.Error("Unknown bean: %v", bean)
|
|
return nil
|
|
}
|
|
|
|
// Check we've not been cancelled
|
|
select {
|
|
case <-ctx.Done():
|
|
return errors.New("aborted")
|
|
default:
|
|
}
|
|
|
|
// Push to the Queue
|
|
if err := PushToQueue(mirrorType, referenceID); err != nil {
|
|
if err == queue.ErrAlreadyInQueue {
|
|
if mirrorType == PushMirrorType {
|
|
log.Trace("PushMirrors for %-v already queued for sync", repo)
|
|
} else {
|
|
log.Trace("PullMirrors for %-v already queued for sync", repo)
|
|
}
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
pullMirrorsRequested := 0
|
|
if pullLimit != 0 {
|
|
if err := repo_model.MirrorsIterate(ctx, pullLimit, func(_ int, bean any) error {
|
|
if err := handler(bean); err != nil {
|
|
return err
|
|
}
|
|
pullMirrorsRequested++
|
|
return nil
|
|
}); err != nil && err != errLimit {
|
|
log.Error("MirrorsIterate: %v", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
pushMirrorsRequested := 0
|
|
if pushLimit != 0 {
|
|
if err := repo_model.PushMirrorsIterate(ctx, pushLimit, func(idx int, bean any) error {
|
|
if err := handler(bean); err != nil {
|
|
return err
|
|
}
|
|
pushMirrorsRequested++
|
|
return nil
|
|
}); err != nil && err != errLimit {
|
|
log.Error("PushMirrorsIterate: %v", err)
|
|
return err
|
|
}
|
|
}
|
|
log.Trace("Finished: Update: %d pull mirrors and %d push mirrors queued", pullMirrorsRequested, pushMirrorsRequested)
|
|
return nil
|
|
}
|
|
|
|
// InitSyncMirrors initializes a go routine to sync the mirrors
|
|
func InitSyncMirrors() {
|
|
StartSyncMirrors()
|
|
}
|