package main

import (

    "github.com/go-martini/martini"

    "github.com/gorilla/websocket"

    "net/http"

    "time"

)

const (

    readBufferSize  =

    writeBufferSize =

)

type Client struct {

    conn *websocket.Conn

}

// 常住监听一个链接的读

func (c *Client) readPump() {

    for {

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

        // .......   该处处理接受用户发来的数据

    }

}

//监听一个链接的写

func (c *Client) writePump() {

    for {

        c.conn.SetWriteDeadline(time.Now().Add( * time.Second))

        // ....... 该处处理推送给用户的数据

    }

}

//  基于martini实现的websocket服务

func main() {

    m := martini.Classic()

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

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

        if err != nil {

            log.Println(err)

            return

        }

        client := &Client{conn: conn}

        go client.writePump() //建立一个goroutine

        client.readPump()

    })

    m.Run()

}

struct epoll_event ev, *events;

for(;;) {

 nfds = epoll_wait(kdpfd, events, maxevents, -1); //等待I/O事件

 for(n = 0; n < nfds; ++n) {

  if(events[n].data.fd == listener) { //如果是主socket的事件，则表示有新连接进入，需要进行新连接的处理。

    client = accept(listener, (struct sockaddr *) &local,  &addrlen);

    if(client < 0){

      perror("accept error");

      continue;

    }

    setnonblocking(client); // 将新连接置于非阻塞模式

    ev.events = EPOLLIN | EPOLLET;

                                   //注意这里的参数EPOLLIN | EPOLLET并没有设置对写socket的监听，

                                   //如果有写操作的话，这个时候epoll是不会返回事件的，

                                   //如果要对写操作也监听的话，应该是EPOLLIN | EPOLLOUT | EPOLLET。

    ev.data.fd = client; // 并且将新连接也加入EPOLL的监听队列

    if (epoll_ctl(kdpfd, EPOLL_CTL_ADD, client, &ev) < 0) {  // 设置好event之后，将这个新的event通过epoll_ctl

                                                             //加入到epoll的监听队列里，这里用EPOLL_CTL_ADD

                                                             //来加一个新的 epoll事件。可以通过EPOLL_CTL_DEL来减少

                                                             //一个epoll事件，通过EPOLL_CTL_MOD来改变一个事件的监听方式。

      fprintf(stderr, "epoll set insertion error: fd=%d"0, client);

      return -1;

    }

  }  else // 如果不是主socket的事件的话，则代表这是一个用户的socket的事件，

            // 则用来处理这个用户的socket的事情是，比如说read(fd,xxx)之类，或者一些其他的处理。

    do_use_fd(events[n].data.fd);

 }

}

func M() {

    for {

        sched.lock.Lock()    //互斥地从就绪G队列中取一个g出来运行

        if sched.allg > 0 {

            g := sched.allg[0]

            sched.allg = sched.allg[1:]

            sched.lock.Unlock()

            g.Run()        //运行它

        } else {

            sched.lock.Unlock()

        }

    }

}

for i:=0; i<GOMAXPROCS; i++ {

    go M()

}