码迷,mamicode.com
首页 > 编程语言 > 详细

[日常] Go语言圣经-基于select的多路复用习题

时间:2018-05-05 20:46:24      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:one   mes   语句   sleep   sele   基于   class   fat   log   

练习 8.8: 使用select来改造8.3节中的echo服务器,为其增加超时,这样服务器可以在客户端10秒中没有任何喊话时自动断开连接。

reverb3.go

package main

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

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处理连接
        }
}
/*
1.启动一个goroutine,for死循环让他不能断掉
select语句case判断两个channel
一个是10秒后断掉连接
另一个是接收标准输入后发送过来的channel,接收到值后,启动goroutinue输出

2.for循环接收标准输入,接收到后发送给message的channel
*/
func handleConn(c net.Conn) {
        input := bufio.NewScanner(c)
        var wg sync.WaitGroup
        var message = make(chan string)
        wg.Add(1)
        go func() {
                defer wg.Done()
                for {
                        select {
                        case <-time.After(10 * time.Second):
                                c.Close()
                        case mes := <-message:
                                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, mes, 1*time.Second)

                        }
                }
        }()
        for input.Scan() {
                text := input.Text()
                message <- text
        }

        wg.Wait()
        //cw := c.(*net.TCPConn)
        //cw.CloseWrite()

        c.Close()
}

  

[日常] Go语言圣经-基于select的多路复用习题

标签:one   mes   语句   sleep   sele   基于   class   fat   log   

原文地址:https://www.cnblogs.com/taoshihan/p/8995716.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!