WarlockFish/proxypool
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
zu1k
2020-08-23 17:32:55 +08:00
parent 32c07bcd60
commit 8ca93891ff
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/app/getter.go
View File

@@ -28,21 +28,23 @@ func InitConfigAndGetters(path string) (err error) {
func initGetters(sourceFiles []string) {
Getters = make([]getter.Getter, 0)
for _, path := range sourceFiles {
source := config.Source{}
data, err := config.ReadFile(path)
if err != nil {
fmt.Errorf("Init SourceFile Error: %s\n", err.Error())
continue
}
err = yaml.Unmarshal(data, &source)
sourceList := make([]config.Source, 0)
err = yaml.Unmarshal(data, &sourceList)
if err != nil {
fmt.Errorf("Init SourceFile Error: %s\n", err.Error())
continue
}
g, err := getter.NewGetter(source.Type, source.Options)
if err == nil && g != nil {
Getters = append(Getters, g)
fmt.Println("init getter:", source.Type, source.Options)
for _, source := range sourceList {
g, err := getter.NewGetter(source.Type, source.Options)
if err == nil && g != nil {
Getters = append(Getters, g)
fmt.Println("init getter:", source.Type, source.Options)
}
}
}
fmt.Println("Getter count:", len(Getters))