feat(app): add IsRunning function

Aniratman · Jan 18, 2021 · 9e393b0 · 9e393b0
1 parent e2e622c
commit 9e393b0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/app.go b/app.go
@@ -87,6 +87,7 @@ type Pitaya interface {
 	StartWorker()
 	RegisterRPCJob(rpcJob worker.RPCJob) error
 	Documentation(getPtrNames bool) (map[string]interface{}, error)
+	IsRunning() bool
 
 	RPC(ctx context.Context, routeStr string, reply proto.Message, arg proto.Message) error
 	RPCTo(ctx context.Context, serverID, routeStr string, reply proto.Message, arg proto.Message) error
@@ -252,6 +253,13 @@ func (app *App) GetServers() []*cluster.Server {
 	return app.serviceDiscovery.GetServers()
 }
 
+// IsRunning indicates if the Pitaya app has been initialized. Note: This
+// doesn't cover acceptors, only the pitaya internal registration and modules
+// initialization.
+func (app *App) IsRunning() bool {
+	return app.running
+}
+
 // SetLogger logger setter
 func SetLogger(l logging.Logger) {
 	logger.Log = l

diff --git a/mocks/app.go b/mocks/app.go