1 files changed, 46 insertions, 0 deletions
diff --git a/service/limit/service.go b/service/limit/service.go
new file mode 100644
index 00000000..72673d1f
--- /dev/null
+++ b/service/limit/service.go
@@ -0,0 +1,46 @@
+package limit
+
+import (
+	"github.com/spiral/roadrunner"
+	"github.com/spiral/roadrunner/service"
+)
+
+// ID defines controller service name.
+const ID = "constrain"
+
+// Controllable defines the ability to attach rr controller.
+type Controllable interface {
+	// AddController attaches controller to the service.
+	AddController(c roadrunner.Controller)
+}
+
+// Services to control the state of rr service inside other services.
+type Service struct {
+	cfg  *Config
+	lsns []func(event int, ctx interface{})
+}
+
+// Init controller service
+func (s *Service) Init(cfg *Config, c service.Container) (bool, error) {
+	// mount Services to designated services
+	for id, watcher := range cfg.Controllers(s.throw) {
+		svc, _ := c.Get(id)
+		if ctrl, ok := svc.(Controllable); ok {
+			ctrl.AddController(watcher)
+		}
+	}
+
+	return true, nil
+}
+
+// AddListener attaches server event controller.
+func (s *Service) AddListener(l func(event int, ctx interface{})) {
+	s.lsns = append(s.lsns, l)
+}
+
+// throw handles service, server and pool events.
+func (s *Service) throw(event int, ctx interface{}) {
+	for _, l := range s.lsns {
+		l(event, ctx)
+	}
+}