mirror of https://github.com/gogits/gogs.git
Unknwon
8 years ago
16 changed files with 345 additions and 242 deletions
@ -0,0 +1,243 @@
|
||||
// Copyright 2016 The Gogs Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package models |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strings" |
||||
"time" |
||||
|
||||
"github.com/Unknwon/com" |
||||
"github.com/go-xorm/xorm" |
||||
"gopkg.in/ini.v1" |
||||
|
||||
"github.com/gogits/gogs/modules/log" |
||||
"github.com/gogits/gogs/modules/process" |
||||
"github.com/gogits/gogs/modules/setting" |
||||
"github.com/gogits/gogs/modules/sync" |
||||
) |
||||
|
||||
var MirrorQueue = sync.NewUniqueQueue(setting.Repository.MirrorQueueLength) |
||||
|
||||
// Mirror represents mirror information of a repository.
|
||||
type Mirror struct { |
||||
ID int64 `xorm:"pk autoincr"` |
||||
RepoID int64 |
||||
Repo *Repository `xorm:"-"` |
||||
Interval int // Hour.
|
||||
EnablePrune bool `xorm:"NOT NULL DEFAULT true"` |
||||
|
||||
Updated time.Time `xorm:"-"` |
||||
UpdatedUnix int64 |
||||
NextUpdate time.Time `xorm:"-"` |
||||
NextUpdateUnix int64 |
||||
|
||||
address string `xorm:"-"` |
||||
} |
||||
|
||||
func (m *Mirror) BeforeInsert() { |
||||
m.NextUpdateUnix = m.NextUpdate.Unix() |
||||
} |
||||
|
||||
func (m *Mirror) BeforeUpdate() { |
||||
m.UpdatedUnix = time.Now().Unix() |
||||
m.NextUpdateUnix = m.NextUpdate.Unix() |
||||
} |
||||
|
||||
func (m *Mirror) AfterSet(colName string, _ xorm.Cell) { |
||||
var err error |
||||
switch colName { |
||||
case "repo_id": |
||||
m.Repo, err = GetRepositoryByID(m.RepoID) |
||||
if err != nil { |
||||
log.Error(3, "GetRepositoryByID[%d]: %v", m.ID, err) |
||||
} |
||||
case "updated_unix": |
||||
m.Updated = time.Unix(m.UpdatedUnix, 0).Local() |
||||
case "next_updated_unix": |
||||
m.NextUpdate = time.Unix(m.NextUpdateUnix, 0).Local() |
||||
} |
||||
} |
||||
|
||||
// ScheduleNextUpdate calculates and sets next update time.
|
||||
func (m *Mirror) ScheduleNextUpdate() { |
||||
m.NextUpdate = time.Now().Add(time.Duration(m.Interval) * time.Hour) |
||||
} |
||||
|
||||
func (m *Mirror) readAddress() { |
||||
if len(m.address) > 0 { |
||||
return |
||||
} |
||||
|
||||
cfg, err := ini.Load(m.Repo.GitConfigPath()) |
||||
if err != nil { |
||||
log.Error(4, "Load: %v", err) |
||||
return |
||||
} |
||||
m.address = cfg.Section("remote \"origin\"").Key("url").Value() |
||||
} |
||||
|
||||
// HandleCloneUserCredentials replaces user credentials from HTTP/HTTPS URL
|
||||
// with placeholder <credentials>.
|
||||
// It will fail for any other forms of clone addresses.
|
||||
func HandleCloneUserCredentials(url string, mosaics bool) string { |
||||
i := strings.Index(url, "@") |
||||
if i == -1 { |
||||
return url |
||||
} |
||||
start := strings.Index(url, "://") |
||||
if start == -1 { |
||||
return url |
||||
} |
||||
if mosaics { |
||||
return url[:start+3] + "<credentials>" + url[i:] |
||||
} |
||||
return url[:start+3] + url[i+1:] |
||||
} |
||||
|
||||
// Address returns mirror address from Git repository config without credentials.
|
||||
func (m *Mirror) Address() string { |
||||
m.readAddress() |
||||
return HandleCloneUserCredentials(m.address, false) |
||||
} |
||||
|
||||
// FullAddress returns mirror address from Git repository config.
|
||||
func (m *Mirror) FullAddress() string { |
||||
m.readAddress() |
||||
return m.address |
||||
} |
||||
|
||||
// SaveAddress writes new address to Git repository config.
|
||||
func (m *Mirror) SaveAddress(addr string) error { |
||||
configPath := m.Repo.GitConfigPath() |
||||
cfg, err := ini.Load(configPath) |
||||
if err != nil { |
||||
return fmt.Errorf("Load: %v", err) |
||||
} |
||||
|
||||
cfg.Section("remote \"origin\"").Key("url").SetValue(addr) |
||||
return cfg.SaveToIndent(configPath, "\t") |
||||
} |
||||
|
||||
// runSync returns true if sync finished without error.
|
||||
func (m *Mirror) runSync() bool { |
||||
repoPath := m.Repo.RepoPath() |
||||
wikiPath := m.Repo.WikiPath() |
||||
timeout := time.Duration(setting.Git.Timeout.Mirror) * time.Second |
||||
|
||||
gitArgs := []string{"remote", "update"} |
||||
if m.EnablePrune { |
||||
gitArgs = append(gitArgs, "--prune") |
||||
} |
||||
|
||||
if _, stderr, err := process.ExecDir( |
||||
timeout, repoPath, fmt.Sprintf("runSync: %s", repoPath), |
||||
"git", gitArgs...); err != nil { |
||||
desc := fmt.Sprintf("Fail to update mirror repository '%s': %s", repoPath, stderr) |
||||
log.Error(4, desc) |
||||
if err = CreateRepositoryNotice(desc); err != nil { |
||||
log.Error(4, "CreateRepositoryNotice: %v", err) |
||||
} |
||||
return false |
||||
} |
||||
if m.Repo.HasWiki() { |
||||
if _, stderr, err := process.ExecDir( |
||||
timeout, wikiPath, fmt.Sprintf("runSync: %s", wikiPath), |
||||
"git", "remote", "update", "--prune"); err != nil { |
||||
desc := fmt.Sprintf("Fail to update mirror wiki repository '%s': %s", wikiPath, stderr) |
||||
log.Error(4, desc) |
||||
if err = CreateRepositoryNotice(desc); err != nil { |
||||
log.Error(4, "CreateRepositoryNotice: %v", err) |
||||
} |
||||
return false |
||||
} |
||||
} |
||||
|
||||
return true |
||||
} |
||||
|
||||
func getMirrorByRepoID(e Engine, repoID int64) (*Mirror, error) { |
||||
m := &Mirror{RepoID: repoID} |
||||
has, err := e.Get(m) |
||||
if err != nil { |
||||
return nil, err |
||||
} else if !has { |
||||
return nil, ErrMirrorNotExist |
||||
} |
||||
return m, nil |
||||
} |
||||
|
||||
// GetMirrorByRepoID returns mirror information of a repository.
|
||||
func GetMirrorByRepoID(repoID int64) (*Mirror, error) { |
||||
return getMirrorByRepoID(x, repoID) |
||||
} |
||||
|
||||
func updateMirror(e Engine, m *Mirror) error { |
||||
_, err := e.Id(m.ID).AllCols().Update(m) |
||||
return err |
||||
} |
||||
|
||||
func UpdateMirror(m *Mirror) error { |
||||
return updateMirror(x, m) |
||||
} |
||||
|
||||
func DeleteMirrorByRepoID(repoID int64) error { |
||||
_, err := x.Delete(&Mirror{RepoID: repoID}) |
||||
return err |
||||
} |
||||
|
||||
// MirrorUpdate checks and updates mirror repositories.
|
||||
func MirrorUpdate() { |
||||
if taskStatusTable.IsRunning(_MIRROR_UPDATE) { |
||||
return |
||||
} |
||||
taskStatusTable.Start(_MIRROR_UPDATE) |
||||
defer taskStatusTable.Stop(_MIRROR_UPDATE) |
||||
|
||||
log.Trace("Doing: MirrorUpdate") |
||||
|
||||
if err := x.Where("next_update_unix<=?", time.Now().Unix()).Iterate(new(Mirror), func(idx int, bean interface{}) error { |
||||
m := bean.(*Mirror) |
||||
if m.Repo == nil { |
||||
log.Error(4, "Disconnected mirror repository found: %d", m.ID) |
||||
return nil |
||||
} |
||||
|
||||
MirrorQueue.Add(m.RepoID) |
||||
return nil |
||||
}); err != nil { |
||||
log.Error(4, "MirrorUpdate: %v", err) |
||||
} |
||||
} |
||||
|
||||
// SyncMirrors checks and syncs mirrors.
|
||||
// TODO: sync more mirrors at same time.
|
||||
func SyncMirrors() { |
||||
// Start listening on new sync requests.
|
||||
for repoID := range MirrorQueue.Queue() { |
||||
log.Trace("SyncMirrors [repo_id: %v]", repoID) |
||||
MirrorQueue.Remove(repoID) |
||||
|
||||
m, err := GetMirrorByRepoID(com.StrTo(repoID).MustInt64()) |
||||
if err != nil { |
||||
log.Error(4, "GetMirrorByRepoID [%d]: %v", repoID, err) |
||||
continue |
||||
} |
||||
|
||||
if !m.runSync() { |
||||
continue |
||||
} |
||||
|
||||
m.ScheduleNextUpdate() |
||||
if err = UpdateMirror(m); err != nil { |
||||
log.Error(4, "UpdateMirror [%d]: %v", repoID, err) |
||||
continue |
||||
} |
||||
} |
||||
} |
||||
|
||||
func InitSyncMirrors() { |
||||
go SyncMirrors() |
||||
} |
||||
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue