Курнат Андрей
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package nats
|
|
|
|
import (
|
|
"time"
|
|
|
|
nserver "github.com/nats-io/nats-server/v2/server"
|
|
)
|
|
|
|
var NATSListenAndServeLoopTimer = 1 * time.Second
|
|
|
|
type NATSServer struct {
|
|
server *nserver.Server
|
|
}
|
|
|
|
// NatsOption configures the new NATSServer instance
|
|
func NewNATSServer(logger nserver.Logger, opts ...NatsOption) (*NATSServer, error) {
|
|
natsOpts := &nserver.Options{}
|
|
|
|
for _, o := range opts {
|
|
o(natsOpts)
|
|
}
|
|
|
|
// enable JetStream
|
|
natsOpts.JetStream = true
|
|
// The NATS server itself runs the signal handling. We set `natsOpts.NoSigs = true` because we want to handle signals ourselves
|
|
natsOpts.NoSigs = true
|
|
|
|
server, err := nserver.NewServer(natsOpts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
server.SetLoggerV2(logger, true, true, false)
|
|
|
|
return &NATSServer{
|
|
server: server,
|
|
}, nil
|
|
}
|
|
|
|
// ListenAndServe runs the NATSServer in a blocking way until the server is shutdown or an error occurs
|
|
func (n *NATSServer) ListenAndServe() (err error) {
|
|
n.server.Start() // it won't block
|
|
n.server.WaitForShutdown() // block until the server is fully shutdown
|
|
return nil
|
|
}
|
|
|
|
// Shutdown stops the NATSServer gracefully
|
|
func (n *NATSServer) Shutdown() {
|
|
n.server.Shutdown()
|
|
n.server.WaitForShutdown()
|
|
}
|