Skip to content

Commit

Permalink
add tcpserver name opt
Browse files Browse the repository at this point in the history
  • Loading branch information
liujianping committed Dec 22, 2019
1 parent 2fde950 commit dd4c94b
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion tcpserver.go
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ type Handler func(ctx context.Context, conn net.Conn) error

//Server represent tcpserver
type Server struct {
name string
network string
address string
handler Handler
Expand All @@ -21,6 +22,13 @@ type Server struct {
wgroup sync.WaitGroup
}

//Name option for tcpserver
func Name(name string) ServerOpt {
return func(srv *Server) {
srv.name = name
}
}

//Network option for listener
func Network(inet string) ServerOpt {
return func(srv *Server) {
Expand Down Expand Up @@ -70,6 +78,7 @@ type ServerOpt func(*Server)
//NewServer create a new tcpserver
func New(opts ...ServerOpt) *Server {
serv := &Server{
name: "tcpserver",
network: "tcp",
}
for _, opt := range opts {
Expand All @@ -85,7 +94,7 @@ func (srv *Server) Serve(ctx context.Context) error {
if err != nil {
return err
}
log.Println("tcpserver serving at ", srv.network, srv.address)
log.Println(srv.name, " serving at ", srv.network, srv.address)
srv.listener = ln
}
if srv.tlsc != nil {
Expand Down

0 comments on commit dd4c94b

Please sign in to comment.