arrgh_example_test.go

// Copyright ©2014 Dan Kortschak. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package arrgh_test

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/url"
	"os"
	"regexp"
	"strings"
	"time"

	"github.com/kortschak/arrgh"
)

func Example_1() {
	r, err := arrgh.NewLocalSession("", "", 3000, 10*time.Second)
	if err != nil {
		log.Fatal(err)
	}
	defer r.Close()

	// Send a query to get a JSON representation of results
	// from rnorm(n=10, mean=10, sd=10).
	resp, err := r.Post(
		"library/stats/R/rnorm/json",
		"application/json",
		url.Values{"digits": []string{"10"}},
		strings.NewReader(`{"n":10, "mean": 10, "sd":10}`),
	)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()

	// Decode the results in to a slice of float64.
	var rnorm []float64
	dec := json.NewDecoder(resp.Body)
	err = dec.Decode(&rnorm)

	fmt.Println(rnorm, err)
}

func mask(r io.Reader) io.Reader {
	re := regexp.MustCompile("x[0-9a-f]{10}")
	var buf bytes.Buffer
	io.Copy(&buf, r)
	return bytes.NewReader(re.ReplaceAll(buf.Bytes(), []byte("xXXXXXXXXXX")))
}

func Example_3() {
	r, err := arrgh.NewRemoteSession("http://public.opencpu.org", "", 10*time.Second)
	if err != nil {
		log.Fatal(err)
	}
	defer r.Close()

	// Upload the contents of the file "mydata.csv" and send
	// it to the read.csv function.
	f, err := os.Open("mydata.csv")
	if err != nil {
		log.Fatal(err)
	}
	content, body, err := arrgh.Multipart(
		arrgh.Params{"header": "FALSE"},
		arrgh.Files{"file": f},
	)
	f.Close()
	if err != nil {
		log.Fatal(err)
	}
	resp, err := r.Post(
		"library/utils/R/read.csv",
		content,
		nil,
		body,
	)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()

	io.Copy(os.Stdout, mask(resp.Body))

	// Output:
	//
	// /ocpu/tmp/xXXXXXXXXXX/R/.val
	// /ocpu/tmp/xXXXXXXXXXX/stdout
	// /ocpu/tmp/xXXXXXXXXXX/source
	// /ocpu/tmp/xXXXXXXXXXX/console
	// /ocpu/tmp/xXXXXXXXXXX/info
	// /ocpu/tmp/xXXXXXXXXXX/files/mydata.csv
}