mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-24 10:07:21 +02:00
40 lines
915 B
Go
40 lines
915 B
Go
|
package jsonrpc2
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"log"
|
||
|
)
|
||
|
|
||
|
// HandlerWithError implements Handler by calling the func for each
|
||
|
// request and handling returned errors and results.
|
||
|
type HandlerWithError func(context.Context, *Conn, *Request) (result interface{}, err error)
|
||
|
|
||
|
// Handle implements Handler.
|
||
|
func (h HandlerWithError) Handle(ctx context.Context, conn *Conn, req *Request) {
|
||
|
result, err := h(ctx, conn, req)
|
||
|
if req.Notif {
|
||
|
if err != nil {
|
||
|
log.Printf("jsonrpc2 handler: notification %q handling error: %s", req.Method, err)
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
resp := &Response{ID: req.ID}
|
||
|
if err == nil {
|
||
|
err = resp.SetResult(result)
|
||
|
}
|
||
|
if err != nil {
|
||
|
if e, ok := err.(*Error); ok {
|
||
|
resp.Error = e
|
||
|
} else {
|
||
|
resp.Error = &Error{Message: err.Error()}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if !req.Notif {
|
||
|
if err := conn.SendResponse(ctx, resp); err != nil {
|
||
|
log.Printf("jsonrpc2 handler: sending response %s: %s", resp.ID, err)
|
||
|
}
|
||
|
}
|
||
|
}
|