diff options
| author | Mikhail Osipov <mike.osipov@gmail.com> | 2020-05-27 17:29:59 +0300 |
|---|---|---|
| committer | Mikhail Osipov <mike.osipov@gmail.com> | 2020-05-27 17:29:59 +0300 |
| commit | e00995cf2a5968ae94d4dc0abfe21a66dae5cd54 (patch) | |
| tree | efc7c389818b9713d92dde5f231a71fde0185699 /pkg/server | |
| parent | a29c4614431c0f7b705ee793f0a98348cdd29595 (diff) | |
rename hook tee to dump
Diffstat (limited to 'pkg/server')
| -rw-r--r-- | pkg/server/hook/dump.go (renamed from pkg/server/hook/tee.go) | 36 |
1 files changed, 18 insertions, 18 deletions
diff --git a/pkg/server/hook/tee.go b/pkg/server/hook/dump.go index 34b2eca..d6bcd90 100644 --- a/pkg/server/hook/tee.go +++ b/pkg/server/hook/dump.go @@ -13,19 +13,19 @@ import ( "tunnel/pkg/server/queue" ) -const teeDefaultFile = "/tmp/tunnel/dump" +const dumpDefaultFile = "/tmp/tunnel/dump" -type tee struct { +type dump struct { f *os.File - h *teeHook + h *dumpHook } -type teeHook struct { +type dumpHook struct { file string time bool } -func (t *tee) dump(s string, p []byte) error { +func (t *dump) write(s string, p []byte) error { var out bytes.Buffer if t.h.time { @@ -46,41 +46,41 @@ func (t *tee) dump(s string, p []byte) error { return nil } -func (t *tee) Send(rq, wq queue.Q) error { +func (t *dump) Send(rq, wq queue.Q) error { for b := range rq { - t.dump(">", b) + t.write(">", b) wq <- b } return nil } -func (t *tee) Recv(rq, wq queue.Q) error { +func (t *dump) Recv(rq, wq queue.Q) error { for b := range rq { - t.dump("<", b) + t.write("<", b) wq <- b } return nil } -func (t *tee) Close() { +func (t *dump) Close() { t.f.Close() } -func (h *teeHook) where(env env.Env) string { +func (h *dumpHook) where(env env.Env) string { if h.file != "" { return h.file } - if v := env.GetLocal("tee.file"); v != "" { + if v := env.GetLocal("dump.file"); v != "" { return v } - return teeDefaultFile + return dumpDefaultFile } -func (h *teeHook) Open(env env.Env) (interface{}, error) { +func (h *dumpHook) Open(env env.Env) (interface{}, error) { file := h.where(env) dir := path.Dir(file) @@ -91,7 +91,7 @@ func (h *teeHook) Open(env env.Env) (interface{}, error) { tid, sid := env.Get("tunnel"), env.Get("stream") name := fmt.Sprintf("%s.%s.%s", file, tid, sid) - t := &tee{h: h} + t := &dump{h: h} if f, err := os.Create(name); err != nil { return nil, err @@ -102,8 +102,8 @@ func (h *teeHook) Open(env env.Env) (interface{}, error) { return t, nil } -func newTeeHook(opts opts.Opts) (hook, error) { - h := &teeHook{ +func newDumpHook(opts opts.Opts) (hook, error) { + h := &dumpHook{ file: opts["file"], time: opts.Bool("time"), } @@ -111,5 +111,5 @@ func newTeeHook(opts opts.Opts) (hook, error) { } func init() { - register("tee", newTeeHook) + register("dump", newDumpHook) } |
