脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang → golang websocket简易聊天室

golang基于websocket实现的简易聊天室程序

 更新时间：2016年07月22日 17:05:48   作者：dotcoo  
这篇文章主要介绍了golang基于websocket实现的简易聊天室,分析了websocket的下载、安装及使用实现聊天室功能的相关技巧,需要的朋友可以参考下

本文实例讲述了golang基于websocket实现的简易聊天室。分享给大家供大家参考，具体如下：

先说点无关的,最近忙于工作没有更新博客,今天休息顺便把golang websocket研究了一下,挺好玩的,写了一个聊天室,分享给大家.

websocket包 : code.google.com/p/go.net/websocket

文档 : http://go.pkgdoc.org/code.google.com/p/go.net/websocket

首先安装websocket包

复制代码 代码如下:
go get code.google.com/p/go.net/websocket

下来就是聊天室代码了,呵呵

golang websocket 聊天室

复制代码 代码如下:
package main
import (
    "fmt"
    "io"
    "bufio"
    "net/http"
    "code.google.com/p/go.net/websocket"
    "container/list"
)
var connid int
var conns *list.List
func ChatroomServer(ws *websocket.Conn) {
    defer ws.Close()
    connid++
    id := connid
    fmt.Printf("connection id: %d\n", id)
    item := conns.PushBack(ws)
    defer conns.Remove(item)
    name := fmt.Sprintf("user%d", id)
    SendMessage(nil, fmt.Sprintf("welcome %s join\n", name))
    r := bufio.NewReader(ws)
    for {
        data, err := r.ReadBytes('\n')
        if err != nil {
            fmt.Printf("disconnected id: %d\n", id)
            SendMessage(item, fmt.Sprintf("%s offline\n", name))
            break
        }
        fmt.Printf("%s: %s", name, data)
        SendMessage(item, fmt.Sprintf("%s\t> %s", name, data))
    }
}
func SendMessage(self *list.Element, data string) {
    // for _, item := range conns {
    for item := conns.Front(); item != nil; item = item.Next() {
        ws, ok := item.Value.(*websocket.Conn)
        if !ok {
            panic("item not *websocket.Conn")
        }
        if item == self {
            continue
        }
        io.WriteString(ws, data)
    }
}
// 网页客户端
func Client(w http.ResponseWriter, r *http.Request) {
    html := `<!doctype html>
<html>
<head>
    <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
    <title>golang websocket chatroom</title>
    <script src="http://ajax.googleapis.com/ajax/libs/jquery/1.7.2/jquery.min.js"></script>
    <script>
        var ws = new WebSocket("ws://127.0.0.1:6611/chatroom");
        ws.onopen = function(e){
            console.log("onopen");
            console.dir(e);
        };
        ws.onmessage = function(e){
            console.log("onmessage");
            console.dir(e);
            $('#log').append('<p>'+e.data+'<p>');
            $('#log').get(0).scrollTop = $('#log').get(0).scrollHeight;
        };
        ws.onclose = function(e){
            console.log("onclose");
            console.dir(e);
        };
        ws.onerror = function(e){
            console.log("onerror");
            console.dir(e);
        };
        $(function(){
            $('#msgform').submit(function(){
                ws.send($('#msg').val()+"\n");
                $('#log').append('<p style="color:red;">My > '+$('#msg').val()+'<p>');
                $('#log').get(0).scrollTop = $('#log').get(0).scrollHeight;
                $('#msg').val('');
                return false;
            });
        });
    </script>
</head>
<body>
    <div id="log" style="height: 300px;overflow-y: scroll;border: 1px solid #CCC;">
    </div>
    <div>
        <form id="msgform">
            <input type="text" id="msg" size="60" />
        </form>
    </div>
</body>
</html>`
    io.WriteString(w, html)
}
func main() {
    fmt.Printf(`Welcome chatroom server!
author: dotcoo zhao
url: http://www.dotcoo.com/golang-websocket-chatroom
`)
    connid = 0
    conns = list.New()
    http.Handle("/chatroom", websocket.Handler(ChatroomServer));
    http.HandleFunc("/", Client);
    err := http.ListenAndServe(":6611", nil);
    if err != nil {
        panic("ListenAndServe: " + err.Error())
    }
}

