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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
package collector
import (
"github.com/golang/glog"
"github.com/prometheus/client_golang/prometheus"
mgo "gopkg.in/mgo.v2"
"gopkg.in/mgo.v2/bson"
)
var (
count = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: Namespace,
Subsystem: "collection",
Name: "total_objects",
Help: "The number of objects or documents in this collection",
}, []string{"ns"})
size = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: Namespace,
Subsystem: "collection",
Name: "size_bytes",
Help: "The total size in memory of all records in a collection",
}, []string{"ns"})
avgObjSize = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: Namespace,
Subsystem: "collection",
Name: "avg_objsize_bytes",
Help: "The average size of an object in the collection",
}, []string{"ns"})
storageSize = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: Namespace,
Subsystem: "collection",
Name: "storage_size_bytes",
Help: "The total amount of storage allocated to this collection for document storage",
}, []string{"ns"})
collIndexSize = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: Namespace,
Subsystem: "collection",
Name: "index_size_bytes",
Help: "The total size of all indexes",
}, []string{"ns"})
)
type CollectionStatus struct {
Name string `bson:"ns"`
Count int `bson:"count"`
Size int `bson:"size"`
AvgSize int `bson:"avgObjSize"`
StorageSize int `bson:"storageSize"`
IndexSize int `bson:"totalIndexSize"`
WiredTiger *CollWiredTigerStats `bson:"wiredTiger"`
}
func (collStatus *CollectionStatus) Export(ch chan<- prometheus.Metric) {
count.WithLabelValues(collStatus.Name).Set(float64(collStatus.Count))
size.WithLabelValues(collStatus.Name).Set(float64(collStatus.Size))
avgObjSize.WithLabelValues(collStatus.Name).Set(float64(collStatus.AvgSize))
storageSize.WithLabelValues(collStatus.Name).Set(float64(collStatus.StorageSize))
collIndexSize.WithLabelValues(collStatus.Name).Set(float64(collStatus.IndexSize))
if collStatus.WiredTiger != nil {
collStatus.WiredTiger.Export(ch, collStatus.Name)
}
count.Collect(ch)
size.Collect(ch)
avgObjSize.Collect(ch)
storageSize.Collect(ch)
collIndexSize.Collect(ch)
count.Reset()
size.Reset()
avgObjSize.Reset()
storageSize.Reset()
collIndexSize.Reset()
}
func (collStatus *CollectionStatus) Describe(ch chan<- *prometheus.Desc) {
count.Describe(ch)
size.Describe(ch)
avgObjSize.Describe(ch)
storageSize.Describe(ch)
collIndexSize.Describe(ch)
if collStatus.WiredTiger != nil {
collStatus.WiredTiger.Describe(ch)
}
}
func GetCollectionStatus(session *mgo.Session, db string, collection string) *CollectionStatus {
var collStatus CollectionStatus
err := session.DB(db).Run(bson.D{{"collStats", collection}, {"scale", 1}}, &collStatus)
if err != nil {
glog.Error(err)
return nil
}
return &collStatus
}
func CollectCollectionStatus(session *mgo.Session, db string, ch chan<- prometheus.Metric) {
collection_names, err := session.DB(db).CollectionNames()
if err != nil {
glog.Error("Failed to get collection names for db=" + db)
return
}
for _, collection_name := range collection_names {
collStats := GetCollectionStatus(session, db, collection_name)
if collStats != nil {
glog.Infof("exporting Database Metrics for db=%q, table=%q", db, collection_name)
collStats.Export(ch)
}
}
}
|
