mirror of https://github.com/q191201771/lal.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package httpflv
|
|
|
|
import (
|
|
"log"
|
|
"net"
|
|
)
|
|
|
|
type Server struct {
|
|
addr string
|
|
sessionChan chan *SubSession
|
|
|
|
ln net.Listener
|
|
}
|
|
|
|
func NewServer(addr string) *Server {
|
|
return &Server{
|
|
addr: addr,
|
|
sessionChan: make(chan *SubSession, 8),
|
|
}
|
|
}
|
|
|
|
func (server *Server) RunLoop() error {
|
|
var err error
|
|
server.ln, err = net.Listen("tcp", server.addr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for {
|
|
conn, err := server.ln.Accept()
|
|
if err != nil {
|
|
log.Println(err)
|
|
return err
|
|
}
|
|
go server.handleSubSessionConnect(conn)
|
|
}
|
|
}
|
|
|
|
func (server *Server) Accept() (session *SubSession, ok bool) {
|
|
session, ok = <-server.sessionChan
|
|
return
|
|
}
|
|
|
|
func (server *Server) Dispose() {
|
|
if err := server.ln.Close(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
close(server.sessionChan)
|
|
}
|
|
|
|
func (server *Server) handleSubSessionConnect(conn net.Conn) {
|
|
log.Println("accept a http flv sub connection. ", conn.RemoteAddr())
|
|
subSession := NewSubSession(conn)
|
|
err := subSession.ReadRequest()
|
|
if err != nil {
|
|
return
|
|
}
|
|
log.Println("read sub session request. ", subSession.StreamName)
|
|
|
|
server.sessionChan <- subSession
|
|
}
|