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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|