New GraphQL schema

schema/float.go

@@ -0,0 +1,66 @@
+package schema
+
+import (
+	"errors"
+	"io"
+	"math"
+	"strconv"
+)
+
+// A custom float type is used so that (Un)MarshalJSON and
+// (Un)MarshalGQL can be overloaded and NaN/null can be used.
+// The default behaviour of putting every nullable value behind
+// a pointer has a bigger overhead.
+type Float float64
+
+var NaN Float = Float(math.NaN())
+
+func (f Float) IsNaN() bool {
+	return math.IsNaN(float64(f))
+}
+
+// NaN will be serialized to `null`.
+func (f Float) MarshalJSON() ([]byte, error) {
+	if f.IsNaN() {
+		return []byte("null"), nil
+	}
+
+	return []byte(strconv.FormatFloat(float64(f), 'f', 2, 64)), nil
+}
+
+// `null` will be unserialized to NaN.
+func (f *Float) UnmarshalJSON(input []byte) error {
+	s := string(input)
+	if s == "null" {
+		*f = NaN
+		return nil
+	}
+
+	val, err := strconv.ParseFloat(s, 64)
+	if err != nil {
+		return err
+	}
+	*f = Float(val)
+	return nil
+}
+
+// UnmarshalGQL implements the graphql.Unmarshaler interface.
+func (f *Float) UnmarshalGQL(v interface{}) error {
+	f64, ok := v.(float64)
+	if !ok {
+		return errors.New("invalid Float scalar")
+	}
+
+	*f = Float(f64)
+	return nil
+}
+
+// MarshalGQL implements the graphql.Marshaler interface.
+// NaN will be serialized to `null`.
+func (f Float) MarshalGQL(w io.Writer) {
+	if f.IsNaN() {
+		w.Write([]byte(`null`))
+	} else {
+		w.Write([]byte(strconv.FormatFloat(float64(f), 'f', 2, 64)))
+	}
+}

schema/metrics.go

@@ -0,0 +1,76 @@
+package schema
+
+import (
+	"fmt"
+	"io"
+)
+
+// Format of `data.json` files.
+type JobData map[string]*JobMetric
+
+type JobMetric struct {
+	Unit     string          `json:"unit"`
+	Scope    MetricScope     `json:"scope"`
+	Timestep int             `json:"timestep"`
+	Series   []*MetricSeries `json:"series"`
+}
+
+type MetricScope string
+
+const (
+	MetricScopeNode   MetricScope = "node"
+	MetricScopeSocket MetricScope = "socket"
+	MetricScopeCpu    MetricScope = "cpu"
+)
+
+func (e *MetricScope) UnmarshalGQL(v interface{}) error {
+	str, ok := v.(string)
+	if !ok {
+		return fmt.Errorf("enums must be strings")
+	}
+
+	*e = MetricScope(str)
+	if *e != "node" && *e != "socket" && *e != "cpu" {
+		return fmt.Errorf("%s is not a valid MetricScope", str)
+	}
+	return nil
+}
+
+func (e MetricScope) MarshalGQL(w io.Writer) {
+	fmt.Fprintf(w, "\"%s\"", e)
+}
+
+type MetricStatistics struct {
+	Avg float64 `json:"avg"`
+	Min float64 `json:"min"`
+	Max float64 `json:"max"`
+}
+
+type MetricSeries struct {
+	NodeID     string            `json:"node_id"`
+	Statistics *MetricStatistics `json:"statistics"`
+	Data       []Float           `json:"data"`
+}
+
+// Format of `meta.json` files.
+type JobMeta struct {
+	JobId     string   `json:"job_id"`
+	UserId    string   `json:"user_id"`
+	ProjectId string   `json:"project_id"`
+	ClusterId string   `json:"cluster_id"`
+	NumNodes  int      `json:"num_nodes"`
+	JobState  string   `json:"job_state"`
+	StartTime int64    `json:"start_time"`
+	Duration  int64    `json:"duration"`
+	Nodes     []string `json:"nodes"`
+	Tags      []struct {
+		Name string `json:"name"`
+		Type string `json:"type"`
+	} `json:"tags"`
+	Statistics map[string]struct {
+		Unit string  `json:"unit"`
+		Avg  float64 `json:"avg"`
+		Min  float64 `json:"min"`
+		Max  float64 `json:"max"`
+	} `json:"statistics"`
+}