mirror of
https://github.com/ScuroNeko/mtg.git
synced 2026-08-31 15:14:03 +03:00
More idiomatic Golang
This commit is contained in:
+13
-16
@@ -3,6 +3,7 @@ package mtglib
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net"
|
"net"
|
||||||
@@ -102,7 +103,7 @@ func newConnProxyProtocol(source, target essentials.Conn) *connProxyProtocol {
|
|||||||
// Both directions update the same timestamp so that activity in one direction
|
// Both directions update the same timestamp so that activity in one direction
|
||||||
// prevents the other (idle) direction from timing out.
|
// prevents the other (idle) direction from timing out.
|
||||||
type idleTracker struct {
|
type idleTracker struct {
|
||||||
lastActive atomic.Int64 // unix nanos
|
lastActive atomic.Pointer[time.Time]
|
||||||
timeout time.Duration
|
timeout time.Duration
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -114,13 +115,12 @@ func newIdleTracker(timeout time.Duration) *idleTracker {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (t *idleTracker) touch() {
|
func (t *idleTracker) touch() {
|
||||||
t.lastActive.Store(time.Now().UnixNano())
|
stamp := time.Now()
|
||||||
|
t.lastActive.Store(&stamp)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *idleTracker) isIdle() bool {
|
func (t *idleTracker) isIdle() bool {
|
||||||
last := time.Unix(0, t.lastActive.Load())
|
return time.Since(*t.lastActive.Load()) >= t.timeout
|
||||||
|
|
||||||
return time.Since(last) >= t.timeout
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type connIdleTimeout struct {
|
type connIdleTimeout struct {
|
||||||
@@ -130,25 +130,22 @@ type connIdleTimeout struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c connIdleTimeout) Read(b []byte) (int, error) {
|
func (c connIdleTimeout) Read(b []byte) (int, error) {
|
||||||
|
var netErr net.Error
|
||||||
|
|
||||||
for {
|
for {
|
||||||
c.SetReadDeadline(time.Now().Add(c.tracker.timeout)) //nolint: errcheck
|
c.SetReadDeadline(time.Now().Add(c.tracker.timeout)) //nolint: errcheck
|
||||||
|
|
||||||
n, err := c.Conn.Read(b)
|
n, err := c.Conn.Read(b)
|
||||||
if n > 0 {
|
|
||||||
|
switch {
|
||||||
|
case err == nil:
|
||||||
c.tracker.touch()
|
c.tracker.touch()
|
||||||
|
return n, nil
|
||||||
return n, err //nolint: wrapcheck
|
case errors.As(err, &netErr) && netErr.Timeout() && !c.tracker.isIdle():
|
||||||
}
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
if netErr, ok := err.(net.Error); ok && netErr.Timeout() && !c.tracker.isIdle() { //nolint: errorlint
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0, err //nolint: wrapcheck
|
return n, err
|
||||||
}
|
|
||||||
|
|
||||||
return 0, nil
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -160,7 +160,6 @@ type ProxyOpts struct {
|
|||||||
|
|
||||||
// DoppelGangerDRS defines if TLS Dynamic Record Sizing is active.
|
// DoppelGangerDRS defines if TLS Dynamic Record Sizing is active.
|
||||||
DoppelGangerDRS bool
|
DoppelGangerDRS bool
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p ProxyOpts) valid() error {
|
func (p ProxyOpts) valid() error {
|
||||||
|
|||||||
Reference in New Issue
Block a user