overflow_backup_2018_0455/long_poller_go
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

poller

Browse Source
This commit is contained in:
insanity@loafle.com 2017-05-16 16:11:50 +09:00
parent 447a3b372b
commit 41d98d4b91
1 changed files with 49 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
poller.go
View File

@ -1,10 +1,8 @@
package long_poller_go
import (
"context"
"google.golang.org/grpc"
cm "loafle.com/overflow/agent_api/config_manager"
pb "loafle.com/overflow/crawler_go/grpc" //temp
//pb "loafle.com/overflow/crawler_go/grpc" //temp
s "loafle.com/overflow/scheduler_go"
"log"
"strconv"
@ -27,22 +25,20 @@ type Poller struct {
taskCh chan interface{}
}
func Start(ch chan string, conf *cm.GlobalConfig) (chan interface{}, error) {
func Start(ch chan bool, conf *cm.GlobalConfig) error {
p := GetInstance()
p.gconf = conf
p.startPolling()
ch <- true
return nil
}
func GetTaskCh() chan interface{} {
p := GetInstance()
p.taskCh = make(chan interface{})
err := p.startPolling(conf)
if err != nil {
return nil, err
}
key := getSecretKey() //todo. getting api key
ch <- key
return p.taskCh, nil
return p.taskCh
}
func getSecretKey() string {
return "tempkey"
}
func Stop(polStopped chan bool) error {
GetInstance().stop()
@ -63,39 +59,40 @@ func UpdateInterval(interval string) error {
return err
}
GetInstance().updateInterval(v)
}
func (p *Poller) startPolling(conf *cm.GlobalConfig) error {
p.gconf = conf
p.scheduler = &s.Scheduler{}
p.scheduler.Start()
err := p.scheduler.NewSchedule(POLLING_ID, uint64(p.gconf.IntervalSecond), p.polling)
if err != nil {
return err
}
return nil
}
func (p *Poller) handleInterval(ch chan interface{}) {
func (p *Poller) startPolling() {
go func() {
data := <-ch
interval := data.(string)
i, err := strconv.Atoi(interval)
if err != nil {
log.Println(err)
return
}
p.scheduler.UpdateInterval(POLLING_ID, uint64(i))
p.scheduler = &s.Scheduler{}
p.scheduler.Start()
p.scheduler.NewSchedule(POLLING_ID, uint64(p.gconf.IntervalSecond), p.polling)
}()
}
//func (p *Poller) handleInterval(ch chan interface{}) {
// go func() {
// data := <-ch
// interval := data.(string)
// i, err := strconv.Atoi(interval)
// if err != nil {
// log.Println(err)
// return
// }
// p.scheduler.UpdateInterval(POLLING_ID, uint64(i))
// }()
//}
func (p *Poller) stop() {
p.scheduler.Stop()
}
func (p *Poller) polling(agentId string) {
go func() {
log.Println("POLLING....")
/*
addr := p.gconf.Central.Address + ":" + string(p.gconf.Central.Port)
conn, err := grpc.Dial(addr, grpc.WithInsecure())
if err != nil {
@ -113,6 +110,8 @@ func (p *Poller) polling(agentId string) {
}
p.taskCh <- out
*/
}()
}
func (p *Poller) updateInterval(interval int) {