go语言编写，简单聊天室程序 1

最新推荐文章于 2022-04-19 22:09:50 发布

eclipser1987

最新推荐文章于 2022-04-19 22:09:50 发布

阅读量3.3k

点赞数

分类专栏： golang

本文链接：https://blog.csdn.net/eclipser1987/article/details/17222541

版权

golang 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用Go语言进行网络编程，通过编写一个简单的聊天室程序作为起点，展示Go语言在网络编程中的应用。文章包含程序的运行命令及预期测试结果。

摘要由CSDN通过智能技术生成

go语言网络编程，先从聊天室开始：

/**
 * Created by Vicky.H on 13-12-6.
 * eclipser@163.com
 */
package main

import (
	"fmt"
	"net"
	"strconv"
	"bytes"
	_ "bufio"
	_ "strings"
)

var m map[string]*net.Conn

func main() {

	m = make(map[string]*net.Conn, 10)

	listener ,err := net.Listen("tcp", ":8080")
	if err != nil {
		fmt.Errorf(err.Error())
		return
	}
	defer listener.Close()

	index := 0
	for {
		conn, err := listener.Accept()
		if err != nil {
			fmt.Errorf(err.Error())
			return
		}
		name := "用户:" + strconv.Itoa(index)
		defer connCloseHandler(name, conn)
		m[name] = &conn
		index++
		go connHandler(name, conn)
	}
}

func connCloseHandler(name string, conn net.Conn) {
	conn.Close()
	delete(m, name)
	for key := range(m)  {
		if key != name {
			(*(m[key])).Write(bytes.NewBufferString(name + " 离开房间\n").Bytes())
			(*(m[key])).Write(bytes.NewBufferString("当前房间人数:" + strconv.Itoa(len(m)) + "\n").Bytes())
		}
	}
}


func connHandler (name string, conn net.Conn) {
	conn.Write(bytes.NewBufferString("当前房间人数:" + strconv.Itoa(len(m)) + "\n").Bytes())
	for key := range(m)  {
		if key != name {
			(*(m[key])).Write(bytes.NewBufferString(name + " 进入房间\n").Bytes())
		}
	}
	for {
		buf := make([]byte,1024)
		conn.Read(buf)
		msg := string(buf)
		conn.Write(bytes.NewBufferString("我说: " + msg).Bytes())
		for key := range(m)  {
			if key != name {
				(*(m[key])).Write(bytes.NewBufferString(name + " 说: " + msg).Bytes())
			}
		}
	}
}

运行：go run main.go

测试结果