go網路編程樣本,伺服器端

來源:互聯網
上載者:User
這是一個建立於 的文章,其中的資訊可能已經有所發展或是發生改變。


package rtmpimport ("bufio"log "github.com/cihub/seelog""net""runtime""sync""time")var shandler ServerHandler = new(DefaultServerHandler)func ListenAndServe(addr string) error {srv := &Server{Addr:         addr,ReadTimeout:  time.Duration(time.Second * 30),WriteTimeout: time.Duration(time.Second * 30),Lock:         new(sync.Mutex)}return srv.ListenAndServe()}type Server struct {Addr         string        //監聽地址ReadTimeout  time.Duration //讀逾時WriteTimeout time.Duration //寫逾時Lock         *sync.Mutex}var gstreamid = uint32(64)func gen_next_stream_id(chunkid uint32) uint32 {gstreamid += 1return gstreamid}func newconn(conn net.Conn, srv *Server) (c *RtmpNetConnection) {c = new(RtmpNetConnection)c.remoteAddr = conn.RemoteAddr().String()c.server = srvc.readChunkSize = RTMP_DEFAULT_CHUNK_SIZEc.writeChunkSize = RTMP_DEFAULT_CHUNK_SIZEc.createTime = time.Now()c.bandwidth = 512 << 10c.conn = connc.br = bufio.NewReader(conn)c.bw = bufio.NewWriter(conn)c.buf = bufio.NewReadWriter(c.br, c.bw)c.lock = new(sync.Mutex)//c.csid_chunk = make(map[uint32]uint32)c.lastReadHeaders = make(map[uint32]*RtmpHeader)c.lastWriteHeaders = make(map[uint32]*RtmpHeader)c.incompletePackets = make(map[uint32]Payload)c.nextStreamId = gen_next_stream_idc.objectEncoding = 0return}func (p *Server) ListenAndServe() error {addr := p.Addrif addr == "" {addr = ":1935"}l, err := net.Listen("tcp", addr)if err != nil {return err}for i := 0; i < runtime.NumCPU(); i++ {go p.loop(l)}return nil}func (srv *Server) loop(l net.Listener) error {defer l.Close()var tempDelay time.Duration // how long to sleep on accept failurefor {grw, e := l.Accept()if e != nil {if ne, ok := e.(net.Error); ok && ne.Temporary() {if tempDelay == 0 {tempDelay = 5 * time.Millisecond} else {tempDelay *= 2}if max := 1 * time.Second; tempDelay > max {tempDelay = max}log.Errorf("rtmp: Accept error: %v; retrying in %v", e, tempDelay)time.Sleep(tempDelay)continue}return e}tempDelay = 0go serve(srv, grw)}}func serve(srv *Server, con net.Conn) {conn := newconn(con, srv)if !handshake1(conn.buf) {conn.Close()return}msg, err := readMessage(conn)if err != nil {log.Error("NetConnecton read error", err)conn.Close()return}cmd, ok := msg.(*ConnectMessage)if !ok || cmd.Command != "connect" {log.Error("NetConnecton Received Invalid ConnectMessage", msg)conn.Close()return}conn.app = getString(cmd.Object, "app")conn.objectEncoding = int(getNumber(cmd.Object, "objectEncoding"))log.Debug(cmd)err = sendAckWinsize(conn, 512<<10)if err != nil {log.Error("NetConnecton sendAckWinsize error", err)conn.Close()return}err = sendPeerBandwidth(conn, 512<<10)if err != nil {log.Error("NetConnecton sendPeerBandwidth error", err)conn.Close()return}err = sendStreamBegin(conn)if err != nil {log.Error("NetConnecton sendStreamBegin error", err)conn.Close()return}err = sendConnectSuccess(conn)if err != nil {log.Error("NetConnecton sendConnectSuccess error", err)conn.Close()return}conn.connected = truenewNetStream(conn, shandler, nil).serve()}



轉載地址:http://www.oschina.net/code/snippet_874_22821


聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.