Add modify backends

src/api/servers.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/yyyar/gobetween/config"
+	"github.com/yyyar/gobetween/core"
 	"github.com/yyyar/gobetween/manager"
 	"github.com/yyyar/gobetween/stats"
 )
@@ -65,6 +66,28 @@ func attachServers(app *gin.RouterGroup) {
 		c.IndentedJSON(http.StatusOK, nil)
 	})
 
+	/**
+	 * Update backends with name :name
+	 * pass in []core.Backend
+	 */
+	app.POST("/servers/:name/backends", func(c *gin.Context) {
+
+		name := c.Param("name")
+
+		cfg := []core.Backend{}
+		if err := c.BindJSON(&cfg); err != nil {
+			c.IndentedJSON(http.StatusBadRequest, err.Error())
+			return
+		}
+
+		if err := manager.UpdateBackends(name, &cfg); err != nil {
+			c.IndentedJSON(http.StatusConflict, err.Error())
+			return
+		}
+
+		c.IndentedJSON(http.StatusOK, nil)
+	})
+
 	/**
 	 * Get server stats
 	 */

src/core/server.go

@@ -26,6 +26,11 @@ type Server interface {
 	 */
 	Stop()
 
+	/**
+	 * UpdateBackends allows you to set a new backend config
+	 */
+	UpdateBackends(backends *[]Backend)
+
 	/**
 	 * Get server configuration
 	 */

src/discovery/discovery.go

@@ -163,11 +163,16 @@ func (this *Discovery) send() bool {
 	case <-this.stop:
 		return false
 	default:
-		this.out <- *this.backends
+		this.SendBackends(this.backends)
 		return true
 	}
 }
 
+// SendBackends updates the backends
+func (this *Discovery) SendBackends(backends *[]core.Backend) {
+	this.out <- *backends
+}
+
 /**
  * wait waits for interval or stop
  * returns true if waiting was successfull

src/manager/manager.go

@@ -217,6 +217,20 @@ func Create(name string, cfg config.Server) error {
 	return nil
 }
 
+// UpdateBackends allows you to update the backends of an existing server
+func UpdateBackends(name string, backends *[]core.Backend) error {
+	servers.RLock()
+	server, ok := servers.m[name]
+	servers.RUnlock()
+	if !ok {
+		return errors.New("Server with this name does not exist: " + name)
+	}
+
+	server.UpdateBackends(backends)
+
+	return nil
+}
+
 /**
  * Delete server stopping all active connections
  */

src/server/tcp/server.go

@@ -181,6 +181,11 @@ func (this *Server) Start() error {
 	return nil
 }
 
+// UpdateBackends updates the backends
+func (this *Server) UpdateBackends(backends *[]core.Backend) {
+	this.scheduler.Discovery.SendBackends(backends)
+}
+
 /**
  * Handle client disconnection
  */

src/server/udp/server.go

@@ -431,3 +431,8 @@ func (this *Server) fireAndForget(pool *connPool, clientAddr *net.UDPAddr, buf [
 func (this *Server) Stop() {
 	this.stop <- true
 }
+
+// UpdateBackends updates the backends
+func (this *Server) UpdateBackends(backends *[]core.Backend) {
+	this.scheduler.Discovery.SendBackends(backends)
+}