blob: 7ff16a201082163d7a5c01577282a6d4f4d6b73d (
plain) (
blame)
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
74
75
76
77
|
package ddp
import (
"fmt"
"sync"
"time"
"github.com/apex/log"
)
// Contains common utility types.
// -------------------------------------------------------------------
// KeyManager provides simple incrementing IDs for ddp messages.
type KeyManager struct {
// nextID is the next ID for API calls
nextID uint64
// idMutex is a mutex to protect ID updates
idMutex *sync.Mutex
}
// NewKeyManager creates a new instance and sets up resources.
func NewKeyManager() *KeyManager {
return &KeyManager{idMutex: new(sync.Mutex)}
}
// Next issues a new ID for use in calls.
func (id *KeyManager) Next() string {
id.idMutex.Lock()
next := id.nextID
id.nextID++
id.idMutex.Unlock()
return fmt.Sprintf("%x", next)
}
// -------------------------------------------------------------------
// PingTracker tracks in-flight pings.
type PingTracker struct {
handler func(error)
timeout time.Duration
timer *time.Timer
}
// -------------------------------------------------------------------
// Call represents an active RPC call.
type Call struct {
ID string // The uuid for this method call
ServiceMethod string // The name of the service and method to call.
Args interface{} // The argument to the function (*struct).
Reply interface{} // The reply from the function (*struct).
Error error // After completion, the error status.
Done chan *Call // Strobes when call is complete.
Owner *Client // Client that owns the method call
}
// done removes the call from any owners and strobes the done channel with itself.
func (call *Call) done() {
delete(call.Owner.calls, call.ID)
select {
case call.Done <- call:
// ok
default:
// We don't want to block here. It is the caller's responsibility to make
// sure the channel has enough buffer space. See comment in Go().
log.Debug("rpc: discarding Call reply due to insufficient Done chan capacity")
}
}
// IgnoreErr logs an error if it occurs and ignores it.
func IgnoreErr(err error, msg string) {
if err != nil {
log.WithError(err).Debug(msg)
}
}
|
