package main import ( "flag" "fmt" "os" "ouchn/config" "ouchn/student" "runtime" "strings" "sync" "time" "github.com/tebeka/selenium" ) var ( configFile = flag.String("config", "./common.yaml", "config file location") version = flag.Bool("version", false, "print the version") GitCommit = "library-import" Version = "library-import" ) /*func handleCount(account []string) { fmt.Printf("账号:%v\n", account) service, wd, err := student.StartOneStudent(account[0], account[1], account[2]) wd.Quit() service.Stop() if err != nil { fmt.Printf("%s %v \n", account[2], err) return } ioutil.WriteFile("record.txt", []byte(fmt.Sprintf("%s %s %s",account[0], account[1], account[2])), 0666) }*/ func writeRecord(filename string, data []byte) { fi, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE, 0666) if err != nil { return } defer fi.Close() fi.Write(data) } func stest() { service, wd, err := student.StartOneStudent("2151101200269", "Ouchn@2021", "方") //service, wd, err := student.StartOneStudent("2251101200243", "Ouchn@2021", "肖") if wd != nil { wd.Quit() } if service != nil { service.Stop() } if err != nil { fmt.Printf("处理账号失败:%v \n", err) return } } func main() { os.Rename("error.txt", fmt.Sprintf("error_%s.txt", time.Now().Format("2006-01-02 15:04:05"))) err := config.LoadConfig(*configFile) if err != nil { panic(err) } student.TokenDbOpen() student.TokenDbClear() defer func() { student.TokenDbClose() }() fmt.Printf("target:%d\n", config.Conf.Target) //stest() // 取账号 for { accounts, _ := student.GetAcounts("account.txt") //accounts = accounts[0:1] //accounts[0][0] = "2051101250176" //accounts[0][1] = "Ouchn@2021" //accounts[0][2] = "将" fmt.Printf("accounts:%v\n", len(accounts)) if len(accounts) == 0 { return } for _, v := range accounts { if len(v) != 3 { fmt.Printf("账号不完整\n") return } } count := 0 var wg sync.WaitGroup serviceArray := []*selenium.Service{} wds := []selenium.WebDriver{} //accounts = [][]string{{"2151101451735", "Ouchn19951103", "罗建琴"}} for k, _ := range accounts { wg.Add(1) count++ go func(account []string) { defer wg.Done() fmt.Printf("账号:%v\n", account) service, wd, err := student.StartOneStudent(account[0], account[1], account[2]) if wd != nil { wd.Quit() wds = append(wds, wd) } if service != nil { //service.Stop() serviceArray = append(serviceArray, service) } if err != nil { fmt.Printf("处理账号%s失败:%v \n", account[2], err) if strings.Contains(err.Error(), "题") || strings.Contains(err.Error(), "panic") { writeRecord("error.txt", []byte(fmt.Sprintf("%s %s %s;%s\n", account[0], account[1], account[2], err.Error()))) } return } writeRecord("record.txt", []byte(fmt.Sprintf("%s %s %s\n", account[0], account[1], account[2]))) }(accounts[k]) //fmt.Printf("count:%d,%d\n", count, config.Conf.Channel) //go handleCount(accounts[k]) if count == config.Conf.Channel { wg.Wait() /*for _, v := range wds { v.Quit() }*/ for _, v := range serviceArray { v.Stop() } serviceArray = []*selenium.Service{} wds = []selenium.WebDriver{} count = 0 } } if count != 0 { wg.Wait() /*for _, v := range wds { v.Quit() }*/ for _, v := range serviceArray { v.Stop() } serviceArray = []*selenium.Service{} wds = []selenium.WebDriver{} } fmt.Printf("等待执行下一批账号\n") runtime.GC() time.Sleep(5 * time.Second) } }