yacloud/tunasync
1
0
Fork 0
mirror of https://github.com/tuna/tunasync.git synced 2025-04-21 04:42:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(worker): added Type() method to provider

Browse Source
This commit is contained in:
bigeagle 2016-05-08 16:26:58 +08:00
parent c25e508a00
commit ccc31d9289
No known key found for this signature in database
GPG Key ID: 9171A4571C27920A
5 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
worker/cmd_provider.go
View File

@ -44,6 +44,10 @@ func newCmdProvider(c cmdConfig) (*cmdProvider, error) {
return provider, nil
}
func (p *cmdProvider) Type() providerEnum {
return provCommand
}
func (p *cmdProvider) Upstream() string {
return p.upstreamURL
}

4
worker/provider.go
View File

@ -10,8 +10,6 @@ import (
// mirror provider is the wrapper of mirror jobs
type providerType uint8
const (
_WorkingDirKey = "working_dir"
_LogDirKey = "log_dir"
@ -24,6 +22,8 @@ type mirrorProvider interface {
Name() string
Upstream() string
Type() providerEnum
// run mirror job in background
Run() error
// run mirror job in background

3
worker/provider_test.go
View File

@ -33,6 +33,7 @@ func TestRsyncProvider(t *testing.T) {
provider, err := newRsyncProvider(c)
So(err, ShouldBeNil)
So(provider.Type(), ShouldEqual, provRsync)
So(provider.Name(), ShouldEqual, c.name)
So(provider.WorkingDir(), ShouldEqual, c.workingDir)
So(provider.LogDir(), ShouldEqual, c.logDir)
@ -126,6 +127,7 @@ func TestCmdProvider(t *testing.T) {
provider, err := newCmdProvider(c)
So(err, ShouldBeNil)
So(provider.Type(), ShouldEqual, provCommand)
So(provider.Name(), ShouldEqual, c.name)
So(provider.WorkingDir(), ShouldEqual, c.workingDir)
So(provider.LogDir(), ShouldEqual, c.logDir)
@ -218,6 +220,7 @@ func TestTwoStageRsyncProvider(t *testing.T) {
provider, err := newTwoStageRsyncProvider(c)
So(err, ShouldBeNil)
So(provider.Type(), ShouldEqual, provTwoStageRsync)
So(provider.Name(), ShouldEqual, c.name)
So(provider.WorkingDir(), ShouldEqual, c.workingDir)
So(provider.LogDir(), ShouldEqual, c.logDir)

4
worker/rsync_provider.go
View File

@ -63,6 +63,10 @@ func newRsyncProvider(c rsyncConfig) (*rsyncProvider, error) {
return provider, nil
}
func (p *rsyncProvider) Type() providerEnum {
return provRsync
}
func (p *rsyncProvider) Upstream() string {
return p.upstreamURL
}

4
worker/two_stage_rsync_provider.go
View File

@ -70,6 +70,10 @@ func newTwoStageRsyncProvider(c twoStageRsyncConfig) (*twoStageRsyncProvider, er
return provider, nil
}
func (p *twoStageRsyncProvider) Type() providerEnum {
return provTwoStageRsync
}
func (p *twoStageRsyncProvider) Upstream() string {
return p.upstreamURL
}