docs/main.go

149 lines
3.2 KiB
Go

package main
import (
"os"
"time"
log "github.com/Sirupsen/logrus"
"github.com/codegangsta/cli"
"github.com/docker/libcluster/api"
"github.com/docker/libcluster/discovery"
"github.com/docker/libcluster/scheduler"
"github.com/docker/libcluster/scheduler/filter"
"github.com/docker/libcluster/scheduler/strategy"
"github.com/docker/libcluster/swarm"
)
type logHandler struct {
}
func (h *logHandler) Handle(e *libcluster.Event) error {
log.Printf("event -> type: %q time: %q image: %q container: %q", e.Type, e.Time.Format(time.RubyDate), e.Container.Image, e.Container.Id)
return nil
}
func main() {
app := cli.NewApp()
app.Name = "swarm"
app.Usage = "docker clustering"
app.Version = "0.0.1"
app.Flags = []cli.Flag{
cli.BoolFlag{
Name: "debug",
Usage: "debug mode",
EnvVar: "DEBUG",
},
}
app.Before = func(c *cli.Context) error {
log.SetOutput(os.Stderr)
if c.Bool("debug") {
log.SetLevel(log.DebugLevel)
}
return nil
}
clusterFlags := []cli.Flag{
cli.StringFlag{
Name: "token",
Value: "",
Usage: "cluster token",
EnvVar: "SWARM_TOKEN",
},
cli.StringFlag{
Name: "addr",
Value: "127.0.0.1:4243",
Usage: "ip to advertise",
EnvVar: "SWARM_ADDR",
},
}
app.Commands = []cli.Command{
{
Name: "manage",
ShortName: "m",
Usage: "manage a docker cluster",
Flags: clusterFlags,
Action: func(c *cli.Context) {
refresh := func(cluster *libcluster.Cluster, nodes []string) error {
for _, addr := range nodes {
if cluster.Node(addr) == nil {
n := libcluster.NewNode(addr, addr)
if err := n.Connect(nil); err != nil {
return err
}
if err := cluster.AddNode(n); err != nil {
return err
}
}
}
return nil
}
cluster := libcluster.NewCluster()
cluster.Events(&logHandler{})
if c.String("token") != "" {
nodes, err := discovery.FetchSlaves(c.String("token"))
if err != nil {
log.Fatal(err)
}
if err := refresh(cluster, nodes); err != nil {
log.Fatal(err)
}
go func() {
for {
time.Sleep(25 * time.Second)
nodes, err = discovery.FetchSlaves(c.String("token"))
if err == nil {
refresh(cluster, nodes)
}
}
}()
} else {
if err := refresh(cluster, c.Args()); err != nil {
log.Fatal(err)
}
}
s := scheduler.NewScheduler(cluster, &strategy.BinPackingPlacementStrategy{}, []filter.Filter{})
log.Fatal(api.ListenAndServe(cluster, s, c.String("addr")))
},
},
{
Name: "join",
ShortName: "j",
Usage: "join a docker cluster",
Flags: clusterFlags,
Action: func(c *cli.Context) {
if c.String("token") == "" {
log.Fatal("--token required to join a cluster")
}
if err := discovery.RegisterSlave(c.String("addr"), c.String("token")); err != nil {
log.Fatal(err)
}
// heartbeat every 25 seconds
go func() {
for {
time.Sleep(25 * time.Second)
if err := discovery.RegisterSlave(c.String("addr"), c.String("token")); err != nil {
log.Error(err)
}
}
}()
},
},
}
log.Fatal(app.Run(os.Args))
}