2015-03-04 19:08:36 +00:00
|
|
|
package task
|
|
|
|
|
|
|
|
import (
|
2015-03-06 13:23:01 +00:00
|
|
|
"log"
|
|
|
|
"time"
|
|
|
|
|
2015-03-04 19:08:36 +00:00
|
|
|
"code.google.com/p/goauth2/oauth"
|
|
|
|
"github.com/google/go-github/github"
|
2015-03-05 08:07:18 +00:00
|
|
|
"github.com/localhots/empact/db"
|
2015-03-04 19:08:36 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func newGithubClient(token string) *github.Client {
|
|
|
|
trans := &oauth.Transport{
|
|
|
|
Token: &oauth.Token{AccessToken: token},
|
|
|
|
}
|
|
|
|
return github.NewClient(trans.Client())
|
|
|
|
}
|
2015-03-04 19:15:05 +00:00
|
|
|
|
2015-03-04 20:14:14 +00:00
|
|
|
func saveResponseMeta(token string, res *github.Response) {
|
|
|
|
if res == nil {
|
|
|
|
return
|
|
|
|
}
|
2015-03-05 15:04:44 +00:00
|
|
|
tok := &db.Token{
|
2015-03-04 20:14:14 +00:00
|
|
|
Token: token,
|
2015-03-06 11:18:15 +00:00
|
|
|
Quota: uint64(res.Limit),
|
|
|
|
Remaining: uint64(res.Remaining),
|
2015-03-04 20:14:14 +00:00
|
|
|
ResetAt: res.Reset.Time,
|
2015-03-05 15:04:44 +00:00
|
|
|
}
|
|
|
|
tok.Save()
|
2015-03-04 20:14:14 +00:00
|
|
|
}
|
2015-03-06 13:23:01 +00:00
|
|
|
|
|
|
|
func report(task string, start time.Time) {
|
|
|
|
duration := time.Since(start).Nanoseconds()
|
2015-03-06 13:35:13 +00:00
|
|
|
outcome := "succeeded"
|
2015-03-06 13:23:01 +00:00
|
|
|
if err := recover(); err != nil {
|
|
|
|
outcome = "failed"
|
|
|
|
}
|
|
|
|
|
|
|
|
log.Printf("Task %s %s; time: %d (%dms)\n", task, outcome, duration, duration/1000000)
|
|
|
|
}
|