88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
|
|
"github.com/zu1k/proxypool/app/cache"
|
|
"github.com/zu1k/proxypool/provider"
|
|
"github.com/zu1k/proxypool/proxy"
|
|
"github.com/zu1k/proxypool/tool"
|
|
)
|
|
|
|
var NeedFetchNewConfigFile = false
|
|
|
|
func Crawl() {
|
|
if NeedFetchNewConfigFile {
|
|
FetchNewConfigFileThenInit()
|
|
}
|
|
proxies := make([]proxy.Proxy, 0)
|
|
for _, g := range Getters {
|
|
proxies = append(proxies, g.Get()...)
|
|
}
|
|
proxies = append(proxies, cache.GetProxies()...)
|
|
proxies = proxy.Deduplication(proxies)
|
|
|
|
num := len(proxies)
|
|
for i := 0; i < num; i++ {
|
|
proxies[i].SetName(strconv.Itoa(rand.Int()))
|
|
}
|
|
log.Println("Crawl node count:", num)
|
|
cache.SetProxies(proxies)
|
|
cache.SetString("clashproxies", provider.Clash{Proxies: proxies}.Provide())
|
|
}
|
|
|
|
func CrawlGo() {
|
|
if NeedFetchNewConfigFile {
|
|
FetchNewConfigFileThenInit()
|
|
}
|
|
wg := &sync.WaitGroup{}
|
|
var pc = make(chan proxy.Proxy)
|
|
for _, g := range Getters {
|
|
wg.Add(1)
|
|
go g.Get2Chan(pc, wg)
|
|
}
|
|
proxies := cache.GetProxies()
|
|
go func() {
|
|
wg.Wait()
|
|
close(pc)
|
|
}()
|
|
for node := range pc {
|
|
if node != nil {
|
|
proxies = append(proxies, node)
|
|
}
|
|
}
|
|
proxies = proxy.Deduplication(proxies)
|
|
|
|
num := len(proxies)
|
|
for i := 0; i < num; i++ {
|
|
proxies[i].SetName(strconv.Itoa(rand.Int()))
|
|
}
|
|
log.Println("CrawlGo node count:", num)
|
|
cache.SetProxies(proxies)
|
|
cache.SetString("clashproxies", provider.Clash{Proxies: proxies}.Provide())
|
|
}
|
|
|
|
func FetchNewConfigFileThenInit() {
|
|
fmt.Println("fetch new config file...")
|
|
resp, err := tool.GetHttpClient().Get("https://raw.githubusercontent.com/zu1k/proxypool/master/source.yaml")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = ioutil.WriteFile("source.yaml", body, os.ModePerm)
|
|
if err != nil {
|
|
return
|
|
}
|
|
InitConfigAndGetters("source.yaml")
|
|
}
|