go_seckill/main.go

package main

import (
	"errors"
	"fmt"
	"github.com/Albert-Zhan/httpc"
	"github.com/tidwall/gjson"
	"github.com/ztino/jd_seckill/common"
	"github.com/ztino/jd_seckill/conf"
	"github.com/ztino/jd_seckill/jd_seckill"
	"log"
	"net/http"
	"os"
	"runtime"
	"sync"
	"time"
)

var client *httpc.HttpClient

var cookieJar *httpc.CookieJar

var config *conf.Config

var wg *sync.WaitGroup

func init()  {
	//客户端设置初始化
	client=httpc.NewHttpClient()
	cookieJar=httpc.NewCookieJar()
	client.SetCookieJar(cookieJar)
	client.SetRedirect(func(req *http.Request, via []*http.Request) error {
		return http.ErrUseLastResponse
	})
	//配置文件初始化
	confFile:="./conf.ini"
	if !common.Exists(confFile) {
		log.Println("配置文件不存在，程序退出")
		os.Exit(0)
	}
	config=&conf.Config{}
	config.InitConfig(confFile)

	wg=new(sync.WaitGroup)
	wg.Add(1)
}

func main()  {
	runtime.GOMAXPROCS(runtime.NumCPU())

	//用户登录
	user:=jd_seckill.NewUser(client,config)
	wlfstkSmdl,err:=user.QrLogin()
	if err!=nil{
		os.Exit(0)
	}
	ticket:=""
	for  {
		ticket,err=user.QrcodeTicket(wlfstkSmdl)
		if err==nil && ticket!=""{
			break
		}
		time.Sleep(2*time.Second)
	}
	_,err=user.TicketInfo(ticket)
	if err==nil {
		log.Println("登录成功")
		//刷新用户状态和获取用户信息
		if status:=user.RefreshStatus();status==nil {
			userInfo,_:=user.GetUserInfo()
			log.Println("用户:"+userInfo)
			//开始预约,预约过的就重复预约
			seckill:=jd_seckill.NewSeckill(client,config)
			seckill.MakeReserve()
			//等待抢购/开始抢购
			nowLocalTime:=time.Now().UnixNano()/1e6
			jdTime,_:=getJdTime()
			buyDate:=config.Read("config","buy_time")
			loc, _ := time.LoadLocation("Local")
			t,_:=time.ParseInLocation("2006-01-02 15:04:05",buyDate,loc)
			buyTime:=t.UnixNano()/1e6
			diffTime:=nowLocalTime-jdTime
			log.Println(fmt.Sprintf("正在等待到达设定时间:%s，检测本地时间与京东服务器时间误差为【%d】毫秒",buyDate,diffTime))
			timerTime:=(buyTime+diffTime)-jdTime
			if timerTime<=0 {
				log.Println("请设置抢购时间")
				os.Exit(0)
			}
			time.Sleep(time.Duration(timerTime)*time.Millisecond)
			//开启任务
			log.Println("时间到达，开始执行……")
			start(seckill,5)
			wg.Wait()
		}
	}else{
		log.Println("登录失败")
	}
}

func getJdTime() (int64,error) {
	req:=httpc.NewRequest(client)
	resp,body,err:=req.SetUrl("https://a.jd.com//ajax/queryServerData.html").SetMethod("get").Send().End()
	if err!=nil || resp.StatusCode!=http.StatusOK {
		log.Println("获取京东服务器时间失败")
		return 0,errors.New("获取京东服务器时间失败")
	}
	return gjson.Get(body,"serverTime").Int(),nil
}

func start(seckill *jd_seckill.Seckill,taskNum int)  {
	for i:=1;i<=taskNum;i++ {
		go func(seckill *jd_seckill.Seckill) {
			seckill.RequestSeckillUrl()
			seckill.SeckillPage()
			seckill.SubmitSeckillOrder()
		}(seckill)
	}
}