clusters.go

package remy

import (
	"bytes"
	"encoding/json"
	"fmt"
)

// Cluster is the underlying struct for a single Cluster in a domain.  Each domain may have multiple Cluster instances,
// each having their own ClusterMaster, deployed state, etc.
type Cluster struct {
	Name    string
	Servers []struct {
		Name                   string
		State                  string
		Health                 string
		IsClusterMaster        bool   `json:"clusterMaster,omitempty"`
		DropOutFrequency       string `json:",omitempty"`
		ResendRequestsCount    int    `json:",omitempty"`
		FragmentsSentCount     int    `json:",omitempty"`
		FragmentsReceivedCount int    `json:",omitempty"`
	} `json:"servers,omitempty"`
}

// GoString creates a GoString of the Cluster type for use in command-line applications.
func (c *Cluster) GoString() string {
	var buffer bytes.Buffer
	buffer.WriteString(fmt.Sprintf("Name: %v\n", c.Name))
	for i := range c.Servers {
		buffer.WriteString(fmt.Sprintf("State:             %-14v| Health:               %-14v| Cluster Master?       %-14v| Drop Out Freq: %-14v\n",
			c.Servers[i].State, c.Servers[i].Health, c.Servers[i].IsClusterMaster, c.Servers[i].DropOutFrequency))
		buffer.WriteString(fmt.Sprintf("Resend Req. Count: %-14v| Fragments Sent Count: %-14v| Fragments Recv Count: %-14v\n",
			c.Servers[i].ResendRequestsCount, c.Servers[i].FragmentsSentCount, c.Servers[i].FragmentsReceivedCount))
	}
	return buffer.String()
}

// Clusters returns all clusters configured in a domain and provides run-time information for each cluster and for each cluster's member servers, including all the member servers' state and health.
func (a *AdminServer) Clusters(fullFormat bool) ([]Cluster, error) {
	url := fmt.Sprintf("%v%v/clusters", a.AdminURL, MonitorPath)
	if fullFormat {
		url = url + "?format=full"
	}
	w, err := requestAndUnmarshal(url, a)
	if err != nil {
		return nil, err
	}
	var clusters []Cluster
	if err := json.Unmarshal(w.Body.Items, &clusters); err != nil {
		return nil, err
	}
	return clusters, nil
}

// Cluster returns run-time information for the specified cluster and its member servers, including the member servers' state and health.
func (a *AdminServer) Cluster(clusterName string) (*Cluster, error) {
	url := fmt.Sprintf("%v%v/clusters/%v", a.AdminURL, MonitorPath, clusterName)
	w, err := requestAndUnmarshal(url, a)
	if err != nil {
		return nil, err
	}
	var cluster Cluster
	if err := json.Unmarshal(w.Body.Item, &cluster); err != nil {
		return nil, err
	}
	return &cluster, nil
}