convox/pkg/manifest/service.go

package manifest

import (
	"crypto/sha1"
	"fmt"
	"sort"
	"strings"
)

type Service struct {
	Name string `yaml:"-"`

	Agent       ServiceAgent          `yaml:"agent,omitempty"`
	Build       ServiceBuild          `yaml:"build,omitempty"`
	Command     string                `yaml:"command,omitempty"`
	Domains     ServiceDomains        `yaml:"domain,omitempty"`
	Drain       int                   `yaml:"drain,omitempty"`
	Environment Environment           `yaml:"environment,omitempty"`
	Health      ServiceHealth         `yaml:"health,omitempty"`
	Image       string                `yaml:"image,omitempty"`
	Init        bool                  `yaml:"init,omitempty"`
	Internal    bool                  `yaml:"internal,omitempty"`
	Port        ServicePortScheme     `yaml:"port,omitempty"`
	Ports       []ServicePortProtocol `yaml:"ports,omitempty"`
	Privileged  bool                  `yaml:"privileged,omitempty"`
	Resources   []string              `yaml:"resources,omitempty"`
	Scale       ServiceScale          `yaml:"scale,omitempty"`
	Singleton   bool                  `yaml:"singleton,omitempty"`
	Sticky      bool                  `yaml:"sticky,omitempty"`
	Test        string                `yaml:"test,omitempty"`
	Volumes     []string              `yaml:"volumes,omitempty"`
}

type Services []Service

type ServiceAgent struct {
	Enabled bool `yaml:"enabled,omitempty"`
}

type ServiceBuild struct {
	Args     []string `yaml:"args,omitempty"`
	Manifest string   `yaml:"manifest,omitempty"`
	Path     string   `yaml:"path,omitempty"`
}

type ServiceDomains []string

type ServiceHealth struct {
	Grace    int
	Interval int
	Path     string
	Timeout  int
}

type ServicePortProtocol struct {
	Port     int    `yaml:"port,omitempty"`
	Protocol string `yaml:"protocol,omitempty"`
}

type ServicePortScheme struct {
	Port   int    `yaml:"port,omitempty"`
	Scheme string `yaml:"scheme,omitempty"`
}

type ServiceScale struct {
	Count   ServiceScaleCount
	Cpu     int
	Memory  int
	Targets ServiceScaleTargets `yaml:"targets,omitempty"`
}

type ServiceScaleCount struct {
	Min int
	Max int
}

type ServiceScaleMetric struct {
	Aggregate  string
	Dimensions map[string]string
	Namespace  string
	Name       string
	Value      float64
}

type ServiceScaleMetrics []ServiceScaleMetric

type ServiceScaleTargets struct {
	Cpu      int
	Custom   ServiceScaleMetrics
	Memory   int
	Requests int
}

func (s Service) BuildHash(key string) string {
	return fmt.Sprintf("%x", sha1.Sum([]byte(fmt.Sprintf("key=%q build[path=%q, manifest=%q, args=%v] image=%q", key, s.Build.Path, s.Build.Manifest, s.Build.Args, s.Image))))
}

func (s Service) Domain() string {
	if len(s.Domains) < 1 {
		return ""
	}

	return s.Domains[0]
}

func (s Service) EnvironmentDefaults() map[string]string {
	defaults := map[string]string{}

	for _, e := range s.Environment {
		switch parts := strings.Split(e, "="); len(parts) {
		case 2:
			defaults[parts[0]] = parts[1]
		}
	}

	return defaults
}

func (s Service) EnvironmentKeys() string {
	kh := map[string]bool{}

	for _, e := range s.Environment {
		kh[strings.Split(e, "=")[0]] = true
	}

	keys := []string{}

	for k := range kh {
		keys = append(keys, k)
	}

	sort.Strings(keys)

	return strings.Join(keys, ",")
}

func (s Service) GetName() string {
	return s.Name
}

func (s Service) Autoscale() bool {
	if s.Agent.Enabled {
		return false
	}

	switch {
	case s.Scale.Count.Min == s.Scale.Count.Max:
		return false
	case s.Scale.Targets.Cpu > 0:
		return true
	case len(s.Scale.Targets.Custom) > 0:
		return true
	case s.Scale.Targets.Memory > 0:
		return true
	case s.Scale.Targets.Requests > 0:
		return true
	}

	return false
}

func (ss Services) External() Services {
	return ss.Filter(func(s Service) bool {
		return !s.Internal
	})
}

func (ss Services) Filter(fn func(s Service) bool) Services {
	fss := Services{}

	for _, s := range ss {
		if fn(s) {
			fss = append(fss, s)
		}
	}

	return fss
}

func (ss Services) Routable() Services {
	return ss.Filter(func(s Service) bool {
		return s.Port.Port > 0
	})
}