func TestPitchfork(t *testing.T) {
var (
config *Config
zk *Zookeeper
p *Pitchfork
storelist StoreList
store *Store
pitchforklist PitchforkList
err error
)
if config, err = conf.NewConfig(configFile); err != nil {
t.Errorf("NewConfig(\"%s\") error(%v)", configFile, err)
return
}
if zk, err = myzk.NewZookeeper([]string{"localhost:2181"}, time.Second*1); err != nil {
t.Errorf("NewZookeeper() error(%v)", err)
t.FailNow()
}
p = NewPitchfork(zk, config)
if err = p.Register(); err != nil {
t.Errorf("pitchfork Register() failed, Quit now")
t.FailNow()
}
storelist, _, err = p.WatchGetStores()
if err != nil {
t.Errorf("pitchfork WatchGetStores() failed, Quit now")
t.FailNow()
}
for _, store = range storelist {
fmt.Println(store.rack, store.ID, store.host, store.status)
}
pitchforklist, _, err = p.WatchGetPitchforks()
if err != nil {
t.Errorf("pitchfork WatchGetPitchforks() failed, Quit now")
t.FailNow()
}
for _, p = range pitchforklist {
fmt.Println(p.ID)
}
for _, store = range storelist {
if err = p.getStore(store); err != nil {
t.Errorf("probeStore() called error(%v)", err)
t.FailNow()
}
}
}
func main() {
var (
config *conf.Config
p *Pitchfork
err error
)
flag.Parse()
defer log.Flush()
log.Infof("bfs pitchfork start")
if config, err = conf.NewConfig(configFile); err != nil {
log.Errorf("NewConfig(\"%s\") error(%v)", configFile, err)
return
}
log.Infof("register pitchfork...")
if p, err = NewPitchfork(config); err != nil {
log.Errorf("pitchfork NewPitchfork() failed, Quit now")
return
}
log.Infof("starts probe stores...")
go p.Probe()
StartSignal()
return
}