slowdown.go 866 B

12345678910111213141516171819202122232425262728293031323334
  1. package proxydialer
  2. import (
  3. "context"
  4. "net"
  5. "net/netip"
  6. "github.com/metacubex/mihomo/component/slowdown"
  7. C "github.com/metacubex/mihomo/constant"
  8. )
  9. type SlowDownDialer struct {
  10. C.Dialer
  11. Slowdown *slowdown.SlowDown
  12. }
  13. func (d SlowDownDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
  14. return slowdown.Do(d.Slowdown, ctx, func() (net.Conn, error) {
  15. return d.Dialer.DialContext(ctx, network, address)
  16. })
  17. }
  18. func (d SlowDownDialer) ListenPacket(ctx context.Context, network, address string, rAddrPort netip.AddrPort) (net.PacketConn, error) {
  19. return slowdown.Do(d.Slowdown, ctx, func() (net.PacketConn, error) {
  20. return d.Dialer.ListenPacket(ctx, network, address, rAddrPort)
  21. })
  22. }
  23. func NewSlowDownDialer(d C.Dialer, sd *slowdown.SlowDown) SlowDownDialer {
  24. return SlowDownDialer{
  25. Dialer: d,
  26. Slowdown: sd,
  27. }
  28. }