github/kcp-go
4
0
Fork 0
mirror of https://github.com/xtaci/kcp-go.git synced 2024-04-21 12:32:15 +00:00
Code Issues Releases Wiki Activity

optimize counting

Browse Source
This commit is contained in:
xtaci
2019-05-08 13:50:07 +08:00
parent e06f294160
commit a537bb2089
2 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tx_generic.go
+1 -3
View File
@@ -11,18 +11,16 @@ func (s *UDPSession) txLoop() {
select {
case txqueue := <-s.chTxQueue:
nbytes := 0
npkts := 0
for k := range txqueue {
if n, err := s.conn.WriteTo(txqueue[k], s.remote); err == nil {
nbytes += n
npkts++
} else {
s.notifyWriteError(err)
}
xmitBuf.Put(txqueue[k])
}
atomic.AddUint64(&DefaultSnmp.OutPkts, uint64(npkts))
atomic.AddUint64(&DefaultSnmp.OutPkts, uint64(len(txqueue)))
atomic.AddUint64(&DefaultSnmp.OutBytes, uint64(nbytes))
case <-s.die:
return
tx_linux.go
+8 -12
View File
@@ -27,15 +27,13 @@ func (s *UDPSession) txLoopIPv4() {
select {
case txqueue := <-s.chTxQueue:
if len(txqueue) > 0 {
var nbytes uint64
var npkts uint64
nbytes := 0
for k := range txqueue {
idx := k % batchSize
msgs[idx].Addr = s.remote
msgs[idx].Buffers = [][]byte{txqueue[k]}
nbytes += uint64(len(txqueue[k]))
npkts++
nbytes += len(txqueue[k])
if (k+1)%batchSize == 0 {
if _, err := conn.WriteBatch(msgs, 0); err != nil {
@@ -54,8 +52,8 @@ func (s *UDPSession) txLoopIPv4() {
xmitBuf.Put(txqueue[k])
}
atomic.AddUint64(&DefaultSnmp.OutPkts, npkts)
atomic.AddUint64(&DefaultSnmp.OutBytes, nbytes)
atomic.AddUint64(&DefaultSnmp.OutPkts, uint64(len(txqueue)))
atomic.AddUint64(&DefaultSnmp.OutBytes, uint64(nbytes))
}
case <-s.die:
return
@@ -71,15 +69,13 @@ func (s *UDPSession) txLoopIPv6() {
select {
case txqueue := <-s.chTxQueue:
if len(txqueue) > 0 {
var nbytes uint64
var npkts uint64
nbytes := 0
for k := range txqueue {
idx := k % batchSize
msgs[idx].Addr = s.remote
msgs[idx].Buffers = [][]byte{txqueue[k]}
nbytes += uint64(len(txqueue[k]))
npkts++
nbytes += len(txqueue[k])
if (k+1)%batchSize == 0 {
if _, err := conn.WriteBatch(msgs, 0); err != nil {
@@ -98,8 +94,8 @@ func (s *UDPSession) txLoopIPv6() {
xmitBuf.Put(txqueue[k])
}
atomic.AddUint64(&DefaultSnmp.OutPkts, npkts)
atomic.AddUint64(&DefaultSnmp.OutBytes, nbytes)
atomic.AddUint64(&DefaultSnmp.OutPkts, uint64(len(txqueue)))
atomic.AddUint64(&DefaultSnmp.OutBytes, uint64(nbytes))
}
case <-s.die:
return