如果只是想了解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. Week 3 有求必应

    [引] 必应词典已经伴我很久了,但那并不意味着我天天都用它查来查去,它总是静静地蹲在我E盘的一角. 从前的它特别任性,总喜欢开机自己冒出头来看我.后来我嫌它每天都播报新闻,于是就把它关进了冷宫.直到不 ...

  2. Opentsdb 启动显示配置文件不存在

    今天 重新启动opentsdb  出现本地配置文件不存在   这不知道  我查了一下官网 了解到 You can use the --config command line argument to s ...

  3. Android中Json数据读取与创建的方法

    转自:http://www.jb51.net/article/70875.htm 首先介绍下JSON的定义,JSON是JavaScript Object Notation的缩写. 一种轻量级的数据交换 ...

  4. how are you

    #include<stdio.h>   int main(){   char sentence[100];   int len=0,j,wordlen=0;   gets(sentence ...

  5. 在XShell中使用sz和rz命令下载和上传文件

    借助XShell,使用linux命令sz可以很方便的将服务器上的文件下载到本地,使用rz命令则是把本地文件上传到服务器 工具/原料   XShell CentOS 6.5 使用sz下载文件   1 输 ...

  6. octave基本指令2

    octave基本指令2 数据移动 >> pwd %显示出当前路径 ans C:\Octave\3.2.4_gcc-4 >> cd 'G:\machine learning' % ...

  7. jmeter 使用csv文件 注意项

    1.首先在jmeter 中导入csv文件时我们程序并不知道csv文件中有多少行 : >1.获取的时候 使用 循环控制器来获取csv文件中的所有数据 : 通过 ${__jexl3("${ ...

  8. delphi创建动态菜单

    1.动态生成菜单项 varFirstItem: TMenuItem;SecondItem: TMenuItem; begin FirstItem := TMenuItem.Create(Self); ...

  9. BZOJ3075[USACO 2013 Mar Gold 3.Necklace]——AC自动机+DP

    题目描述 给你一个长度为n的字符串A,再给你一个长度为m的字符串B,求至少在A中删去多少个字符才能使得B不是A的子串.注:该题只读入A和B,不读入长度,先读入A,再读入B.数据保证A和B中只含小写字母 ...

  10. Code First 重复外键（简单方法）

    之前有说过  Code First 重复外键   的一种解决方案.   http://blog.csdn.net/hanjun0612/article/details/50478134 虽然可以解决问 ...