mirror of https://github.com/q191201771/naza
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
713 B
Go
45 lines
713 B
Go
// Copyright 2019, Chef. All rights reserved.
|
|
// https://github.com/q191201771/naza
|
|
//
|
|
// Use of this source code is governed by a MIT-style license
|
|
// that can be found in the License file.
|
|
//
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
package mockserver
|
|
|
|
import (
|
|
"net"
|
|
"sync"
|
|
)
|
|
|
|
type MockAcceptServer struct {
|
|
l net.Listener
|
|
conns []net.Conn
|
|
m sync.Mutex
|
|
}
|
|
|
|
func (s *MockAcceptServer) Run(addr string) (err error) {
|
|
s.m.Lock()
|
|
s.l, err = net.Listen("tcp", addr)
|
|
s.m.Unlock()
|
|
if err != nil {
|
|
return
|
|
}
|
|
c, err := s.l.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
s.conns = append(s.conns, c)
|
|
return
|
|
}
|
|
|
|
func (s *MockAcceptServer) Dispose() {
|
|
s.m.Lock()
|
|
if s.l != nil {
|
|
s.l.Close()
|
|
|
|
}
|
|
s.m.Unlock()
|
|
}
|