Informer plugin ready

1 files changed, 56 insertions, 0 deletions

diff --git a/plugins/informer/plugin.go b/plugins/informer/plugin.go
new file mode 100644
index 00000000..09d933fd
--- /dev/null
+++ b/plugins/informer/plugin.go
@@ -0,0 +1,56 @@
+package informer
+
+import (
+	"github.com/spiral/endure"
+	"github.com/spiral/errors"
+	"github.com/spiral/roadrunner/v2"
+	"github.com/spiral/roadrunner/v2/interfaces/informer"
+	"github.com/spiral/roadrunner/v2/interfaces/log"
+)
+
+const PluginName = "informer"
+
+type Plugin struct {
+	registry map[string]informer.Informer
+	log      log.Logger
+}
+
+func (p *Plugin) Init(log log.Logger) error {
+	p.registry = make(map[string]informer.Informer)
+	p.log = log
+	return nil
+}
+
+// Reset named service.
+func (p *Plugin) Workers(name string) ([]roadrunner.WorkerBase, error) {
+	const op = errors.Op("get workers")
+	svc, ok := p.registry[name]
+	if !ok {
+		return nil, errors.E(op, errors.Errorf("no such service: %s", name))
+	}
+
+	return svc.Workers(), nil
+}
+
+// CollectTarget resettable service.
+func (p *Plugin) CollectTarget(name endure.Named, r informer.Informer) error {
+	p.registry[name.Name()] = r
+	return nil
+}
+
+// Collects declares services to be collected.
+func (p *Plugin) Collects() []interface{} {
+	return []interface{}{
+		p.CollectTarget,
+	}
+}
+
+// Name of the service.
+func (p *Plugin) Name() string {
+	return PluginName
+}
+
+// RPCService returns associated rpc service.
+func (p *Plugin) RPC() interface{} {
+	return &rpc{srv: p, log: p.log}
+}