Merge pull request #463 from spiral/experiment/core_pluginsv2.0.0-beta1

[RR2] Plugins

1 files changed, 61 insertions, 0 deletions

diff --git a/tests/plugins/http/plugin_middleware.go b/tests/plugins/http/plugin_middleware.go
new file mode 100644
index 00000000..8d02524d
--- /dev/null
+++ b/tests/plugins/http/plugin_middleware.go
@@ -0,0 +1,61 @@
+package http
+
+import (
+	"net/http"
+
+	"github.com/spiral/roadrunner/v2/plugins/config"
+)
+
+type PluginMiddleware struct {
+	config config.Configurer
+}
+
+func (p *PluginMiddleware) Init(cfg config.Configurer) error {
+	p.config = cfg
+	return nil
+}
+
+func (p *PluginMiddleware) Middleware(next http.Handler) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/halt" {
+			w.WriteHeader(500)
+			_, err := w.Write([]byte("halted"))
+			if err != nil {
+				panic("error writing the data to the http reply")
+			}
+		} else {
+			next.ServeHTTP(w, r)
+		}
+	}
+}
+
+func (p *PluginMiddleware) Name() string {
+	return "pluginMiddleware"
+}
+
+type PluginMiddleware2 struct {
+	config config.Configurer
+}
+
+func (p *PluginMiddleware2) Init(cfg config.Configurer) error {
+	p.config = cfg
+	return nil
+}
+
+func (p *PluginMiddleware2) Middleware(next http.Handler) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/boom" {
+			w.WriteHeader(555)
+			_, err := w.Write([]byte("boom"))
+			if err != nil {
+				panic("error writing the data to the http reply")
+			}
+		} else {
+			next.ServeHTTP(w, r)
+		}
+	}
+}
+
+func (p *PluginMiddleware2) Name() string {
+	return "pluginMiddleware2"
+}