Golang学习笔记之简易聊天系统服务器的搭建

下面先列举一下程序使用到的函数,省的大家去找，直接拷贝官方api的解释吧。

func DialTCP(net string, laddr, raddr *TCPAddr) (*TCPConn, error)
DialTCP在网络协议net上连接本地地址laddr和远端地址raddr。
net必须是"tcp"、"tcp4"、"tcp6"；如果laddr不是nil，将使用它作为本地地址，否则自动选择一个本地地址。

func ResolveTCPAddr(net, addr string) (*TCPAddr, error)
ResolveTCPAddr将addr作为TCP地址解析并返回。
参数addr格式为"host:port"或"[ipv6-host%zone]:port"，解析得到网络名和端口名；net必须是"tcp"、"tcp4"或"tcp6"。

func ListenTCP(net string, laddr *TCPAddr) (*TCPListener, error)
ListenTCP在本地TCP地址laddr上声明并返回一个*TCPListener，
net参数必须是"tcp"、"tcp4"、"tcp6"，如果laddr的端口字段为0，函数将选择一个当前可用的端口，可以用Listener的Addr方法获得该端口。TCPListener代表一个TCP网络的监听者。使用者应尽量使用Listener接口而不是假设（网络连接为）TCP。

func (l *TCPListener) AcceptTCP() (*TCPConn, error)
//AcceptTCP接收下一个呼叫，并返回一个新的*TCPConn。
//TCPConn代表一个TCP网络连接，实现了Conn接口。

Write(b []byte) (n int, err error)
Write从连接中写入数据
Write方法可能会在超过某个固定时间限制后超时返回错误，该错误的Timeout()方法返回真
Read(b []byte) (n int, err error)
Read从连接中读取数据
Read方法可能会在超过某个固定时间限制后超时返回错误，该错误的Timeout()方法返回真
var Args []string
Args保管了命令行参数，第一个是程序名。

服务端代码

server.go

package main

import (
    "encoding/json"
    "fmt"
    "log"
    "net"
)

func main() {
    //端口号
    StartServer("8080")
}

//结构体
type person struct {
    news string
    ip   string
}

//StartServer 启动服务器
func StartServer(s string) {

    // 获取tcp地址
    tcpAddr, err := net.ResolveTCPAddr("tcp4", ":"+s)
    if err != nil {
        log.Printf("resolve tcp addr failed: %v\n", err)
        return
    }
    //连接池，用来保存所有人的数据
    conns := make(map[string]net.Conn)
    //消息信道，为有缓冲信道，当然缓冲内存也可以增大
    messages := make(chan person, 10)

    // 监听
    listener, err := net.ListenTCP("tcp", tcpAddr)
    if err != nil {
        log.Printf("listen tcp port failed: %v\n", err)
        return
    }
    //开启发送消息的协程
    go BroadCastMessage(conns, messages)

    //时刻监测有没有新的消息发送过来
    for {
        /*
            func (l *TCPListener) AcceptTCP() (*TCPConn, error)

            AcceptTCP接收下一个呼叫，并返回一个新的*TCPConn。
            TCPConn代表一个TCP网络连接，实现了Conn接口。
        */
        conn, err := listener.AcceptTCP()
        if err != nil {
            fmt.Println("链接失败")
            continue
        }
        conns[conn.RemoteAddr().String()] = conn
        go HandlerMessage(conn, conns, messages)
    }
}

//HandlerMessage 检查发送来的消息
/*
    //conn:返回的新的*TCPConn
    conns:连接池
    messages:消息通道
*/
func HandlerMessage(conn net.Conn, conns map[string]net.Conn, messages chan person) {
    //切片用来暂存消息
    buf := make([]byte, 1024)
    for {
        //从链接里面读取数据，写给buf
        len, err := conn.Read(buf)
        if err != nil {
            conn.Close()
            delete(conns, conn.RemoteAddr().String())
            break
        }
        //消息写入结构体，并且把发送者的ip一块写入
        p := person{
            news: string(buf[:len]),
            ip:   conn.RemoteAddr().String(),
        }
        //发送给信道
        messages <- p
        fmt.Println(string(buf[:len]))
    }
}

