From cb234ca4a77131ec7cf95cc0f82a818c831398ff Mon Sep 17 00:00:00 2001
From: FUJIWARA Shunichiro <fujiwara.shunichiro@gmail.com>
Date: Fri, 13 Jan 2017 16:41:35 +0900
Subject: [PATCH] Accept service name as node name

---
 assets/scripts/dashboard.js |  4 ++--
 dashboard.go                | 47 ++++++++++++++++++++++++++++++++-----
 2 files changed, 43 insertions(+), 8 deletions(-)

diff --git a/assets/scripts/dashboard.js b/assets/scripts/dashboard.js
index eb127b4..7a9181d 100644
--- a/assets/scripts/dashboard.js
+++ b/assets/scripts/dashboard.js
@@ -19,7 +19,7 @@ var StatusSelector = React.createClass({
                 <button type="button" value="warning" className="btn btn-default navbar-btn alert-warning" onClick={this.handleChange}>Warning</button>
                 <button type="button" value="danger" className="btn btn-default navbar-btn alert-danger" onClick={this.handleChange}>Danger</button>
                 <button type="button" value="info" className="btn btn-default navbar-btn alert-info" onClick={this.handleChange}>Info</button>
-                <input type="text" id="nodeFilter" className="form-control" placeholder="nodename"  onKeyUp={this.handleNodeGrep}/>
+                <input type="text" id="nodeFilter" className="form-control" placeholder="node | service"  onKeyUp={this.handleNodeGrep}/>
                 <input type="text" id="keyFilter" className="form-control" placeholder="key"  onKeyUp={this.handleKeyGrep}/>
               </form>
             </div>
@@ -208,7 +208,7 @@ var Dashboard = React.createClass({
         <table className="table table-bordered">
           <thead>
             <tr>
-              <th>node</th>
+              <th>node | service</th>
               <th>address</th>
               <th>key</th>
               <th className="item_timestamp_col">timestamp</th>
diff --git a/dashboard.go b/dashboard.go
index 628c6dc..b98f808 100644
--- a/dashboard.go
+++ b/dashboard.go
@@ -24,6 +24,7 @@ var (
 	Version     string
 	ExtAssetDir string
 	Nodes       []Node
+	Services    map[string][]string
 	mutex       sync.RWMutex
 )
 
@@ -126,7 +127,8 @@ func main() {
 		log.Println("trigger:", trigger)
 		go watchForTrigger(trigger)
 	}
-	go updateNodeList()
+	go updateNodes()
+	go updateServices()
 
 	log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), nil))
 }
@@ -205,7 +207,7 @@ func kvApiProxy(w http.ResponseWriter, r *http.Request) {
 		items := make([]Item, 0, len(kvps))
 		for _, kv := range kvps {
 			item := kv.NewItem()
-			if itemInNodes(&item) {
+			if itemInCatalog(&item) {
 				items = append(items, item)
 			}
 		}
@@ -236,7 +238,7 @@ func watchForTrigger(command string) {
 		currentItem := make(map[string]Item)
 		for _, kv := range kvps {
 			item := kv.NewItem()
-			if !itemInNodes(&item) {
+			if !itemInCatalog(&item) {
 				continue
 			}
 
@@ -248,7 +250,7 @@ func watchForTrigger(command string) {
 		}
 		for _, kv := range kvps {
 			item := kv.NewItem()
-			if !itemInNodes(&item) {
+			if !itemInCatalog(&item) {
 				continue
 			}
 			if _, exist := currentItem[item.Category]; !exist {
@@ -336,7 +338,7 @@ func invokePipe(command string, src io.Reader) error {
 	return cmdErr
 }
 
-func updateNodeList() {
+func updateNodes() {
 	var index int64
 	for {
 		resp, newIndex, err := callConsulAPI(
@@ -358,7 +360,28 @@ func updateNodeList() {
 	}
 }
 
-func itemInNodes(item *Item) bool {
+func updateServices() {
+	var index int64
+	for {
+		resp, newIndex, err := callConsulAPI(
+			"/v1/catalog/services?index=" + strconv.FormatInt(index, 10) + "&wait=55s",
+		)
+		if err != nil {
+			log.Println("[error]", err)
+			time.Sleep(10 * time.Second)
+			continue
+		}
+		index = newIndex
+		defer resp.Body.Close()
+		dec := json.NewDecoder(resp.Body)
+		mutex.Lock()
+		dec.Decode(&Services)
+		mutex.Unlock()
+		time.Sleep(1 * time.Second)
+	}
+}
+
+func itemInCatalog(item *Item) bool {
 	mutex.RLock()
 	defer mutex.RUnlock()
 	for _, node := range Nodes {
@@ -367,6 +390,18 @@ func itemInNodes(item *Item) bool {
 			return true
 		}
 	}
+	for name, tags := range Services {
+		if item.Node == name {
+			item.Address = "service"
+			return true
+		}
+		for _, tag := range tags {
+			if item.Node == fmt.Sprintf("%s.%s", tag, name) {
+				item.Address = "service"
+				return true
+			}
+		}
+	}
 	return false
 }