lal/httpflv/manager.go

package httpflv

import (
	"log"
	"sync"
	"time"
)

type Manager struct {
	Config

	server   *Server
	groups   map[string]*Group // TODO chef: with appName
	mutex    sync.Mutex
	exitChan chan bool
}

func NewManager(config Config) *Manager {
	return &Manager{
		Config:   config,
		server:   NewServer(config.SubListenAddr),
		groups:   make(map[string]*Group),
		exitChan: make(chan bool),
	}
}

func (manager *Manager) SubSessionHandler(session *SubSession) {
	group := manager.getOrCreateGroup(session.AppName, session.StreamName)
	group.AddSubSession(session)
	if manager.PullAddr != "" {
		group.PullIfNeeded(manager.PullAddr)
	}
}

func (manager *Manager) RunLoop() {
	go func() {
		for {
			if subSession, ok := manager.server.Accept(); ok {
				manager.SubSessionHandler(subSession)
			} else {
				log.Println("accept sub session from httpflv server failed.")
				return
			}
		}
	}()

	go func() {
		if err := manager.server.RunLoop(); err != nil {
			log.Println(err)
		}
	}()

	t := time.NewTicker(1 * time.Second)
	defer t.Stop()
	for {
		select {
		case <-manager.exitChan:
			return
		case <-t.C:
			manager.check()
		}
	}
}

func (manager *Manager) Dispose() {
	log.Println("Dispose manager.")
	manager.server.Dispose()
	manager.exitChan <- true
	manager.mutex.Lock()
	defer manager.mutex.Unlock()
	for _, group := range manager.groups {
		group.Dispose()
	}
	manager.groups = nil
}

func (manager *Manager) check() {
	manager.mutex.Lock()
	defer manager.mutex.Unlock()
	for k, group := range manager.groups {
		if group.IsTotalEmpty() {
			log.Println("erase empty group.", k)
			group.Dispose()
			delete(manager.groups, k)
		}
	}
}

func (manager *Manager) getOrCreateGroup(appName string, streamName string) *Group {
	manager.mutex.Lock()
	defer manager.mutex.Unlock()
	group, exist := manager.groups[streamName]
	if !exist {
		log.Println("create group. ", streamName)
		group = NewGroup(appName, streamName, manager.Config)
		manager.groups[streamName] = group
	}
	go group.RunLoop()
	return group
}