diff options
| author | Mikhail Osipov <mike.osipov@gmail.com> | 2020-02-28 23:34:45 +0300 |
|---|---|---|
| committer | Mikhail Osipov <mike.osipov@gmail.com> | 2020-02-28 23:34:45 +0300 |
| commit | 11501b56a751d2959480aaeaf2036eff586e5629 (patch) | |
| tree | 69f6d7aa7534c2d5198381e84cee2f55250996d4 /pkg/server/tunnel.go | |
| parent | e0ca83d77c748a8d55707a101c5fc6ab41c6727b (diff) | |
add tee module
Diffstat (limited to 'pkg/server/tunnel.go')
| -rw-r--r-- | pkg/server/tunnel.go | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/pkg/server/tunnel.go b/pkg/server/tunnel.go index 189703d..91d7533 100644 --- a/pkg/server/tunnel.go +++ b/pkg/server/tunnel.go @@ -6,6 +6,7 @@ import ( "io" "log" "sort" + "strconv" "strings" "sync" "time" @@ -140,6 +141,7 @@ func (t *tunnel) newStream(in, out socket.Channel) *stream { } s.env.Set("tunnel", t.id) + s.env.Set("stream", strconv.Itoa(s.id)) s.run() @@ -208,7 +210,12 @@ func (s *stream) run() { s.channel(s.in, rq, wq) for _, m := range s.t.m { - send, recv, _ := module.Open(m, s.env) + send, recv, err := module.Open(m, s.env) + if err != nil { + // FIXME: abort stream on error + log.Println(s.t, s, m, err) + continue + } if send != nil { q := queue.New() |