telemetry.go

/*
 * Copyright 2018 Dgraph Labs, Inc. and Contributors
 *
 * 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
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package telemetry

import (
	"bytes"
	"encoding/json"
	"io/ioutil"
	"net/http"
	"runtime"

	"github.com/dgraph-io/dgraph/protos/pb"
	"github.com/dgraph-io/dgraph/x"
	"github.com/golang/glog"
)

// Telemetry holds information about the state of the zero and alpha server.
type Telemetry struct {
	Arch                 string
	Cid                  string
	ClusterSize          int
	DiskUsageMB          int64
	NumAlphas            int
	NumGroups            int
	NumTablets           int
	NumZeros             int
	OS                   string
	SinceHours           int
	Version              string
	GraphqlQueryCount    uint64
	NonGraphqlQueryCount uint64
}

var keenURL = "https://ping.dgraph.io/3.0/projects/5b809dfac9e77c0001783ad0/events"

// NewZeroTelemetry returns a Telemetry struct that holds information about the state of zero
// server.
func NewZeroTelemetry(ms *pb.MembershipState) *Telemetry {
	if len(ms.Cid) == 0 {
		glog.V(2).Infoln("No CID found yet")
		return nil
	}
	t := &Telemetry{
		Cid:       ms.Cid,
		NumGroups: len(ms.GetGroups()),
		NumZeros:  len(ms.GetZeros()),
		Version:   x.Version(),
		OS:        runtime.GOOS,
		Arch:      runtime.GOARCH,
	}
	for _, g := range ms.GetGroups() {
		t.NumAlphas += len(g.GetMembers())
		for _, tablet := range g.GetTablets() {
			t.NumTablets++
			t.DiskUsageMB += tablet.GetSpace()
		}
	}
	t.DiskUsageMB /= (1 << 20)
	t.ClusterSize = t.NumAlphas + t.NumZeros
	return t
}

// NewAlphaTelemetry returns a Telemetry struct that holds information about the state of alpha
// server.
func NewAlphaTelemetry(ms *pb.MembershipState, graphqlQueryCount uint64,
	nonGraphqlQueryCount uint64) *Telemetry {
	t := &Telemetry{
		Cid:                  ms.Cid,
		Version:              x.Version(),
		OS:                   runtime.GOOS,
		Arch:                 runtime.GOARCH,
		GraphqlQueryCount:    graphqlQueryCount,
		NonGraphqlQueryCount: nonGraphqlQueryCount,
	}
	return t
}

// Post reports the Telemetry to the stats server.
func (t *Telemetry) Post() error {
	data, err := json.Marshal(t)
	if err != nil {
		return err
	}
	url := keenURL + "/dev"
	if len(t.Version) > 0 {
		url = keenURL + "/pings"
	}
	req, err := http.NewRequest("POST", url, bytes.NewBuffer(data))
	if err != nil {
		return err
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Authorization", "D0398E8C83BB30F67C519FDA6175975F680921890C35B36C34BE1095445"+
		"97497CA758881BD7D56CC2355A2F36B4560102CBC3279AC7B27E5391372C36A31167EB0D06BF3764894AD20"+
		"A0554BAFF14C292A40BC252BB9FF008736A0FD1D44E085")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return err
	}

	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return err
	}
	glog.V(2).Infof("Telemetry response status: %v", resp.Status)
	glog.V(2).Infof("Telemetry response body: %s", body)
	return nil
}