92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
package application
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"git.derfenix.pro/fenix/astontest/adapters/randomstring"
|
|
"git.derfenix.pro/fenix/astontest/adapters/stdoutstore"
|
|
"git.derfenix.pro/fenix/astontest/pkg/discovery"
|
|
"git.derfenix.pro/fenix/astontest/pkg/messenger"
|
|
)
|
|
|
|
type Application struct {
|
|
discoverySet discovery.DiscoverySet
|
|
messenger *messenger.Messenger
|
|
log *zap.Logger
|
|
}
|
|
|
|
func NewApplication(cfg *Config, log *zap.Logger) (*Application, error) {
|
|
discoveryOpts := []discovery.Option{
|
|
discovery.WithBroadcastInterval(cfg.BroadcastInterval),
|
|
}
|
|
|
|
if cfg.Debug {
|
|
discoveryOpts = append(discoveryOpts, discovery.WithDebug())
|
|
}
|
|
|
|
discoverySet, err := discovery.NewDiscoverySet(log.Named("discovery"), cfg.DiscoveryPort, discoveryOpts...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("new discovery set: %w", err)
|
|
|
|
}
|
|
|
|
messengerSrv := messenger.NewMessenger(
|
|
discoverySet.NewNodes(),
|
|
cfg.MessagingPort,
|
|
stdoutstore.New(),
|
|
randomstring.New(cfg.RandomMessageSize),
|
|
cfg.MessagingInterval,
|
|
log.Named("messenger"),
|
|
)
|
|
|
|
go discoverySet.FailNodes(messengerSrv.FailNodesCh)
|
|
|
|
return &Application{
|
|
discoverySet: discoverySet,
|
|
messenger: messengerSrv,
|
|
log: log,
|
|
}, nil
|
|
}
|
|
|
|
func (a *Application) Start(ctx context.Context, wg *sync.WaitGroup) {
|
|
wg.Add(len(a.discoverySet))
|
|
for _, discover := range a.discoverySet {
|
|
go discover.Start(ctx, wg)
|
|
}
|
|
|
|
wg.Add(2)
|
|
go a.messenger.StartServer(ctx, wg)
|
|
go a.messenger.Start(ctx, wg)
|
|
}
|
|
|
|
var _ = `
|
|
s1-1 | Received 'apq34yod73' from <172.19.0.5>
|
|
s1-1 | Received 'cbtfl716li' from <172.19.0.3>
|
|
s1-1 | Received 'n84fg3mx9o' from <172.19.0.2>
|
|
s1-1 | Received 'zb98146fqz' from <172.19.0.4>
|
|
|
|
s2-1 | Received '7h1s9ruilr' from <172.19.0.2>
|
|
s2-1 | Received 'aij179r0ck' from <172.19.0.3>
|
|
s2-1 | Received 'm3k0snj7ma' from <172.19.0.6>
|
|
s2-1 | Received 'tw5726fo7e' from <172.19.0.5>
|
|
|
|
s3-1 | Received '32d20marhg' from <172.19.0.6>
|
|
s3-1 | Received 'ffs9pi6o9j' from <172.19.0.3>
|
|
s3-1 | Received 'wcso6aashe' from <172.19.0.5>
|
|
s3-1 | Received 'wfhcy9xbdj' from <172.19.0.4>
|
|
|
|
s4-1 | Received '2le3u1ikyg' from <172.19.0.2>
|
|
s4-1 | Received '5locacst6w' from <172.19.0.4>
|
|
s4-1 | Received 'cf47nyd2ca' from <172.19.0.6>
|
|
s4-1 | Received 'guvfhb7wud' from <172.19.0.5>
|
|
|
|
s5-1 | Received 'dsr7qt2x5l' from <172.19.0.3>
|
|
s5-1 | Received 'l67gc5xdt3' from <172.19.0.2>
|
|
s5-1 | Received 'mof0yp4vxt' from <172.19.0.6>
|
|
s5-1 | Received 'slc9tw30r4' from <172.19.0.4>
|
|
`
|