-
Notifications
You must be signed in to change notification settings - Fork 101
/
odr_cache.go
65 lines (52 loc) · 1.34 KB
/
odr_cache.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
/**
* @file
* @copyright defined in go-seele/LICENSE
*/
package light
import (
"reflect"
"github.com/hashicorp/golang-lru"
"github.com/seeleteam/go-seele/common"
"github.com/seeleteam/go-seele/crypto"
)
const (
cacheSizeOdrTx = 1024
)
type odrCache struct {
typedCaches map[string]*lru.Cache
}
func newOdrCache() *odrCache {
return &odrCache{
typedCaches: map[string]*lru.Cache{
odrRequestType(&odrTxByHashRequest{}): common.MustNewCache(cacheSizeOdrTx),
},
}
}
func odrRequestType(request odrRequest) string {
return reflect.TypeOf(request).String()
}
func odrRequestKey(request odrRequest) common.Hash {
reqID := request.getRequestID()
defer request.setRequestID(reqID)
request.setRequestID(0)
return crypto.MustHash([]interface{}{odrRequestType(request), request})
}
func (cache *odrCache) get(request odrRequest) (odrResponse, bool) {
reqType := odrRequestType(request)
lruCache := cache.typedCaches[reqType]
if lruCache == nil {
return nil, false
}
reqKey := odrRequestKey(request)
if resp, ok := lruCache.Get(reqKey); ok {
return resp.(odrResponse), true
}
return nil, false
}
func (cache *odrCache) add(request odrRequest, response odrResponse) {
reqType := odrRequestType(request)
if lruCache := cache.typedCaches[reqType]; lruCache != nil {
reqKey := odrRequestKey(request)
lruCache.Add(reqKey, response)
}
}