mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 02:00:43 +00:00
2ba50b8b3d
```
$ gvt delete golang.org/x/net/context
$ gvt fetch golang.org/x/net/context
2018/07/23 18:03:49 Fetching: golang.org/x/net/context
$ git grep -l "golang.org/x/net/context" | grep -v vendor | xargs sed -i '' 's:golang.org/x/net/context:context:g'
$ git grep -l "context/ctxhttp" | grep -v vendor | xargs sed -i '' 's:context/ctxhttp:golang.org/x/net/context/ctxhttp:g'
$ gofmt -s -w app
$ gofmt -s -w common
$ gofmt -s -w probe
$ gofmt -s -w prog
$ gofmt -s -w tools
```
fixed a bunch of:
```
cannot use func literal (type func("github.com/weaveworks/scope/vendor/golang.org/x/net/context".Context) error) as type func("context".Context) error
```
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
package app
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"context"
|
|
"github.com/gorilla/mux"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/weaveworks/scope/common/xfer"
|
|
)
|
|
|
|
// RegisterPipeRoutes registers the pipe routes
|
|
func RegisterPipeRoutes(router *mux.Router, pr PipeRouter) {
|
|
router.Methods("GET").
|
|
Name("api_pipe_pipeid_check").
|
|
Path("/api/pipe/{pipeID}/check").
|
|
HandlerFunc(requestContextDecorator(checkPipe(pr)))
|
|
|
|
router.Methods("GET").
|
|
Name("api_pipe_pipeid").
|
|
Path("/api/pipe/{pipeID}").
|
|
HandlerFunc(requestContextDecorator(handlePipeWs(pr, UIEnd)))
|
|
|
|
router.Methods("GET").
|
|
Name("api_pipe_pipeid_probe").
|
|
Path("/api/pipe/{pipeID}/probe").
|
|
HandlerFunc(requestContextDecorator(handlePipeWs(pr, ProbeEnd)))
|
|
|
|
router.Methods("DELETE", "POST").
|
|
Name("api_pipe_pipeid").
|
|
Path("/api/pipe/{pipeID}").
|
|
HandlerFunc(requestContextDecorator(deletePipe(pr)))
|
|
}
|
|
|
|
func checkPipe(pr PipeRouter) CtxHandlerFunc {
|
|
return func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
|
|
id := mux.Vars(r)["pipeID"]
|
|
exists, err := pr.Exists(ctx, id)
|
|
if err != nil {
|
|
respondWith(w, http.StatusInternalServerError, err)
|
|
} else if exists {
|
|
w.WriteHeader(http.StatusNoContent)
|
|
} else {
|
|
http.NotFound(w, r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func handlePipeWs(pr PipeRouter, end End) CtxHandlerFunc {
|
|
return func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
|
|
id := mux.Vars(r)["pipeID"]
|
|
pipe, endIO, err := pr.Get(ctx, id, end)
|
|
if err != nil {
|
|
// this usually means the pipe has been closed
|
|
log.Debugf("Error getting pipe %s: %v", id, err)
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
defer pr.Release(ctx, id, end)
|
|
|
|
conn, err := xfer.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
log.Errorf("Error upgrading pipe %s (%d) websocket: %v", id, end, err)
|
|
return
|
|
}
|
|
defer conn.Close()
|
|
|
|
if err := pipe.CopyToWebsocket(endIO, conn); err != nil && !xfer.IsExpectedWSCloseError(err) {
|
|
log.Errorf("Error copying to pipe %s (%d) websocket: %v", id, end, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func deletePipe(pr PipeRouter) CtxHandlerFunc {
|
|
return func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
|
|
pipeID := mux.Vars(r)["pipeID"]
|
|
log.Debugf("Deleting pipe %s", pipeID)
|
|
if err := pr.Delete(ctx, pipeID); err != nil {
|
|
respondWith(w, http.StatusInternalServerError, err)
|
|
}
|
|
}
|
|
}
|