如果只是想了解chat的实现方式，在gorilla和revel框架里面都有完整的chat实例可以提供参考。本篇讲解的是，如何基于martini实现websocket的聊天。

配置步骤:

1 已经安装了go命令，配置了GOPATH

2 安装gorilla/websocket

go get github.com/gorilla/websocket

3 安装martini

go get github.com/go-martini/martini

项目文件列表

$GOPATH/demo/home.html

$GOPATH/demo/martini_ws.go

home.html 代码

<!DOCTYPE html>

<html lang="en">

<head>

<title>Chat Example</title>

<script type="text/javascript">

window.onload = function () {

    var conn;

    var msg = document.getElementById("msg");

    var log = document.getElementById("log");

    function appendLog(item) {

        var doScroll = log.scrollTop === log.scrollHeight - log.clientHeight;

        log.appendChild(item);

        if (doScroll) {

            log.scrollTop = log.scrollHeight - log.clientHeight;

        }

    }

    document.getElementById("form").onsubmit = function () {

        if (!conn) {

            return false;

        }

        if (!msg.value) {

            return false;

        }

        conn.send(msg.value);

        msg.value = "";

        return false;

    };

    if (window["WebSocket"]) {

        conn = new WebSocket("ws://{{$}}/ws");

        conn.onclose = function (evt) {

            var item = document.createElement("div");

            item.innerHTML = "<b>Connection closed.</b>";

            appendLog(item);

        };

        conn.onmessage = function (evt) {

            var messages = evt.data.split('\n');

            for (var i = ; i < messages.length; i++) {

                var item = document.createElement("div");

                item.innerText = messages[i];

                appendLog(item);

            }

        };

    } else {

        var item = document.createElement("div");

        item.innerHTML = "<b>Your browser does not support WebSockets.</b>";

        appendLog(item);

    }

};

</script>

<style type="text/css">

html {

    overflow: hidden;

}

body {

    overflow: hidden;

    padding: ;

    margin: ;

    width: %;

    height: %;

    background: gray;

}

#log {

    background: white;

    margin: ;

    padding: .5em .5em .5em .5em;

    position: absolute;

    top: .5em;

    left: .5em;

    right: .5em;

    bottom: 3em;

    overflow: auto;

}

#form {

    padding:  .5em  .5em;

    margin: ;

    position: absolute;

    bottom: 1em;

    left: 0px;

    width: %;

    overflow: hidden;

}

</style>

</head>

<body>

<div id="log"></div>

<form id="form">

    <input type="submit" value="Send" />

    <input type="text" id="msg" size=""/>

</form>

</body>

</html>

martini_ws.go代码

package main

import (

    "fmt"

    "github.com/go-martini/martini"

    "github.com/gorilla/websocket"

    "log"

    "net/http"

    "text/template"

)

const (

    readBufferSize  =

    writeBufferSize =

)

type Client struct {

    conn     *websocket.Conn

    messages chan []byte

}

var clients map[*Client]bool // 存储所有的链接

var broadcast chan []byte    // 广播聊天的chan

var addClients chan *Client  // 新链接进来的chan

func (c *Client) readPump() {

    for {

        _, message, err := c.conn.ReadMessage()

        if err != nil {

            if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {

                log.Printf("error: %v", err)

            }

            break

        }

        fmt.Printf("receive message is :%s\n", message)

        broadcast <- message

    }

}

func (c *Client) writePump() {

    for {

        select {

        case message := <-c.messages:

            fmt.Printf("send message is :%s\n", message)

            c.conn.WriteMessage(, message)

        }

    }

}

func manager() {

    clients = make(map[*Client]bool)

    broadcast = make(chan []byte, )

    addClients = make(chan *Client)

    for {

        select {

        case message := <-broadcast:

            for client := range clients {

                select {

                case client.messages <- message:

                default:

                    close(client.messages)

                    delete(clients, client)

                }

            }

        case itemClient := <-addClients:

            clients[itemClient] = true

        }

    }

}

func main() {

    var homeTemplate = template.Must(template.ParseFiles("home.html"))

    m := martini.Classic()

    m.Get("/", func(res http.ResponseWriter, req *http.Request) {

        res.Header().Set("Content-Type", "text/html; charset=utf-8")

        homeTemplate.Execute(res, req.Host)

    })

    m.Get("/ws", func(res http.ResponseWriter, req *http.Request) { // res and req are injected by Martini

        conn, err := websocket.Upgrade(res, req, nil, readBufferSize, writeBufferSize)

        if err != nil {

            log.Println(err)

            return

        }

        client := &Client{conn: conn, messages: make(chan []byte, )}

        addClients <- client

        go client.writePump()

        client.readPump()

    })

    go manager()

    m.RunOnAddr(":3010")

    //m.Run()

}

按照上面的步骤建好文件，就可以运行了。m.RunOnAddr 可以改成你需要的端口

执行命令:

go run martini_ws.go

在浏览器输入 http://localhost:3010　就能访问你本机的聊天服务了

参考地址

https://github.com/patcito/martini-gorilla-websocket-chat-example

　　

