proxycheck/cli.go at main · memclutter/proxycheck · GitHub

History

78 lines (67 loc) · 1.77 KB

Raw

package proxycheck

import (

"fmt"

cli "github.com/urfave/cli/v2"

"os"

"strings"

"sync"

)

// NewApp builds the proxycheck CLI application. It is shared by the cmd binary

// and the tests so both exercise the same flag wiring.

func NewApp() *cli.App {

return &cli.App{

Name: "proxycheck",

Description: "Proxy checker tool",

Version: "0.0.6",

Flags: []cli.Flag{

&cli.StringFlag{Name: "judge", Usage: "Set judge", Value: "proxyjudge.us"},

&cli.IntFlag{Name: "threads", Usage: "Count of threads", Value: 10},

Action: Action,

}

func Action(c *cli.Context) error {

// Resolve the judge from the --judge flag before doing any work; an unknown

// name aborts here, so no proxy is checked.

judge, err := ResolveJudge(c.String("judge"))

if err != nil {

return err

}

// If pass args, use it or read stdin as input file with proxies

var feed Feed

if c.Args().Len() > 0 {

feed = NewSliceFeed(c.Args().Slice())

} else {

feed = NewFileFeed(os.Stdin)

}

// Worker pool

poolSize := c.Int("threads")

proxyAddrs := make(chan string, poolSize)

var wg sync.WaitGroup

for i := 0; i < poolSize; i++ {

wg.Add(1)

go func(proxyAddrs chan string) {

defer wg.Done()

for proxyAddr := range proxyAddrs {

if res := Check(proxyAddr, judge); res.Online {

fmt.Printf("%s\t%s\t%s\n", proxyAddr, strings.Join(res.Protocols, ","), res.Speed.String())

} else {

fmt.Fprintf(os.Stderr, "invalid proxy %s: %v\n", proxyAddr, res.Err)

}

}(proxyAddrs)

}

// Start one thread proxy check

for {

proxyAddr, err := feed.Next()

if err == FeedEnd {

break

}

proxyAddrs <- proxyAddr

}

// Closing the channel lets the workers' range loops finish so wg.Wait

// returns once the feed is exhausted.

close(proxyAddrs)

wg.Wait()

return nil

}

proxycheck/cli.go at main · memclutter/proxycheck · GitHub

Expand file tree

Breadcrumbs

cli.go

Latest commit

History

Breadcrumbs

cli.go

File metadata and controls