From b8a7b08984d9ecb8cce9c053c67c10d94009d7ad Mon Sep 17 00:00:00 2001
From: "xinfei.wu" <jiyeyuran@gmail.com>
Date: Fri, 1 Feb 2019 17:11:25 +0800
Subject: [PATCH] handle RegisterInterval option

---
 http.go | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/http.go b/http.go
index 9b1385c..c063573 100644
--- a/http.go
+++ b/http.go
@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"sort"
 	"sync"
+	"time"
 
 	log "github.com/micro/go-log"
 	"github.com/micro/go-micro/broker"
@@ -252,7 +253,31 @@ func (h *httpServer) Start() error {
 	go http.Serve(ln, handler)
 
 	go func() {
-		ch := <-h.exit
+		t := new(time.Ticker)
+
+		// only process if it exists
+		if opts.RegisterInterval > time.Duration(0) {
+			// new ticker
+			t = time.NewTicker(opts.RegisterInterval)
+		}
+
+		// return error chan
+		var ch chan error
+
+	Loop:
+		for {
+			select {
+			// register self on interval
+			case <-t.C:
+				if err := h.Register(); err != nil {
+					log.Log("Server register error: ", err)
+				}
+			// wait for exit
+			case ch = <-h.exit:
+				break Loop
+			}
+		}
+
 		ch <- ln.Close()
 
 		// deregister