initial middleware refactor

server/pool.go

@@ -0,0 +1,53 @@
+// Copyright 2016 The Chihaya Authors. All rights reserved.
+// Use of this source code is governed by the BSD 2-Clause license,
+// which can be found in the LICENSE file.
+
+package server
+
+import (
+	"sync"
+
+	"github.com/chihaya/chihaya/config"
+	"github.com/chihaya/chihaya/tracker"
+)
+
+// Pool represents a running pool of servers.
+type Pool struct {
+	servers []Server
+	wg      sync.WaitGroup
+}
+
+// StartPool creates a new pool of servers specified by the provided config and
+// runs them.
+func StartPool(cfgs []config.ServerConfig, tkr *tracker.Tracker) (*Pool, error) {
+	var servers []Server
+	var wg sync.WaitGroup
+
+	for _, cfg := range cfgs {
+		srv, err := New(&cfg, tkr)
+		if err != nil {
+			return nil, err
+		}
+
+		wg.Add(1)
+		go func(srv Server) {
+			defer wg.Done()
+			srv.Start()
+		}(srv)
+
+		servers = append(servers, srv)
+	}
+
+	return &Pool{
+		servers: servers,
+		wg:      wg,
+	}, nil
+}
+
+// Stop safely shuts down a pool of servers.
+func (p *Pool) Stop() {
+	for _, srv := range p.servers {
+		srv.Stop()
+	}
+	p.wg.Wait()
+}