improved debug handlers

1 files changed, 18 insertions, 0 deletions

diff --git a/cmd/util/rpc.go b/cmd/util/rpc.go
new file mode 100644
index 00000000..ee3414a6
--- /dev/null
+++ b/cmd/util/rpc.go
@@ -0,0 +1,18 @@
+package util
+
+import (
+	"errors"
+	"github.com/spiral/roadrunner/service"
+	rrpc "github.com/spiral/roadrunner/service/rpc"
+	"net/rpc"
+)
+
+// RPCClient returns RPC client associated with given roadrunner service container.
+func RPCClient(container service.Container) (*rpc.Client, error) {
+	svc, st := container.Get(rrpc.ID)
+	if st < service.StatusOK {
+		return nil, errors.New("RPC service is not configured")
+	}
+
+	return svc.(*rrpc.Service).Client()
+}