a2a0b27bc4
-made defer adjustments inline and deleted the separate function. -adjusted method in upload to take the only parameter it requires. -Implemented stopper param for reflector server -Aligned Cluster New to NewCluster -Adjusted DHT to use StopAndWait -Removed blocking waitgroup add -Unified all components under prism. -Moved defer done outside of functions. -renamed NewCluster to New -fixed travis errors.
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package reflector
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/lbryio/lbry.go/stopOnce"
|
|
"github.com/lbryio/reflector.go/cluster"
|
|
"github.com/lbryio/reflector.go/dht"
|
|
"github.com/lbryio/reflector.go/peer"
|
|
"github.com/lbryio/reflector.go/store"
|
|
)
|
|
|
|
// Prism is the root instance of the application and houses the DHT, Peer Server, Reflector Server, and Cluster.
|
|
type Prism struct {
|
|
dht *dht.DHT
|
|
peer *peer.Server
|
|
reflector *Server
|
|
cluster *cluster.Cluster
|
|
|
|
stop *stopOnce.Stopper
|
|
}
|
|
|
|
// NewPrism returns an initialized Prism instance pointer.
|
|
func NewPrism(store store.BlobStore, clusterSeedAddr string) *Prism {
|
|
d, err := dht.New(nil)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return &Prism{
|
|
dht: d,
|
|
peer: peer.NewServer(store),
|
|
reflector: NewServer(store),
|
|
cluster: cluster.New(cluster.DefaultClusterPort, clusterSeedAddr),
|
|
stop: stopOnce.New(),
|
|
}
|
|
}
|
|
|
|
// Start starts the components of the application.
|
|
func (p *Prism) Start() error {
|
|
if err := p.dht.Start(); err != nil {
|
|
return err
|
|
}
|
|
if err := p.cluster.Connect(); err != nil {
|
|
return err
|
|
}
|
|
if err := p.peer.Start("localhost:" + strconv.Itoa(peer.DefaultPort)); err != nil {
|
|
return err
|
|
}
|
|
if err := p.reflector.Start("localhost:" + strconv.Itoa(DefaultPort)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Shutdown gracefully shuts down the different prism components before exiting.
|
|
func (p *Prism) Shutdown() {
|
|
p.stop.StopAndWait()
|
|
p.reflector.Shutdown()
|
|
p.peer.Shutdown()
|
|
p.cluster.Shutdown()
|
|
p.dht.Shutdown()
|
|
}
|