//BroadCastMessage 发送消息
/*
    conns:连接池
    messages:消息通道
*/
func BroadCastMessage(conns map[string]net.Conn, messages chan person) {
    for {
        //读取信道里面的消息
        mess := <-messages
        //发送给所有人
        for k, v := range conns {
            //不发送给自己
            if k != mess.ip {
                var m map[string]interface{}
                //将发送过来消息的news序列化为map
                json.Unmarshal([]byte(mess.news), &m)
                //拼接字符串
                msg := m["time"].(string) + "\n" + m["userID"].(string) + ":" + m["message"].(string)
                //发送
                _, err := v.Write([]byte(msg))
                //如果失败关闭v所属人的协程，继续监测
                if err != nil {
                    delete(conns, k)
                    v.Close()
                    continue
                }
            }
        }
    }
}

个人端代码
net.go

package main

import (
    "encoding/json"
    "fmt"
    "log"
    "net"
    "os"
    "time"
)

func main() {
    /*
        var Args []string
        Args保管了命令行参数，第一个是程序名。
    */
    tcpAddr, _ := net.ResolveTCPAddr("tcp", os.Args[1])
    //拨号
    conn, _ := net.DialTCP("tcp", nil, tcpAddr)
    var p = make(map[string]interface{})
    p["userID"] = os.Args[2]
    go HandlerMessage(conn, p)
    ReceivesMessage(conn)
}

//HandlerMessage 向服务器发送数据
func HandlerMessage(conn net.Conn, p map[string]interface{}) {
    for {
        var input string

        // 接收输入消息，放到input变量中
        fmt.Scanln(&input)

        //用户端退出啊
        if input == "/q" || input == "/quit" {
            fmt.Println("Byebye ...")
            conn.Close()
            os.Exit(0)
        }
        //发送的信息
        p["message"] = input
        //时间戳
        p["time"] = time.Now()
        // 只处理有内容的消息
        if len(input) > 0 {
            //序列化为json
            msg, err := json.Marshal(p)
            if err != nil {
                //如果不成功，返回错误
                fmt.Println("错误", err)
            } else {
                //发送数据
                _, err := conn.Write([]byte(string(msg)))
                //没有发送成功
                if err != nil {
                    conn.Close()
                    break
                }
            }
        }
    }
}

//ReceivesMessage 接收服务器消息
func ReceivesMessage(conn net.Conn) {
    // 接收来自服务器端的广播消息
    buf := make([]byte, 1024)
    for {
        length, err := conn.Read(buf)
        if err != nil {
            log.Printf("recv server msg failed: %v\n", err)
            conn.Close()
            os.Exit(0)
            break
        }

        fmt.Println(string(buf[0:length]))
    }
}

测试步骤，只用本地测试一下，懒的上传云服务器了

一、启动服务器
二、启动个人端，当然这里我们启动两个，两个以上才能看出来效果不是吗

三、发送消息
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 213,864评论 6赞 494
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 91,175评论 3赞 387
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 159,401评论 0赞 349
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 57,170评论 1赞 286
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 66,276评论 6赞 385
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 50,364评论 1赞 292
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 39,401评论 3赞 412
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 38,179评论 0赞 269
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 44,604评论 1赞 306
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,902评论 2赞 328
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,070评论 1赞 341
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 34,751评论 4赞 337
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 40,380评论 3赞 319
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,077评论 0赞 21
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,312评论 1赞 267
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 46,924评论 2赞 365
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 43,957评论 2赞 351

推荐阅读更多精彩内容

  • 高效阅读的4点建议
    移动互联网时代，信息传播与分享越来越方便，其渠道和形式也是多种多样。一方面，我们接触到的信息量大大增加，知识的获取...
    星禾笔记阅读 447评论 0赞 2
  • 天城往事—和朋友的骑行
    我高中那会儿，不长不短，也有三年光载，说到朋友的话，不多不少，只有两三个。两个还是三个？到现在可能还没弄清楚。有的...
    云牧1阅读 380评论 0赞 2
  • 时光记忆2
    皮皮虾2019阅读 208评论 0赞 0