Commit 389f14c3 authored by vicotor's avatar vicotor

update scheduler

parent 4b3eff99
......@@ -2,6 +2,7 @@ metrics_port = 28012
routines = 1
max_nm_update_ex = 40
dispatch_timeout = 3
poper_version = "v1"
[mongodb]
url="mongodb://127.0.0.1:27017"
......
......@@ -37,6 +37,7 @@ type Config struct {
Routines int `json:"routines" toml:"routines"`
MaxNmUpdateEx int `json:"max_nm_update_ex" toml:"max_nm_update_ex"`
DispatchTimeout int `json:"dispatch_timeout" toml:"dispatch_timeout"`
PoperVersion string `json:"poper_version" toml:"poper_version`
Redis RedisConfig `json:"redis" toml:"redis"`
Kafka KafkaConfig `json:"kafka" toml:"kafka"`
DbConfig MysqlConfig `json:"mysql" toml:"mysql"`
......
package apibackend
type Backend interface {
BloomStatus() (uint64, uint64)
}
type nodeBackend struct {
}
func (b *nodeBackend) BloomStatus() (uint64, uint64) {
return 0, 0
}
func NewBackend() Backend {
return &nodeBackend{}
}
......@@ -4,6 +4,7 @@ import (
"context"
"errors"
odysseus "github.com/odysseus/odysseus-protocol/gen/proto/go/base/v1"
"github.com/odysseus/scheduler/config"
"github.com/odysseus/scheduler/types"
"github.com/redis/go-redis/v9"
log "github.com/sirupsen/logrus"
......@@ -21,10 +22,29 @@ type WorkerPoper interface {
}
func NewPopWorker() WorkerPoper {
v2, err := newPoperV2()
if err != nil {
log.WithField("err", err).Error("failed to create poperV2")
return nil
switch config.GetConfig().PoperVersion {
case "v1":
v1, err := newPoperV1()
if err != nil {
log.WithFields(log.Fields{
"version": "v1",
"err": err,
}).Error("failed to create poper")
return nil
}
return v1
case "v2":
v2, err := newPoperV2()
if err != nil {
log.WithFields(log.Fields{
"version": "v2",
"err": err,
}).Error("failed to create poper")
return nil
}
return v2
default:
log.WithField("version", config.GetConfig().PoperVersion).Error("unknown poper version")
}
return v2
return nil
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment