github
/
v2ray-core
mirror of https://github.com/v2ray/v2ray-core
1
0
Fork 0
Code Issues Releases Wiki Activity

refine locker in sending worker

pull/215/head
v2ray 2016-07-06 00:02:22 +02:00
parent 78ef65e17b
commit 5448ec1fe4
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
transport/internet/kcp/sending.go
View File

@ -307,6 +307,8 @@ func (this *SendingWorker) ProcessSegment(current uint32, seg *AckSegment) {
func (this *SendingWorker) Push(b []byte) int {
nBytes := 0
this.Lock()
defer this.Unlock()
for len(b) > 0 && !this.queue.IsFull() {
var size int
if len(b) > int(this.conn.mss) {
@ -316,9 +318,7 @@ func (this *SendingWorker) Push(b []byte) int {
}
seg := NewDataSegment()
seg.Data = alloc.NewSmallBuffer().Clear().Append(b[:size])
this.Lock()
this.queue.Push(seg)
this.Unlock()
b = b[size:]
nBytes += size
}
@ -331,7 +331,7 @@ func (this *SendingWorker) Write(seg Segment) {
dataSeg.Conv = this.conn.conv
dataSeg.SendingNext = this.firstUnacknowledged
dataSeg.Opt = 0
if this.conn.state == StateReadyToClose {
if this.conn.State() == StateReadyToClose {
dataSeg.Opt = SegmentOptionClose
}