summaryrefslogtreecommitdiff
path: root/src/mongo/gotools/src/github.com/mongodb/mongo-tools/mongostat/stat_consumer/line/line.go
blob: 4f6d8b4d0ca5f26d37527db607ee9f4b7e8f8360 (plain)
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
// Copyright (C) MongoDB, Inc. 2014-present.
//
// 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

package line

import (
	"github.com/mongodb/mongo-tools/mongostat/status"
)

// StatLine is a wrapper for all metrics reported by mongostat for monitored hosts
type StatLine struct {
	Fields  map[string]string
	Error   error
	Printed bool
}

type StatLines []*StatLine

func (slice StatLines) Len() int {
	return len(slice)
}

func (slice StatLines) Less(i, j int) bool {
	return slice[i].Fields["host"] < slice[j].Fields["host"]
}

func (slice StatLines) Swap(i, j int) {
	slice[i], slice[j] = slice[j], slice[i]
}

// NewStatLine constructs a StatLine object from two ServerStatus objects
func NewStatLine(oldStat, newStat *status.ServerStatus, headerKeys []string, c *status.ReaderConfig) *StatLine {
	line := &StatLine{
		Fields: make(map[string]string),
	}
	for _, key := range headerKeys {
		_, ok := StatHeaders[key]
		if ok {
			line.Fields[key] = StatHeaders[key].ReadField(c, newStat, oldStat)
		} else {
			line.Fields[key] = status.InterpretField(key, newStat, oldStat)
		}
	}
	// We always need host and storage_engine, even if they aren't being displayed
	line.Fields["host"] = StatHeaders["host"].ReadField(c, newStat, oldStat)
	line.Fields["storage_engine"] = StatHeaders["storage_engine"].ReadField(c, newStat, oldStat)
	return line
}