4ea494e8c5
In order to be able to register the subservers with the root grpc server before we have all dependencies available, we wrap them in an GrpcHandler struct. This struct will initially hold an empty reference to the subservers, which allows us to register with the GRPC server, and later populate and create the subserver instance.
61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
package wtclientrpc
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/lightningnetwork/lnd/lnrpc"
|
|
)
|
|
|
|
// createNewSubServer is a helper method that will create the new sub server
|
|
// given the main config dispatcher method. If we're unable to find the config
|
|
// that is meant for us in the config dispatcher, then we'll exit with an
|
|
// error.
|
|
func createNewSubServer(configRegistry lnrpc.SubServerConfigDispatcher) (
|
|
*WatchtowerClient, lnrpc.MacaroonPerms, error) {
|
|
|
|
// We'll attempt to look up the config that we expect, according to our
|
|
// subServerName name. If we can't find this, then we'll exit with an
|
|
// error, as we're unable to properly initialize ourselves without this
|
|
// config.
|
|
subServerConf, ok := configRegistry.FetchConfig(subServerName)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("unable to find config for "+
|
|
"subserver type %s", subServerName)
|
|
}
|
|
|
|
// Now that we've found an object mapping to our service name, we'll
|
|
// ensure that it's the type we need.
|
|
config, ok := subServerConf.(*Config)
|
|
if !ok {
|
|
return nil, nil, fmt.Errorf("wrong type of config for "+
|
|
"subserver %s, expected %T got %T", subServerName,
|
|
&Config{}, subServerConf)
|
|
}
|
|
|
|
// Before we try to make the new service instance, we'll perform
|
|
// some sanity checks on the arguments to ensure that they're useable.
|
|
switch {
|
|
case config.Resolver == nil:
|
|
return nil, nil, errors.New("a lncfg.TCPResolver is required")
|
|
}
|
|
|
|
return New(config)
|
|
}
|
|
|
|
func init() {
|
|
subServer := &lnrpc.SubServerDriver{
|
|
SubServerName: subServerName,
|
|
NewGrpcHandler: func() lnrpc.GrpcHandler {
|
|
return &ServerShell{}
|
|
},
|
|
}
|
|
|
|
// If the build tag is active, then we'll register ourselves as a
|
|
// sub-RPC server within the global lnrpc package namespace.
|
|
if err := lnrpc.RegisterSubServer(subServer); err != nil {
|
|
panic(fmt.Sprintf("failed to register sub server driver "+
|
|
"'%s': %v", subServerName, err))
|
|
}
|
|
}
|