add webserver handler to get orchestrator info

Author: ad-astra-video

server/handlers.go

@@ -24,6 +24,7 @@ import (
 	"github.com/livepeer/go-livepeer/eth"
 	"github.com/livepeer/go-livepeer/eth/types"
 	"github.com/livepeer/go-livepeer/monitor"
+	"github.com/livepeer/go-livepeer/net"
 	"github.com/livepeer/go-livepeer/pm"
 	"github.com/livepeer/lpms/ffmpeg"
 	"github.com/pkg/errors"
@@ -400,6 +401,67 @@ func (s *LivepeerServer) getSessionPoolInfoHandler() http.Handler {
 
 }
 
+type orchInfoResponse struct {
+	OrchInfo *net.OrchestratorInfo `json:"orchestrator_info"`
+	Status   string                `json:"status"`
+	Took     int64                 `json:"took_ms"`
+}
+
+func (s *LivepeerServer) getOrchestratorInfoHandler() http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if s.LivepeerNode.NodeType == core.BroadcasterNode {
+			//create GetOrchestrator request
+			b := core.NewBroadcaster(s.LivepeerNode)
+			b_sig, err := b.Sign([]byte(fmt.Sprintf("%v", b.Address().Hex())))
+			if err != nil {
+				respond500(w, "could not create sig")
+				return
+			}
+			req := &net.OrchestratorRequest{Address: b.Address().Bytes(), Sig: b_sig}
+			//send GetOrchestrator request
+			orch_url := r.Header.Get("Url")
+			if orch_url != "" {
+				url, err := url.ParseRequestURI(orch_url)
+				if err != nil {
+					respond400(w, "could not parse url")
+					return
+				}
+				orchInfoResp := orchInfoResponse{}
+
+				client, conn, err := startOrchestratorClient(context.Background(), url)
+				defer conn.Close()
+				if conn != nil && err == nil {
+					start := time.Now()
+					orch_info, err := client.GetOrchestrator(context.Background(), req)
+					if err != nil {
+						orchInfoResp.Status = "failed"
+						orchInfoResp.Took = time.Since(start).Milliseconds()
+						glog.Errorf("getorchestrator request failed: %v", err)
+						respondJson(w, orchInfoResp)
+						return
+					}
+					//parse net.OrchestratorInfo to json
+					orchInfoResp.Status = "success"
+					orchInfoResp.Took = time.Since(start).Milliseconds()
+					orchInfoResp.OrchInfo = orch_info
+					//send orchestrator info received
+					respondJson(w, orchInfoResp)
+					return
+				} else {
+					respond500(w, "getorchestrator request failed: "+err.Error())
+					return
+				}
+			} else {
+				respond400(w, "url not provided")
+				return
+			}
+		} else {
+			respond400(w, "not broadcaster node")
+			return
+		}
+	})
+}
+
 // Rounds
 func currentRoundHandler(client eth.LivepeerEthClient) http.Handler {
 	return mustHaveClient(client, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

server/webserver.go

@@ -52,6 +52,7 @@ func (s *LivepeerServer) cliWebServerHandlers(bindAddr string) *http.ServeMux {
 	mux.Handle("/setMaxPriceForCapability", mustHaveFormParams(s.setMaxPriceForCapability(), "maxPricePerUnit", "pixelsPerUnit", "currency", "pipeline", "modelID"))
 	mux.Handle("/getAISessionPoolsInfo", s.getAIPoolsInfoHandler())
 	mux.Handle("/getSessionPoolInfo", s.getSessionPoolInfoHandler())
+	mux.Handle("/getOrchestratorInfo", s.getOrchestratorInfoHandler())
 
 	// Rounds
 	mux.Handle("/currentRound", currentRoundHandler(client))