-
Notifications
You must be signed in to change notification settings - Fork 46
/
middleware_stat.go
77 lines (67 loc) · 1.98 KB
/
middleware_stat.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
74
75
76
77
//
// Copyright 2019 Insolar Technologies GmbH
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package pubsubwrap
import (
"context"
"sort"
"sync"
"github.com/ThreeDotsLabs/watermill/message"
"github.com/insolar/insolar/instrumentation/introspector/introproto"
)
// MessageStatByType holds publish statistic per message type.
type MessageStatByType struct {
sync.Mutex
stat map[string]int64
}
// NewMessageStatByType is a constructor for MessageStatByType.
func NewMessageStatByType() *MessageStatByType {
return &MessageStatByType{
stat: map[string]int64{},
}
}
// Filter counts published messages by type.
func (ms *MessageStatByType) Filter(m *message.Message) (*message.Message, error) {
typ, err := decodeType(m)
key := typ.String()
if err != nil {
if de, ok := err.(decodeError); ok {
key = "legacy." + de.metadataType
}
}
ms.Lock()
defer ms.Unlock()
ms.stat[key]++
return m, err
}
// GetMessagesStat returns publish statistic per message type.
func (ms *MessageStatByType) GetMessagesStat(context.Context, *introproto.EmptyArgs) (*introproto.AllMessageStatByType, error) {
ms.Lock()
all := make([]*introproto.MessageStatByType, 0, len(ms.stat))
for name, count := range ms.stat {
all = append(all, &introproto.MessageStatByType{
Name: name,
Count: count,
})
}
ms.Unlock()
// frequent messages goes first
sort.Slice(all, func(i, j int) bool {
return all[i].Count > all[j].Count
})
return &introproto.AllMessageStatByType{
Counters: all,
}, nil
}