featurebase/bench/prettify.go

package bench

import "time"

// wrapper type to force human-readable JSON output
type PrettyDuration time.Duration

// MarshalJSON returns a nicely formatted duration, instead of it just being
// treated like an int.
func (d PrettyDuration) MarshalJSON() ([]byte, error) {
	s := time.Duration(d).String()
	return []byte("\"" + s + "\""), nil
}

// Recursively replaces elements of ugly types with their pretty wrappers
func Prettify(m map[string]interface{}) map[string]interface{} {
	newmap := make(map[string]interface{})
	for k, v := range m {
		switch v.(type) {
		case map[string]interface{}:
			newmap[k] = Prettify(v.(map[string]interface{}))
		case []time.Duration:
			newslice := make([]PrettyDuration, len(v.([]time.Duration)))
			slice := v.([]time.Duration)
			for n, e := range slice {
				newslice[n] = PrettyDuration(e)
			}
			newmap[k] = newslice
		case time.Duration:
			newmap[k] = PrettyDuration(v.(time.Duration))
		default:
			if interv, ok := v.([]map[string]interface{}); ok {
				for i, iv := range interv {
					interv[i] = Prettify(iv)
				}
			}
			newmap[k] = v
		}
	}
	return newmap
}