简陋了些,但是基本功能都有了.

golang websocket 客户端

复制代码 代码如下:
package main
import (
    "fmt"
    "bufio"
    "io"
    "os"
    "code.google.com/p/go.net/websocket"
)
func Command(ws *websocket.Conn) {
    r := bufio.NewReader(os.Stdin)
    w := bufio.NewWriter(ws)
    for {
        data, err := r.ReadBytes('\n')
        if err != nil {
            panic(err)
        }
        _, err = w.Write(data)
        if err != nil {
            panic(err)
        }
        w.Flush()
        fmt.Printf("My\t> ")
    }
}
func main() {
    fmt.Printf(`Welcome chatroom!
    author: dotcoo zhao
    url: http://www.dotcoo.com/golang-websocket-chatroom
    `)
    origin := "http://127.0.0.1:6611/"     
    url := "ws://127.0.0.1:6611/chatroom"
    ws, err := websocket.Dial(url, "", origin)
    if err != nil {
        panic(err)
    }
    defer ws.Close()
    r := bufio.NewReader(ws)
    //w := bufio.NewWriter(os.Stdout)
    go Command(ws)
    for {
        data, err := r.ReadBytes('\n')
        if err == io.EOF {
            fmt.Printf("disconnected\n")
            os.Exit(0)
        }
        if err != nil {
            panic(err)
        }
        fmt.Printf("\r%sMy\t> ", data)
    }
}

希望本文所述对大家Go语言程序设计有所帮助。

您可能感兴趣的文章:

相关文章

  • golang中for循环遍历channel时需要注意的问题详解

    golang中for循环遍历channel时需要注意的问题详解

    这篇文章主要给大家介绍了关于golang中for循环遍历channel时需要注意的问题的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。
    2018-04-04
  • 多阶段构建优化Go 程序Docker镜像

    多阶段构建优化Go 程序Docker镜像

    这篇文章主要为大家介绍了多阶段构建优化Go 程序Docker镜像，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-08-08
  • Go设计模式之观察者模式图解

    Go设计模式之观察者模式图解

    观察者模式是一种行为设计模式， 允许你定义一种订阅机制， 可在对象事件发生时通知多个 “观察” 该对象的其他对象，下面这篇文章主要给大家介绍了关于图解Go观察者模式的相关资料,需要的朋友可以参考下
    2023-07-07
  • Go-家庭收支记账软件项目实现

    Go-家庭收支记账软件项目实现

    这篇文章主要介绍了Go-家庭收支记账软件项目实现，本文章内容详细，具有很好的参考价值，希望对大家有所帮助，需要的朋友可以参考下
    2023-01-01
  • Goland调节字体大小的设置(编辑区,terminal区,页面字体)

    Goland调节字体大小的设置(编辑区,terminal区,页面字体)

    这篇文章主要介绍了Goland调节字体大小的设置(编辑区,terminal区,页面字体)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • Go1.18新特性之泛型使用三步曲(小结)

    Go1.18新特性之泛型使用三步曲(小结)

    本文主要介绍了Go1.18新特性之泛型，是Go1.18的新特性，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-04-04
  • Golang学习之map的用法详解

    Golang学习之map的用法详解

    在Golang（又称Go语言）中，map是一种非常有用的数据结构，所以这篇文章小编就来带大家一起深入了解一下map的用法，感兴趣的小伙伴可以了解一下
    2023-06-06
  • Go操作etcd的实现示例

    Go操作etcd的实现示例

    etcd是近几年比较火热的一个开源的、分布式的键值对数据存储系统，提供共享配置、服务的注册和发现，本文主要介绍etcd的安装和使用，感兴趣的可以了解一下
    2021-09-09
  • golang中for range的取地址操作陷阱介绍

    golang中for range的取地址操作陷阱介绍

    这篇文章主要介绍了golang中for range的取地址操作陷阱，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-04-04
  • Go语言展现快速排序算法全过程的思路及代码示例

    Go语言展现快速排序算法全过程的思路及代码示例

    这篇文章主要介绍了Go语言展现快速排序算法全过程的思路及代码示例,文章最后作者还提到了对Quick Sort算法优化的一些想法,需要的朋友可以参考下
    2016-04-04

最新评论