52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package mongodb
|
|
|
|
import (
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"gopkg.in/mgo.v2"
|
|
"gopkg.in/mgo.v2/bson"
|
|
)
|
|
|
|
type Server struct {
|
|
Url *url.URL
|
|
Session *mgo.Session
|
|
lastResult *ServerStatus
|
|
}
|
|
|
|
func (s *Server) getDefaultTags() map[string]string {
|
|
tags := make(map[string]string)
|
|
tags["hostname"] = s.Url.Host
|
|
return tags
|
|
}
|
|
|
|
func (s *Server) gatherData(acc telegraf.Accumulator) error {
|
|
s.Session.SetMode(mgo.Eventual, true)
|
|
s.Session.SetSocketTimeout(0)
|
|
result := &ServerStatus{}
|
|
err := s.Session.DB("admin").Run(bson.D{{"serverStatus", 1}, {"recordStats", 0}}, result)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
s.lastResult = result
|
|
}()
|
|
|
|
result.SampleTime = time.Now()
|
|
if s.lastResult != nil && result != nil {
|
|
duration := result.SampleTime.Sub(s.lastResult.SampleTime)
|
|
durationInSeconds := int64(duration.Seconds())
|
|
if durationInSeconds == 0 {
|
|
durationInSeconds = 1
|
|
}
|
|
data := NewMongodbData(
|
|
NewStatLine(*s.lastResult, *result, s.Url.Host, true, durationInSeconds),
|
|
s.getDefaultTags(),
|
|
)
|
|
data.AddDefaultStats()
|
|
data.flush(acc)
|
|
}
|
|
return nil
|
|
}
|