From 11501b56a751d2959480aaeaf2036eff586e5629 Mon Sep 17 00:00:00 2001 From: Mikhail Osipov Date: Fri, 28 Feb 2020 23:34:45 +0300 Subject: add tee module --- pkg/server/tunnel.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'pkg/server/tunnel.go') 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() -- cgit v1.2.3-70-g09d2