diff --git a/app/rproxy/forwarder.go b/app/rproxy/forwarder.go index 6c7efd7..669baf5 100644 --- a/app/rproxy/forwarder.go +++ b/app/rproxy/forwarder.go @@ -30,7 +30,7 @@ func NewForwarder(ctx context.Context, lport, dport uint32, addrs ...string) (*F ctx: ctx, cancel: cancel, } - id := strconv.FormatUint(uint64(lport), 10) + id := strconv.FormatUint(uint64(lport), 10) forw.log = logger.NewLogger("forwarder:" + id) for _, addr := range addrs { dest := NewDestination(addr) @@ -38,7 +38,7 @@ func NewForwarder(ctx context.Context, lport, dport uint32, addrs ...string) (*F } portinfo := ":" + strconv.FormatUint(uint64(forw.Lport), 10) - laddr, err := net.ResolveTCPAddr("tcp", portinfo) + laddr, err := net.ResolveTCPAddr("tcp", portinfo) if err != nil { return forw, err } @@ -68,18 +68,18 @@ func (forw *Forwarder) Stop() error { } type Streamer struct { - source string - dest string - ctx context.Context - cancel context.CancelFunc + source string + dest string + ctx context.Context + cancel context.CancelFunc } func NewStreamer(ctx context.Context) *Streamer { - ctx, cancel := context.WithCancel(ctx) - return &Streamer{ - ctx: ctx, - cancel: cancel, - } + ctx, cancel := context.WithCancel(ctx) + return &Streamer{ + ctx: ctx, + cancel: cancel, + } } func (forw *Forwarder) handle(ctx context.Context, inconn net.Conn) { @@ -95,7 +95,7 @@ func (forw *Forwarder) handle(ctx context.Context, inconn net.Conn) { if err != nil { return } - defer outconn.Close() + defer outconn.Close() var wg sync.WaitGroup wg.Add(1) go forw.stream(&wg, inconn, outconn)