go语言从零学起(四) -- 基于martini和gorilla实现的websocket聊天实例的更多相关文章

  1. go语言从零学起(三) -- chat实现的思考

    要通过go实现一个应用场景: 1 建立一个websocket服务 2 维护在线用户的链接 3 推送消息和接受用户的操作 列出需求,很显然的想到了chat模型.于是研究了revel框架提供的sample ...

  2. go语言从零学起(一) -- 文档教程篇

    先记录一下自己学go语言的出发点 作为一个phper,精通一门底层语言一直是努力的目标. 相对于c,c++,go语言不需要过多的关注指针,内存释放,一两行代码就能跑起一个server服务,简直不要太简 ...

  3. go语言从零学起(二)--list循环删除元素(转载)

    本篇系转载 在使用go的container/list的package时,你可能会无意间踩一个小坑,那就是list的循环删除元素. list删除元素,直观写下来的代码如下: package main i ...

  4. 从零学脚手架(四)---babel

    如果此篇对您有所帮助,在此求一个star.项目地址: OrcasTeam/my-cli 接下来介绍一个打包编译过程中一个极为重要的工具--babel. ES6的枷锁 细心的朋友可以知道,在之前打包编译 ...

  5. 带你从零学ReactNative开发跨平台App开发（四）

    ReactNative跨平台开发系列教程: 带你从零学ReactNative开发跨平台App开发(一) 带你从零学ReactNative开发跨平台App开发(二) 带你从零学ReactNative开发 ...

  6. 带你从零学ReactNative开发跨平台App开发-[react native 仿boss直聘]（十三）

    ReactNative跨平台开发系列教程: 带你从零学ReactNative开发跨平台App开发(一) 带你从零学ReactNative开发跨平台App开发(二) 带你从零学ReactNative开发 ...

  7. 带你从零学ReactNative开发跨平台App开发[expo 打包发布]（八）

    ReactNative跨平台开发系列教程: 带你从零学ReactNative开发跨平台App开发(一) 带你从零学ReactNative开发跨平台App开发(二) 带你从零学ReactNative开发 ...

  8. 从头开始学JavaScript (四)——操作符

    原文:从头开始学JavaScript (四)--操作符 一.一元操作符 1.自增自减操作符:分为前置型和后置型: 前置型:++a;--a; 后置型:a++;a--; 例: <script typ ...

  9. C语言老司机学Python （五）

    今天看的是标准库概览. 操作系统接口: 用os模块实现. 针对文件和目录管理,还有个shutil模块可以用. 例句: import os os.getcwd() # 返回当前的工作目录 os.chdi ...

随机推荐

  1. 毕业设计 之 三 mooodle及bigbluebutton使用笔记（未完成）

    毕业设计 之 三 mooodle及bigbluebutton使用笔记 作者:20135216 平台:windows10 备注:N把辛酸泪 附:在准备过程中其他的一些零碎小问题 一.关于moodle 1 ...

  2. Linux课题实践四——ELF文件格式分析

    2.4   ELF文件格式分析 20135318 刘浩晨 ELF全称Executable and Linkable Format,可执行连接格式,ELF格式的文件用于存储Linux程序.ELF文件(目 ...

  3. [转帖] 打开加密SQLite文件的方法

    Copy From http://blog.csdn.net/sean4m/article/details/50211565 mark 下 正好工作用到了这个东西. 版本:SQLiteExpertPr ...

  4. centos7 修改网络配置问题

    ip adr 查看ip 查看报错详细信息 journalctl -xe 查看网络故障命令 cat /var/log/messages |grep network 报错现象 Error: Connect ...

  5. 自定义Label控件

    最近开发过程中有一个需求就是修改label控件的模板,使其能够在鼠标移近的时候变成TextBox,从而方便输入,然后进行相应的修改,最终达到动态修改Label的目的,这里贴出相应的代码,并做简要的分析 ...

  6. Python中print/format字符串格式化实例

    Python 字符串格式化使用 "字符 %格式1 %格式2 字符"%(变量1,变量2),%格式表示接受变量的类型.简单的使用例子如下 # 例:字符串格式化Name = '17jo' ...

  7. html 塊 div span

    塊級元素:元素結束后,另起一行:如p,table,h1,ul,ol 內聯元素:元素結束后,不會另起一行:如a,td.image.b <div>是塊級元素,div本身沒有含義,div是htm ...

  8. 【题解】ID分配

    题目描述 您正在处理要为每个客户端分配唯一ID的特定系统的后端.但是,系统是分布式的,并且有许多组件,每个组件都必须能够为客户端分配ID.换句话说,您希望每个组件都使用尽可能少的组件之间的通信来分配I ...

  9. 【Linux】Centos7 解压zip文件

    如果输入unzip无反应那么请安装相应软件包 yum install -y unzip 语法 unzip [参数] [文件] 参数 -c:将解压缩的结果显示到屏幕上,并对字符做适当的转换: -f:更新 ...

  10. ubuntu eclipse 无法打开

    报错: The catalog could not be loaded... 这个问题,我上网查了很久.. 基本上网上可用的办法我都试过了.全是失败的 firefox+autoproxy,shadow ...