mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-08-28 10:54:59 +00:00
903e234c69
* tweak a bunch of logging and config make overall logs less verbose and chatty 1 minute computer check-in interval 3 minute snapshot interval remove CaptureLogger as it has same functionality as buffer logger add a WithPrefix to the Logger interface so sub-services can have different prefixes * fix some lint * fix lint... confused why this is coming up now
119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
package service
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/featurebasedb/featurebase/v3/dax"
|
|
"github.com/featurebasedb/featurebase/v3/dax/boltdb"
|
|
"github.com/featurebasedb/featurebase/v3/dax/controller"
|
|
balancerboltdb "github.com/featurebasedb/featurebase/v3/dax/controller/balancer/boltdb"
|
|
controllerhttp "github.com/featurebasedb/featurebase/v3/dax/controller/http"
|
|
schemarboltdb "github.com/featurebasedb/featurebase/v3/dax/controller/schemar/boltdb"
|
|
"github.com/featurebasedb/featurebase/v3/errors"
|
|
"github.com/featurebasedb/featurebase/v3/logger"
|
|
fbnet "github.com/featurebasedb/featurebase/v3/net"
|
|
)
|
|
|
|
// Ensure type implements interface.
|
|
var _ dax.Service = (*controllerService)(nil)
|
|
|
|
type controllerService struct {
|
|
uri *fbnet.URI
|
|
controller *controller.Controller
|
|
|
|
// Because we stopped using a storage method interface, and always use bolt,
|
|
// we need to be sure to close the boltDBs that are created in controller.New()
|
|
// whenever controller.Stop() is called. These are pointers to that DB so we can
|
|
// close it.
|
|
boltDB *boltdb.DB
|
|
|
|
logger logger.Logger
|
|
}
|
|
|
|
func New(uri *fbnet.URI, cfg controller.Config) *controllerService {
|
|
// Set up logger.
|
|
var logr logger.Logger = logger.StderrLogger
|
|
if cfg.Logger != nil {
|
|
logr = cfg.Logger.WithPrefix("Controller: ")
|
|
}
|
|
|
|
// Storage methods.
|
|
if cfg.StorageMethod != "boltdb" && cfg.StorageMethod != "" {
|
|
logr.Printf("storagemethod %s not supported, try 'boltdb'", cfg.StorageMethod)
|
|
}
|
|
|
|
if cfg.DataDir == "" {
|
|
dir, err := os.MkdirTemp("", "controller_*")
|
|
if err != nil {
|
|
logr.Printf("Making temp dir for Controller storage: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
cfg.DataDir = dir
|
|
logr.Warnf("no DataDir given (like '/path/to/directory'); using temp dir at '%s'", cfg.DataDir)
|
|
}
|
|
|
|
buckets := append(schemarboltdb.SchemarBuckets, balancerboltdb.BalancerBuckets...)
|
|
|
|
controllerDB, err := boltdb.NewSvcBolt(cfg.DataDir, "controller", buckets...)
|
|
if err != nil {
|
|
logr.Printf(errors.Wrap(err, "creating controller bolt").Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
schemar := schemarboltdb.NewSchemar(controllerDB, logr)
|
|
balancer := balancerboltdb.NewBalancer(controllerDB, schemar, logr)
|
|
|
|
// Directive version.
|
|
if err := controllerDB.InitializeBuckets(boltdb.DirectiveBuckets...); err != nil {
|
|
logr.Panicf("initializing directive buckets: %v", err)
|
|
}
|
|
directiveVersion := boltdb.NewDirectiveVersion(controllerDB)
|
|
|
|
// Controller.
|
|
controller := controller.New(cfg)
|
|
controller.Schemar = schemar
|
|
controller.Balancer = balancer
|
|
controller.DirectiveVersion = directiveVersion
|
|
controller.BoltDB = controllerDB
|
|
|
|
if cfg.Director != nil {
|
|
controller.Director = cfg.Director
|
|
}
|
|
|
|
return &controllerService{
|
|
uri: uri,
|
|
controller: controller,
|
|
boltDB: controllerDB,
|
|
logger: logr,
|
|
}
|
|
}
|
|
|
|
func (m *controllerService) Start() error {
|
|
// Start controller service.
|
|
if err := m.controller.Start(); err != nil {
|
|
return errors.Wrap(err, "starting controller")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *controllerService) Stop() error {
|
|
err := m.controller.Stop()
|
|
if err != nil {
|
|
m.logger.Warnf("error stopping controller: %v", err)
|
|
}
|
|
|
|
if m.boltDB != nil {
|
|
m.boltDB.Close()
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (m *controllerService) Address() dax.Address {
|
|
return dax.Address(m.uri.HostPort() + "/" + dax.ServicePrefixController)
|
|
}
|
|
|
|
func (m *controllerService) HTTPHandler() http.Handler {
|
|
return controllerhttp.Handler(m.controller)
|
|
}
|