Bumped v0.5.8

Signed-off-by: Vishal Rana <[email protected]>

Author: vishr

Makefile

@@ -1,5 +1,5 @@
 IMAGE = labstack/tunnel
-VERSION = 0.5.7
+VERSION = 0.5.8
 
 publish:
 	git tag v$(VERSION)

daemon/connection.go

@@ -37,7 +37,7 @@ type (
 
   Connection struct {
     server        *Server
-    startChan     chan bool
+    startChan     chan error
     acceptChan    chan net.Conn
     reconnectChan chan error
     stopChan      chan bool
@@ -87,7 +87,7 @@ func (s *Server) newConnection(req *ConnectRequest) (c *Connection, err error) {
   id, _ := gonanoid.Nanoid()
   c = &Connection{
     server:        s,
-    startChan:     make(chan bool),
+    startChan:     make(chan error),
     acceptChan:    make(chan net.Conn),
     reconnectChan: make(chan error),
     stopChan:      make(chan bool),
@@ -206,7 +206,7 @@ RECONNECT:
   c.retries = 0
   if !c.started {
     c.started = true
-    c.startChan <- true
+    c.startChan <- c.server.findConnection(c)
   }
   log.Infof("connection %s is online", c.Name)
 

daemon/server.go

@@ -60,8 +60,7 @@ func (s *Server) Connect(req *ConnectRequest, rep *ConnectReply) (err error) {
     return
   }
   go c.connect()
-  <-c.startChan
-  return s.findConnection(c)
+  return <-c.startChan
 }
 
 func (s *Server) PS(req *PSRequest, rep *PSReply) (err error) {
@@ -100,6 +99,7 @@ func Start() {
   go func() {
     <-c
     log.Warn("stopping daemon")
+    os.Exit(0)
   }()
 
   // Listen