Beispiel #1
0
func initManager(mgr_cfg *ManagerConfig) (adaptor.Manager, error) {
	options := make([]func(*manager.Manager), 0)

	if mgr_cfg.Connection_limit != 0 {
		limit := mgr_cfg.Connection_limit
		options = append(options, manager.SetConnectionLimit(limit))
	}

	if mgr_cfg.Connection_rate != 0 {
		rate := time.Second / time.Duration(mgr_cfg.Connection_rate)
		options = append(options, manager.SetConnectionRate(rate))
	}

	if mgr_cfg.Protocol_magic != 0 {
		magic := wire.BitcoinNet(mgr_cfg.Protocol_magic)
		options = append(options, manager.SetProtocolMagic(magic))
	}

	if mgr_cfg.Protocol_version != 0 {
		version := mgr_cfg.Protocol_version
		options = append(options, manager.SetProtocolVersion(version))
	}

	if mgr_cfg.Ticker_interval != 0 {
		interval := time.Second * time.Duration(mgr_cfg.Ticker_interval)
		options = append(options, manager.SetTickerInterval(interval))
	}

	return manager.New(options...)
}
Beispiel #2
0
func New() (*Supervisor, error) {
	// load configuration file
	cfg := &Config{}
	err := gcfg.ReadFileInto(cfg, "pbtc.cfg")
	if err != nil {
		return nil, err
	}

	// initialize struct with maps
	supervisor := &Supervisor{
		logr: make(map[string]adaptor.Logger),
		repo: make(map[string]adaptor.Repository),
		tkr:  make(map[string]adaptor.Tracker),
		svr:  make(map[string]adaptor.Server),
		pro:  make(map[string]adaptor.Processor),
		mgr:  make(map[string]adaptor.Manager),
	}

	if len(cfg.Logger) == 0 {
		logr, err := logger.New()
		if err != nil {
			return nil, err
		}

		supervisor.logr[""] = logr

	} else if cfg.Logger[""] == nil {
		for _, logr_cfg := range cfg.Logger {
			logr, err := initLogger(logr_cfg)
			if err != nil {
				return nil, err
			}

			supervisor.logr[""] = logr
			break
		}
	} else {
		logr_cfg := cfg.Logger[""]
		logr, err := initLogger(logr_cfg)
		if err != nil {
			return nil, err
		}

		supervisor.logr[""] = logr
	}

	level, err := logger.ParseLevel(cfg.Supervisor.Log_level)
	if err != nil {
		level = logging.CRITICAL
	}

	supervisor.log = supervisor.logr[""].GetLog("supervisor")
	supervisor.logr[""].SetLevel("supervisor", level)
	supervisor.log.Info("[SUP] Init: started")
	supervisor.log.Info("[SUP] Init: default logger initialized")
	supervisor.log.Info("[SUP] Init: initializing modules")

	// initialize remaining modules
	for name, logr_cfg := range cfg.Logger {
		if name == "" {
			continue
		}

		logr, err := initLogger(logr_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: logger init failed (%v)", err)
			continue
		}

		supervisor.logr[name] = logr
	}

	for name, repo_cfg := range cfg.Repository {
		repo, err := initRepository(repo_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: repo init failed (%v)", err)
			continue
		}

		supervisor.repo[name] = repo
	}

	for name, tkr_cfg := range cfg.Tracker {
		tkr, err := initTracker(tkr_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: tracker init failed (%v)", err)
			continue
		}

		supervisor.tkr[name] = tkr
	}

	for name, svr_cfg := range cfg.Server {
		svr, err := initServer(svr_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: server init failed (%v)", err)
			continue
		}

		supervisor.svr[name] = svr
	}

	for name, pro_cfg := range cfg.Processor {
		pro, err := initProcessor(pro_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: proc init failed (%v)", err)
			continue
		}

		supervisor.pro[name] = pro
	}

	for name, mgr_cfg := range cfg.Manager {
		mgr, err := initManager(mgr_cfg)
		if err != nil {
			supervisor.log.Warning("[SUP] Init: manager init failed (%v)", err)
			continue
		}

		supervisor.mgr[name] = mgr
	}

	supervisor.log.Info("[SUP] Init: checking module cardinality")

	// check remaining modules for missing values
	if len(supervisor.repo) == 0 {
		supervisor.log.Warning("[SUP] Init: missing repository module")
		repo, err := repository.New()
		if err != nil {
			return nil, err
		}

		supervisor.repo["default"] = repo
	}

	if len(supervisor.tkr) == 0 {
		supervisor.log.Warning("[SUP] Init: missing tracker module")
		tkr, err := tracker.New()
		if err != nil {
			return nil, err
		}

		supervisor.tkr["default"] = tkr
	}

	if len(supervisor.mgr) == 0 {
		supervisor.log.Warning("[SUP] Init: missing manager module")
		mgr, err := manager.New()
		if err != nil {
			return nil, err
		}

		supervisor.mgr["default"] = mgr
	}

	if len(supervisor.svr) == 0 {
		supervisor.log.Notice("[SUP] Init: no server module")
	}

	if len(supervisor.pro) == 0 {
		supervisor.log.Notice("[SUP] Init: no processor module")
	}

	supervisor.log.Info("[SUP] Init: injecting logging capabilities")

	// inject logging dependencies
	for key, logr := range supervisor.logr {
		logr_cfg, ok := cfg.Logger[key]
		if !ok {
			continue
		}

		level, err := logger.ParseLevel(logr_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "logr___" + key
		logr.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	for key, repo := range supervisor.repo {
		repo_cfg, ok := cfg.Repository[key]
		if !ok {
			continue
		}

		logr, ok := supervisor.logr[repo_cfg.Logger]
		if !ok {
			logr = supervisor.logr[""]
		}

		level, err := logger.ParseLevel(repo_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "repo___" + key
		repo.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	for key, tkr := range supervisor.tkr {
		tkr_cfg, ok := cfg.Tracker[key]
		if !ok {
			continue
		}

		logr, ok := supervisor.logr[tkr_cfg.Logger]
		if !ok {
			logr = supervisor.logr[""]
		}

		level, err := logger.ParseLevel(tkr_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "tkr___" + key
		tkr.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	for key, svr := range supervisor.svr {
		svr_cfg, ok := cfg.Server[key]
		if !ok {
			continue
		}

		logr, ok := supervisor.logr[svr_cfg.Logger]
		if !ok {
			logr = supervisor.logr[""]
		}

		level, err := logger.ParseLevel(svr_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "svr___" + key
		svr.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	for key, pro := range supervisor.pro {
		pro_cfg, ok := cfg.Processor[key]
		if !ok {
			continue
		}

		logr, ok := supervisor.logr[pro_cfg.Logger]
		if !ok {
			logr = supervisor.logr[""]
		}

		level, err := logger.ParseLevel(pro_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "pro___" + key
		pro.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	for key, mgr := range supervisor.mgr {
		mgr_cfg, ok := cfg.Manager[key]
		if !ok {
			continue
		}

		logr, ok := supervisor.logr[mgr_cfg.Logger]
		if !ok {
			logr = supervisor.logr[""]
		}

		level, err := logger.ParseLevel(mgr_cfg.Log_level)
		if err != nil {
			level = logging.CRITICAL
		}

		log := "mgr___" + key
		mgr.SetLog(logr.GetLog(log))
		logr.SetLevel(log, level)
	}

	supervisor.log.Info("[SUP] Init: injecting module dependencies")

	// inject manager into server
	for key, svr := range supervisor.svr {
		svr_cfg, ok := cfg.Server[key]
		if !ok {
			continue
		}

		mgr, ok := supervisor.mgr[svr_cfg.Manager]
		if !ok {
			for _, def := range supervisor.mgr {
				mgr = def
				break
			}
		}

		svr.SetManager(mgr)
	}

	// inject repository into manager
	for key, mgr := range supervisor.mgr {
		mgr_cfg, ok := cfg.Manager[key]
		if !ok {
			continue
		}

		repo, ok := supervisor.repo[mgr_cfg.Repository]
		if !ok {
			for _, def := range supervisor.repo {
				repo = def
				break
			}
		}

		mgr.SetRepository(repo)
	}

	// inject tracker into manager
	for key, mgr := range supervisor.mgr {
		mgr_cfg, ok := cfg.Manager[key]
		if !ok {
			continue
		}

		tkr, ok := supervisor.tkr[mgr_cfg.Tracker]
		if !ok {
			for _, def := range supervisor.tkr {
				tkr = def
				break
			}
		}

		mgr.SetTracker(tkr)
	}

	// inject processors into managers
	for key, mgr := range supervisor.mgr {
		mgr_cfg, ok := cfg.Manager[key]
		if !ok {
			continue
		}

		for _, name := range mgr_cfg.Processor {
			pro, ok := supervisor.pro[name]
			if !ok {
				continue
			}

			mgr.AddProcessor(pro)
		}
	}

	// inject processors into processors
	for key, pro := range supervisor.pro {
		pro_cfg, ok := cfg.Processor[key]
		if !ok {
			continue
		}

		for _, name := range pro_cfg.Next {
			next, ok := supervisor.pro[name]
			if !ok {
				continue
			}

			pro.AddNext(next)
		}
	}

	supervisor.log.Info("[SUP] Init: completed")

	return supervisor, nil
}