|
|
// Copyright 2019, Chef. All rights reserved.
|
|
|
// https://github.com/q191201771/lal
|
|
|
//
|
|
|
// Use of this source code is governed by a MIT-style license
|
|
|
// that can be found in the License file.
|
|
|
//
|
|
|
// Author: Chef (191201771@qq.com)
|
|
|
|
|
|
package httpflv
|
|
|
|
|
|
import (
|
|
|
"net"
|
|
|
"sync"
|
|
|
|
|
|
log "github.com/q191201771/naza/pkg/nazalog"
|
|
|
)
|
|
|
|
|
|
type ServerObserver interface {
|
|
|
// 通知上层有新的拉流者
|
|
|
// 返回值： true则允许拉流，false则关闭连接
|
|
|
NewHTTPFLVSubSessionCB(session *SubSession) bool
|
|
|
|
|
|
DelHTTPFLVSubSessionCB(session *SubSession)
|
|
|
}
|
|
|
|
|
|
type Server struct {
|
|
|
obs ServerObserver
|
|
|
addr string
|
|
|
|
|
|
m sync.Mutex
|
|
|
ln net.Listener
|
|
|
}
|
|
|
|
|
|
func NewServer(obs ServerObserver, addr string) *Server {
|
|
|
return &Server{
|
|
|
obs: obs,
|
|
|
addr: addr,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (server *Server) RunLoop() error {
|
|
|
var err error
|
|
|
|
|
|
server.m.Lock()
|
|
|
server.ln, err = net.Listen("tcp", server.addr)
|
|
|
server.m.Unlock()
|
|
|
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
|
|
|
log.Infof("start httpflv listen. addr=%s", server.addr)
|
|
|
for {
|
|
|
conn, err := server.ln.Accept()
|
|
|
if err != nil {
|
|
|
return err
|
|
|
}
|
|
|
go server.handleConnect(conn)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (server *Server) Dispose() {
|
|
|
server.m.Lock()
|
|
|
defer server.m.Unlock()
|
|
|
if server.ln == nil {
|
|
|
return
|
|
|
}
|
|
|
if err := server.ln.Close(); err != nil {
|
|
|
log.Error(err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (server *Server) handleConnect(conn net.Conn) {
|
|
|
log.Infof("accept a httpflv connection. remoteAddr=%v", conn.RemoteAddr())
|
|
|
session := NewSubSession(conn)
|
|
|
if err := session.ReadRequest(); err != nil {
|
|
|
log.Errorf("read httpflv SubSession request error. [%s]", session.UniqueKey)
|
|
|
return
|
|
|
}
|
|
|
log.Infof("-----> http request. [%s] uri=%s", session.UniqueKey, session.URI)
|
|
|
|
|
|
if !server.obs.NewHTTPFLVSubSessionCB(session) {
|
|
|
session.Dispose()
|
|
|
}
|
|
|
|
|
|
err := session.RunLoop()
|
|
|
log.Debugf("httpflv sub session loop done. err=%v", err)
|
|
|
server.obs.DelHTTPFLVSubSessionCB(session)
|
|
|
}
|