Sansui233_proxypool/pkg/proxy/proxies.go

package proxy

import (
	"fmt"
	"sort"
)

type ProxyList []Proxy

func (ps ProxyList) Len() int {
	return len(ps)
}

func (ps ProxyList) TypeLen(t string) int {
	l := 0
	for _, p := range ps {
		if p.TypeName() == t {
			l++
		}
	}
	return l
}

func (ps ProxyList) Less(i, j int) bool {
	return ps[i].BaseInfo().Name < ps[j].BaseInfo().Name
}

func (ps ProxyList) Swap(i, j int) {
	ps[i], ps[j] = ps[j], ps[i]
}

func (ps ProxyList) Deduplication() ProxyList {
	result := make(ProxyList, 0, len(ps))
	temp := map[string]struct{}{}
	for _, item := range ps {
		if item != nil {
			if _, ok := temp[item.Identifier()]; !ok {
				temp[item.Identifier()] = struct{}{}
				result = append(result, item)
			}
		}
	}
	return result
}

func (ps ProxyList) Sort() ProxyList {
	sort.Sort(ps)
	return ps
}

func (ps ProxyList) NameAddCounrty() ProxyList {
	num := len(ps)
	for i := 0; i < num; i++ {
		country, err := geoIp.Find(ps[i].BaseInfo().Server)
		if err != nil || country == "" {
			country = "Earth"
		}
		ps[i].SetName(fmt.Sprintf("%s", country))
	}
	return ps
}

func (ps ProxyList) NameAddIndex() ProxyList {
	num := len(ps)
	for i := 0; i < num; i++ {
		ps[i].SetName(fmt.Sprintf("%s_%d", ps[i].BaseInfo().Name, i+1))
	}
	return ps
}

func Deduplication(src ProxyList) ProxyList {
	result := make(ProxyList, 0, len(src))
	temp := map[string]struct{}{}
	for _, item := range src {
		if item != nil {
			if _, ok := temp[item.Identifier()]; !ok {
				temp[item.Identifier()] = struct{}{}
				result = append(result, item)
			}
		}
	}
	return result
}

func (ps ProxyList) Clone() ProxyList {
	result := make(ProxyList, 0, len(ps))
	for _, pp := range ps {
		if pp != nil {
			result = append(result, pp.Clone())
		}
	}
	return result
}