github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-09-07 07:46:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix windows

Browse Source
This commit is contained in:
ffdfgdfg
2019-12-27 23:24:19 +08:00
parent 5e1b0be81c
commit 9b29bd64bd
3 changed files with 73 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
lib/mux/mux.go
View File

@@ -7,7 +7,6 @@ import (
"net"
"os"
"sync/atomic"
"syscall"
"time"
"github.com/astaxie/beego/logs"
@@ -64,26 +63,6 @@ func NewMux(c net.Conn, connType string) *Mux {
return m
}
func getConnFd(c net.Conn) (fd *os.File, err error) {
switch c.(type) {
case *net.TCPConn:
fd, err = c.(*net.TCPConn).File()
if err != nil {
return
}
return
case *net.UDPConn:
fd, err = c.(*net.UDPConn).File()
if err != nil {
return
}
return
default:
err = errors.New("mux:unknown conn type, only tcp or kcp")
return
}
}
func (s *Mux) NewConn() (*conn, error) {
if s.IsClose {
return nil, errors.New("the mux has closed")
@@ -442,7 +421,7 @@ func (Self *bandwidth) SetCopySize(n uint16) {
func (Self *bandwidth) calcBandWidth() {
t := Self.readStart.Sub(Self.lastReadStart)
bufferSize, err := syscall.GetsockoptInt(int(Self.fd.Fd()), syscall.SOL_SOCKET, syscall.SO_RCVBUF)
bufferSize, err := sysGetSock(Self.fd)
//logs.Warn(bufferSize)
if err != nil {
logs.Warn(err)
@@ -451,7 +430,7 @@ func (Self *bandwidth) calcBandWidth() {
}
if Self.bufLength >= uint32(bufferSize) {
atomic.StoreUint64(&Self.readBandwidth, math.Float64bits(float64(Self.bufLength)/t.Seconds()))
// calculate the hole socket buffer, the time meaning to fill the buffer
// calculate the whole socket buffer, the time meaning to fill the buffer
//logs.Warn(Self.Get())
} else {
Self.calcThreshold = uint32(bufferSize)