代码拉取完成,页面将自动刷新
同步操作将从 CloudWeGo/netpoll 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
Listener
-> net.Listener
(#29)
// Copyright 2021 CloudWeGo Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package netpoll
import (
"context"
"net"
"sync/atomic"
"testing"
"time"
)
func TestListenerDialer(t *testing.T) {
network := "tcp"
addr := ":1234"
ln, err := CreateListener(network, addr)
MustNil(t, err)
defer time.Sleep(10 * time.Millisecond)
defer ln.Close()
stop := make(chan int)
trigger := make(chan int)
defer close(stop)
defer close(trigger)
msg := []byte("0123456789")
go func() {
for {
select {
case <-stop:
err := ln.Close()
MustNil(t, err)
return
default:
}
conn, err := ln.Accept()
if conn == nil && err == nil {
continue
}
go func(conn net.Conn) {
<-trigger
buf := make([]byte, 10)
n, err := conn.Read(buf)
MustNil(t, err)
Equal(t, n, len(msg))
Equal(t, string(buf[:n]), string(msg))
n, err = conn.Write(buf)
MustNil(t, err)
Equal(t, n, len(msg))
}(conn)
}
}()
// trigger
var closed, read int32
dialer := NewDialer()
callback := func(connection Connection) error {
atomic.StoreInt32(&closed, 1)
return nil
}
onRequest := func(ctx context.Context, connection Connection) error {
atomic.StoreInt32(&read, 1)
err := connection.Close()
MustNil(t, err)
return err
}
for i := 0; i < 10; i++ {
conn, err := dialer.DialConnection(network, addr, time.Second)
if err != nil {
continue
}
conn.AddCloseCallback(callback)
conn.SetOnRequest(onRequest)
MustNil(t, err)
n, err := conn.Write([]byte(msg))
MustNil(t, err)
Equal(t, n, len(msg))
time.Sleep(10 * time.Millisecond)
trigger <- 1
time.Sleep(10 * time.Millisecond)
Equal(t, atomic.LoadInt32(&read), int32(1))
Equal(t, atomic.LoadInt32(&closed), int32(1))
}
}
func TestConvertListener(t *testing.T) {
network, address := "unix", "mock.test.sock"
ln, err := net.Listen(network, address)
if err != nil {
panic(err)
}
udsln, _ := ln.(*net.UnixListener)
// udsln.SetUnlinkOnClose(false)
nln, err := ConvertListener(udsln)
if err != nil {
panic(err)
}
err = nln.Close()
if err != nil {
panic(err)
}
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。