More idiomatic Golang

This commit is contained in:
9seconds
2026-03-31 15:07:01 +02:00
parent 0c9fa5e710
commit b6427ee321
4 changed files with 22 additions and 26 deletions
+13 -16
View File
@@ -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
} }
} }
-1
View File
@@ -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 {