112 lines
1.4 KiB
Go
112 lines
1.4 KiB
Go
package rpc
|
|
|
|
import (
|
|
"google.golang.org/grpc"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
server *grpc.Server
|
|
o *ServerOptions
|
|
)
|
|
|
|
var (
|
|
inUse sync.Mutex
|
|
isRunning bool
|
|
)
|
|
|
|
func RunWith(r func(srv *grpc.Server), opts ...ServerOption) {
|
|
|
|
if !inUse.TryLock() {
|
|
log.E.Ln("rpc server is in use.")
|
|
return
|
|
}
|
|
|
|
log.I.Ln("initializing the rpc server")
|
|
|
|
o = &ServerOptions{
|
|
&storeMem{},
|
|
unixPathDefault,
|
|
false,
|
|
NullPort,
|
|
[]string{},
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
opt.apply(o)
|
|
}
|
|
|
|
if o.unixPath != "" {
|
|
log.I.Ln("enabling rpc unix listener:")
|
|
log.I.F("- [/unix%s]", o.unixPath)
|
|
|
|
isUnixSockEnabled = true
|
|
}
|
|
|
|
if o.tunEnable {
|
|
configureTunnel()
|
|
}
|
|
|
|
isConfigured <- true
|
|
|
|
server = grpc.NewServer()
|
|
r(server)
|
|
|
|
go start()
|
|
}
|
|
|
|
func Options() *ServerOptions {
|
|
return o
|
|
}
|
|
|
|
func start() {
|
|
|
|
log.I.Ln("starting rpc server")
|
|
|
|
var err error
|
|
|
|
createTunnel()
|
|
|
|
if err = startTunnel(server); check(err) {
|
|
startupErrors <- err
|
|
return
|
|
}
|
|
|
|
if err = startUnixSocket(server); check(err) {
|
|
startupErrors <- err
|
|
return
|
|
}
|
|
|
|
isRunning = true
|
|
|
|
log.I.Ln("rpc server is ready")
|
|
isReady <- true
|
|
}
|
|
|
|
func Shutdown() {
|
|
|
|
if !isRunning {
|
|
return
|
|
}
|
|
|
|
log.I.Ln("shutting down rpc server")
|
|
|
|
server.GracefulStop()
|
|
|
|
var err error
|
|
|
|
//err = stopUnixSocket()
|
|
//
|
|
//check(err)
|
|
|
|
err = stopTunnel()
|
|
|
|
check(err)
|
|
|
|
isRunning = false
|
|
|
|
inUse.Unlock()
|
|
|
|
log.I.Ln("- rpc server shutdown completed")
|
|
}
|