Beispiel #1
0
// sendData handles the actual sending of data to the aggregator.
func sendData(cmd, svc string, u *whiplash.ClientUpdate) {
	// create a new aclient instance
	pc, err := pclient.NewTCP(pcconf)
	if err != nil {
		log.Println(err)
		return
	}
	defer pc.Close()
	// success, so turn the update into json
	jupdate, err := json.Marshal(u)
	if err != nil {
		log.Println(err)
		return
	}
	req := []byte(cmd)
	req = append(req, 32)
	req = append(req, jupdate...)
	_, err = pc.Dispatch(req)
	if err != nil {
		log.Println("err dispatching", cmd, "for", svc, ":", err)
		return
	}
}
Beispiel #2
0
func main() {
	flag.Parse()
	args = flag.Args()
	// read whiplash config. genconf should be FALSE in the
	// whiplash.New call for a wlq instance: we don't expect to hacve
	// ceph services around on a machine running a query.
	wl, err := whiplash.New(whipconf, false)
	if err != nil {
		quit(fmt.Errorf("can't read configuration file: %s\n", err))
	}

	// validate user input
	err = validateInput()
	if err != nil {
		quit(err)
	}
	// handle non-networked commands
	switch args[0] {
	case "help":
		showHelp()
		quit(nil)
	case "version":
		showVersion()
		quit(nil)
	}

	// set up configuration and create aclient instance
	pcconf := &pclient.Config{
		Addr:    wl.Aggregator.BindAddr + ":" + wl.Aggregator.QueryPort,
		Timeout: 100,
	}
	c, err := pclient.NewTCP(pcconf)
	if err != nil {
		quit(fmt.Errorf("can't connect to aggregator: %s", err))
	}
	defer c.Close()

	// stitch together the non-option arguments into our request
	req := strings.Join(flag.Args(), " ")
	// and dispatch it to the server!
	respj, err := c.Dispatch([]byte(req))
	if err != nil {
		quit(fmt.Errorf("sending request to aggregator failed: %s", err))
	}

	// vivify response and handle errors
	resp := new(whiplash.QueryResponse)
	err = json.Unmarshal(respj, &resp)
	if err != nil {
		quit(fmt.Errorf("failure unmarshaling json\nerror: %s\njson: %s", err, string(respj)))
	}
	if resp.Code >= 400 {
		quit(fmt.Errorf("there was a problem with the request:\n%s", string(respj)))
	}

	// if -j has been specified, print the raw response data and exit
	if dumpjson {
		fmt.Println(string(resp.Data))
		os.Exit(0)
	}

	// else, we have to hand off to a pretty-printing routine
	// TODO write a pretty-printing routine
}