/
server_example.go
73 lines (64 loc) · 1.41 KB
/
server_example.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package main
import (
"context"
"encoding/json"
"log"
"net/http"
"github.com/penguinn/jsonrpc2"
"time"
"sync"
"github.com/mitchellh/mapstructure"
)
type Echo int
type AddParams struct {
AddStart int `json:"addStart"`
AddEnd int `json:"addEnd"`
}
type AddResult struct {
Sum int `json:"sum"`
}
func(p *Echo)Add(c context.Context, params *json.RawMessage) (interface{}, *jsonrpc2.Error) {
var param AddParams
if err := jsonrpc2.Unmarshal(params, ¶m); err != nil {
log.Fatal(err)
return nil, err
}
return AddResult{
Sum: param.AddStart + param.AddEnd,
}, nil
}
func main() {
var wg sync.WaitGroup
wg.Add(2)
go serClient(wg)
go serServer(wg)
wg.Wait()
}
func serClient(wg sync.WaitGroup) {
time.Sleep(1)
client := jsonrpc2.NewClient("http://127.0.0.1:8080/v1/jrpc")
params := AddParams{AddStart:1, AddEnd:2}
resp, err := client.Call("Add", params, 1)
if err != nil && resp.Error != nil{
log.Fatal(err)
}else {
var sum AddResult
err := mapstructure.WeakDecode(resp.Result, &sum)
if err != nil {
log.Fatal(err)
}
log.Println(sum.Sum)
}
wg.Done()
}
func serServer(wg sync.WaitGroup) {
log.Println(1)
p := new(Echo)
jsonrpc2.RegisterMethod("Add", p.Add, AddParams{}, AddResult{})
http.HandleFunc("/v1/jrpc", jsonrpc2.Handler)
http.HandleFunc("/v1/jrpc/debug", jsonrpc2.DebugHandler)
if err := http.ListenAndServe(":8080", nil); err != nil {
log.Fatalln(err)
}
wg.Done()
}