mirror of
https://github.com/umputun/reproxy.git
synced 2025-06-30 22:13:42 +02:00
d563c2a648
- Add proper synchronization for DNS mock server - Fix race condition with thread-safe token access - Add timeouts to certificate acquisition to prevent test hanging - Improve error handling in DNS server - Normalize comments with unfuck-ai-comments
52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/umputun/reproxy/lib"
|
|
)
|
|
|
|
func main() {
|
|
// create demo plugin on port 1234 with two methods: HeaderThing and ErrorThing
|
|
// both called via RPC from reproxy core with fully formed lib.Request
|
|
plugin := lib.Plugin{
|
|
Name: "TestPlugin",
|
|
Address: "plugin-example:1234",
|
|
Methods: []string{"HeaderThing", "ErrorThing"},
|
|
}
|
|
log.Printf("start demo plugin")
|
|
// do start the plugin listener and register with reproxy plugin conductor
|
|
if err := plugin.Do(context.TODO(), "http://reproxy:8081", new(Handler)); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// Handler is an example of middleware handler altering headers and status
|
|
type Handler struct{}
|
|
|
|
// HeaderThing adds key:val header to the response
|
|
// nolint:unparam // error response is required by the interface
|
|
func (h *Handler) HeaderThing(req lib.Request, res *lib.Response) (err error) {
|
|
log.Printf("req: %+v", req)
|
|
res.HeadersOut = http.Header{}
|
|
res.HeadersOut.Add("key", "val")
|
|
res.HeadersIn = http.Header{}
|
|
res.HeadersIn.Add("token", "something")
|
|
res.StatusCode = 200 // each handler has to set status code
|
|
return
|
|
}
|
|
|
|
// ErrorThing returns status 500 on "/fail" url. This terminated processing chain on reproxy side immediately
|
|
// nolint:unparam // error response is required by the interface
|
|
func (h *Handler) ErrorThing(req lib.Request, res *lib.Response) (err error) {
|
|
log.Printf("req: %+v", req)
|
|
if req.URL == "/fail" {
|
|
res.StatusCode = 500
|
|
return
|
|
}
|
|
res.StatusCode = 200
|
|
return
|
|
}
|