如何一边监听客户端连接一边监听终端输入？-Golang-PHP中文网

如何一边监听客户端连接一边监听终端输入？

心靈之曲

发布： 2024-11-04 12:54:50

原创

648人浏览过

如何一边监听客户端连接一边监听终端输入？

一边监听客户端连接一边监听终端输入的go代码实践

在go中构建一个可同时监听客户端连接和终端输入的服务器程序需要解决并发更新和阻塞问题。

问题背景

在最初的代码中，服务端通过阻塞性的fmt.scan 读取终端输入。这会导致监听客户端连接的goroutine长期阻塞，无法正常响应新连接请求。

优化方案

听脑AI

听脑AI语音，一款专注于音视频内容的工作学习助手，为用户提供便捷的音视频内容记录、整理与分析功能。

378

查看详情

为了解决问题，需要使用非阻塞方式读取终端输入。改进后的代码如下：

package main

import (
    "fmt"
    "os"
)

var recvFromTerminalChan = make(chan string)
var recvFromNetChan = make(chan string)

func main() {
    go readFromTerminal()

    for {
        select {
        // 从终端输入通道接收数据
        case str := <-recvFromTerminalChan:
            go sendCommandToNet(str)
        // 从网络通道接收数据
        case str := <-recvFromNetChan:
            go doContentFromNet(str)
        }
    }
}

func readFromTerminal() {
    for {
        var input string
        fmt.Scan(&input)
        fmt.Println("input content:", input)
        recvFromTerminalChan <- input
    }
}

func sendCommandToNet(command string) {
    fmt.Println("command:", command)
    content := command

    recvFromNetChan <- content
}

func doContentFromNet(content string) {
    fmt.Println("content from net:", content)
}

登录后复制

改进要点