Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 1 | package main |
| 2 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 3 | import ( |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 4 | "context" |
| 5 | "fmt" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 6 | "net/http" |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 7 | "strings" |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 8 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 9 | "code.hackerspace.pl/q3k/topo/graph" |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 10 | "github.com/gobuffalo/packr" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 11 | "github.com/golang/glog" |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 12 | "github.com/q3k/statusz" |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 13 | ) |
| 14 | |
| 15 | type ServiceConfig struct { |
| 16 | DebugListen string |
| 17 | } |
| 18 | |
| 19 | type Service struct { |
| 20 | gr *graph.Graph |
| 21 | config ServiceConfig |
| 22 | } |
| 23 | |
| 24 | func NewService(gr *graph.Graph, c ServiceConfig) *Service { |
| 25 | return &Service{ |
| 26 | gr: gr, |
| 27 | config: c, |
| 28 | } |
| 29 | } |
| 30 | |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 31 | type TopologyStatus struct { |
| 32 | } |
| 33 | |
| 34 | const topologyFragment = ` |
| 35 | <script src="/assets/viz.js"></script> |
| 36 | <script> |
| 37 | |
| 38 | var viz = new Viz({ workerURL: '/assets/full.render.js' }); |
| 39 | var xmlhttp = new XMLHttpRequest(); |
| 40 | xmlhttp.onreadystatechange = function() { |
| 41 | if (this.readyState == 4 && this.status == 200) { |
| 42 | var dot = this.responseText; |
| 43 | viz.renderSVGElement(dot) |
| 44 | .then(function(element) { |
| 45 | document.body.appendChild(element); |
| 46 | }); |
| 47 | } |
| 48 | }; |
| 49 | xmlhttp.open('GET', '/debug/graphviz'); |
| 50 | xmlhttp.send(); |
| 51 | |
| 52 | </script> |
| 53 | ` |
| 54 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 55 | func (s *Service) Run() { |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 56 | assets := packr.NewBox("./assets") |
| 57 | http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(assets))) |
| 58 | |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 59 | http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { |
| 60 | http.Redirect(w, r, "/debug/status", http.StatusSeeOther) |
| 61 | }) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 62 | |
| 63 | http.HandleFunc("/debug/graphviz", func(w http.ResponseWriter, r *http.Request) { |
| 64 | fmt.Fprintf(w, "graph G {\n") |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 65 | fmt.Fprintf(w, " rankdir = LR\n") |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 66 | for _, machine := range s.gr.Machines { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 67 | portNames := []string{} |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 68 | for _, port := range machine.Ports { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 69 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 70 | portNames = append(portNames, name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 71 | } |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 72 | ports := strings.Join(portNames, "|") |
| 73 | fmt.Fprintf(w, " %s [shape=record label=\"{ %s | { %s }}\"]\n", machine.Name, machine.Name, ports) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 74 | } |
| 75 | for _, sw := range s.gr.Switches { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 76 | portNames := []string{} |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 77 | for _, port := range sw.Ports { |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 78 | name := fmt.Sprintf("<%s> %s", port.Name, port.Name) |
| 79 | portNames = append(portNames, name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 80 | } |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 81 | ports := strings.Join(portNames, "|") |
| 82 | fmt.Fprintf(w, " %s [shape=record label=\"{{ %s } | %s}\"]\n", sw.Name, ports, sw.Name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 83 | } |
| 84 | for _, machine := range s.gr.Machines { |
| 85 | for _, port := range machine.Ports { |
| 86 | if port.OtherEnd == nil { |
| 87 | continue |
| 88 | } |
Serge Bazanski | de869df | 2018-10-06 13:55:49 +0100 | [diff] [blame^] | 89 | fmt.Fprintf(w, " %s:%q -- %s:%q\n", machine.Name, port.Name, port.OtherEnd.Switch.Name, port.OtherEnd.Name) |
Serge Bazanski | c7be4a1 | 2018-10-06 13:18:05 +0100 | [diff] [blame] | 90 | } |
| 91 | } |
| 92 | fmt.Fprintf(w, "}\n") |
| 93 | }) |
| 94 | |
| 95 | statusz.AddStatusPart("Topology", topologyFragment, func(ctx context.Context) interface{} { |
| 96 | return &TopologyStatus{} |
| 97 | }) |
Serge Bazanski | 4676508 | 2018-10-06 12:32:01 +0100 | [diff] [blame] | 98 | glog.Infof("Debug listening on %s....", s.config.DebugListen) |
| 99 | http.ListenAndServe(s.config.DebugListen, nil) |
Serge Bazanski | 8d7843c | 2018-10-04 10:37:36 +0100 | [diff] [blame] | 100 | } |