前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >[日常] Go语言圣经--并发的循环习题

[日常] Go语言圣经--并发的循环习题

作者头像
唯一Chat
发布2019-09-10 12:40:35
6710
发布2019-09-10 12:40:35
举报
文章被收录于专栏:陶士涵的菜地陶士涵的菜地

练习 8.4: 修改reverb2服务器,在每一个连接中使用sync.WaitGroup来计数活跃的echo goroutine。当计数减为零时,关闭TCP连接的写入,像练习8.3中一样。验证一下你的修改版netcat3客户端会一直等待所有的并发“喊叫”完成,即使是在标准输入流已经关闭的情况下。

重点:等待所有的goroutine执行完成后再关闭TCP连接

package main

import (
        "bufio"
        "fmt"
        "log"
        "net"
        "strings"
        "time"
        "sync"
)

func main() {
        listener, err := net.Listen("tcp", ":8040")
        if err != nil {
                log.Fatal(err)
        }   

        for {
                conn, err := listener.Accept()
                if err != nil {
                        log.Print(err) // e.g., connection aborted
                        continue
                }   
                go handleConn(conn) //新建goroutines处理连接
        }   
}

func handleConn(c net.Conn) {
        input := bufio.NewScanner(c)
        var wg sync.WaitGroup
        //var  ch =make(chan struct{})
        for input.Scan() {
                wg.Add(1)
                go func(c net.Conn, shout string, delay time.Duration) {
                        defer wg.Done()
                        fmt.Fprintln(c, "\t", strings.ToUpper(shout))
                        time.Sleep(delay)
                        fmt.Fprintln(c, "\t", shout)
                        time.Sleep(delay)
                        fmt.Fprintln(c, "\t", strings.ToLower(shout))
                        //ch<-struct{}{}

                }(c, input.Text(), 1*time.Second)
        }   
        wg.Wait()
                //cw := c.(*net.TCPConn)
          //cw.CloseWrite()
        
        c.Close()
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2018-05-02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档