You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
2.9 KiB
118 lines
2.9 KiB
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)
|
|
}
|
|
}
|