frodovdr/vsphere-influxdb-go
1
0
Fork 0
mirror of https://github.com/Oxalide/vsphere-influxdb-go.git synced 2023-10-10 11:36:51 +00:00
Code Issues Releases Wiki Activity

Merge c60748773f8b658cb74446eaa222a5afa814f5ad into b656d6be889ce3b4286f8353ff89fbb2ab9cbd9c

Browse Source
This commit is contained in:
heldri 2017-05-11 09:47:12 +00:00 committed by GitHub
commit d1835bfa81
2 changed files with 92 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vsphere-influxdb-go.json
View File

@ -66,4 +66,4 @@
]
}
]
}
}

96
vsphere-influxdb.go
View File

@ -21,8 +21,6 @@ import (
"encoding/json"
"flag"
"fmt"
"github.com/davecgh/go-spew/spew"
"golang.org/x/net/context"
"log"
"math"
"net/url"
@ -32,6 +30,9 @@ import (
"strings"
"time"
"github.com/davecgh/go-spew/spew"
"golang.org/x/net/context"
influxclient "github.com/influxdata/influxdb/client/v2"
"github.com/vmware/govmomi"
"github.com/vmware/govmomi/property"
@ -219,6 +220,7 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
objectTypes = append(objectTypes, group.ObjectType)
}
objectTypes = append(objectTypes, "ClusterComputeResource")
objectTypes = append(objectTypes, "ResourcePool")
// Loop trought datacenters and create the intersting object reference list
mors := []types.ManagedObjectReference{}
@ -246,6 +248,7 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
vm_refs := []types.ManagedObjectReference{}
host_refs := []types.ManagedObjectReference{}
cluster_refs := []types.ManagedObjectReference{}
respool_refs := []types.ManagedObjectReference{}
new_mors := []types.ManagedObjectReference{}
@ -260,6 +263,8 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
new_mors = append(new_mors, mor)
} else if mor.Type == "ClusterComputeResource" {
cluster_refs = append(cluster_refs, mor)
} else if mor.Type == "ResourcePool" {
respool_refs = append(respool_refs, mor)
}
}
// Copy the mors without the clusters
@ -283,6 +288,43 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
return
}
//Retrieve properties for ResourcePool
var rpmo []mo.ResourcePool
err = pc.Retrieve(ctx, respool_refs, []string{"summary"}, &rpmo)
if err != nil {
fmt.Println(err)
return
}
// Initialize the map that will hold the VM MOR to ResourcePool reference
vmToPool := make(map[types.ManagedObjectReference]string)
// Retrieve properties for ResourcePools
if len(respool_refs) > 0 {
if debug == true {
stdlog.Println("going inside ResourcePools")
}
var respool []mo.ResourcePool
err = pc.Retrieve(ctx, respool_refs, []string{"name", "config", "vm"}, &respool)
if err != nil {
fmt.Println(err)
return
}
for _, pool := range respool {
if debug == true {
stdlog.Println("---resourcepool name - you should see every resourcepool here (+VMs inside)----")
stdlog.Println(pool.Name)
}
for _, vm := range pool.Vm {
if debug == true {
stdlog.Println("--VM ID - you should see every VM ID here--")
stdlog.Println(vm)
}
vmToPool[vm] = pool.Name
}
}
}
// Initialize the map that will hold the VM MOR to cluster reference
vmToCluster := make(map[types.ManagedObjectReference]string)
@ -317,6 +359,13 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
}
}
// Retrieve properties for the pools
respool_summary := make(map[types.ManagedObjectReference]map[string]string)
for _, pools := range rpmo {
respool_summary[pools.Self] = make(map[string]string)
respool_summary[pools.Self]["name"] = pools.Summary.GetResourcePoolSummary().Name
}
// Retrieve properties for the hosts
host_summary := make(map[types.ManagedObjectReference]map[string]string)
host_extra_metrics := make(map[types.ManagedObjectReference]map[string]int64)
@ -343,6 +392,9 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
if vmToCluster[vm.Self] != "" {
vm_summary[vm.Self]["cluster"] = vmToCluster[vm.Self]
}
if vmToPool[vm.Self] != "" {
vm_summary[vm.Self]["respool"] = vmToPool[vm.Self]
}
vm_summary[vm.Self]["esx"] = host_summary[*vm.Summary.Runtime.Host]["name"]
}
@ -456,6 +508,12 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
}
}
if summary, ok := respool_summary[pem.Entity]; ok {
for key, tag := range summary {
tags[key] = tag
}
}
special_fields := make(map[string]map[string]map[string]map[string]interface{})
special_tags := make(map[string]map[string]map[string]map[string]string)
nowTime := time.Now()
@ -538,6 +596,26 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
}
}
}
var respool []mo.ResourcePool
err = pc.Retrieve(ctx, respool_refs, []string{"name", "config", "vm"}, &respool)
if err != nil {
fmt.Println(err)
return
}
for _, pool := range respool {
respoolFields := map[string]interface{}{
"cpu_limit": pool.Config.CpuAllocation.GetResourceAllocationInfo().Limit,
"memory_limit": MBToKB(pool.Config.MemoryAllocation.GetResourceAllocationInfo().Limit),
}
respoolTags := map[string]string{"pool_name": pool.Name}
pt3, err := influxclient.NewPoint("resourcepool", respoolTags, respoolFields, time.Now())
if err != nil {
errlog.Println(err)
}
bp.AddPoint(pt3)
}
}
//InfluxDB send
err = InfluxDBClient.Write(bp)
@ -549,6 +627,13 @@ func (vcenter *VCenter) Query(config Configuration, InfluxDBClient influxclient.
}
func MBToKB(n int64) int64 {
if n != -1 {
n = 1000 * n
}
return n
}
func min(n ...int64) int64 {
var min int64 = -1
for _, i := range n {
@ -613,6 +698,7 @@ func queryVCenter(vcenter VCenter, config Configuration, InfluxDBClient influxcl
func main() {
flag.BoolVar(&debug, "debug", false, "Debug mode")
var cfgFile = flag.String("config","/etc/" + path.Base(os.Args[0])+".json", "Config file to use. Default is /etc/"+path.Base(os.Args[0])+".json")
flag.Parse()
stdlog = log.New(os.Stdout, "", log.Ldate|log.Ltime)
@ -620,16 +706,16 @@ func main() {
stdlog.Println("Starting :", path.Base(os.Args[0]))
// read the configuration
file, err := os.Open("/etc/" + path.Base(os.Args[0]) + ".json")
file, err := os.Open(*cfgFile)
if err != nil {
errlog.Println("Could not open configuration file")
errlog.Println("Could not open configuration file " + *cfgFile)
errlog.Println(err)
}
jsondec := json.NewDecoder(file)
config := Configuration{}
err = jsondec.Decode(&config)
if err != nil {
errlog.Println("Could not decode configuration file")
errlog.Println("Could not decode configuration file " + *cfgFile)
errlog.Println(err)
}