|
|
@@ -4,19 +4,10 @@ import (
|
|
4
|
4
|
"context"
|
|
5
|
5
|
"errors"
|
|
6
|
6
|
"io"
|
|
7
|
|
- "sync"
|
|
8
|
7
|
|
|
9
|
8
|
"github.com/9seconds/mtg/v2/essentials"
|
|
10
|
|
- "github.com/9seconds/mtg/v2/mtglib/internal/tls"
|
|
11
|
9
|
)
|
|
12
|
10
|
|
|
13
|
|
-var bufPool = sync.Pool{
|
|
14
|
|
- New: func() any {
|
|
15
|
|
- b := make([]byte, tls.MaxRecordPayloadSize)
|
|
16
|
|
- return &b
|
|
17
|
|
- },
|
|
18
|
|
-}
|
|
19
|
|
-
|
|
20
|
11
|
func Relay(ctx context.Context, log Logger, telegramConn, clientConn essentials.Conn) {
|
|
21
|
12
|
defer telegramConn.Close() //nolint: errcheck
|
|
22
|
13
|
defer clientConn.Close() //nolint: errcheck
|
|
|
@@ -44,13 +35,13 @@ func Relay(ctx context.Context, log Logger, telegramConn, clientConn essentials.
|
|
44
|
35
|
}
|
|
45
|
36
|
|
|
46
|
37
|
func pump(log Logger, src, dst essentials.Conn, direction string) {
|
|
47
|
|
- bp := bufPool.Get().(*[]byte)
|
|
48
|
|
- defer bufPool.Put(bp)
|
|
|
38
|
+ buf := acquireBuffer()
|
|
|
39
|
+ defer releaseBuffer(buf)
|
|
49
|
40
|
|
|
50
|
41
|
defer src.CloseRead() //nolint: errcheck
|
|
51
|
42
|
defer dst.CloseWrite() //nolint: errcheck
|
|
52
|
43
|
|
|
53
|
|
- n, err := io.CopyBuffer(src, dst, *bp)
|
|
|
44
|
+ n, err := io.CopyBuffer(src, dst, *buf)
|
|
54
|
45
|
|
|
55
|
46
|
switch {
|
|
56
|
47
|
case err == nil:
|