lal/pkg/hls/server_sub_session.go

// Copyright 2022, 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 hls

import (
	"github.com/q191201771/lal/pkg/base"
	"github.com/q191201771/naza/pkg/connection"
	"github.com/q191201771/naza/pkg/nazamd5"
	"net/http"
	"strings"
	"time"
)

type SubSession struct {
	LastRequestTime time.Time
	urlCtx          base.UrlContext
	hlsUrlPattern   string
	appName         string
	timeout         time.Duration
	sessionIdHash   string // Because session.UniqueKey() too easy to guess so that we need to hash it with a key to prevent client guess session id

	stat     base.StatSession
	prevStat connection.Stat
	currStat connection.StatAtomic
}

func (s *SubSession) UniqueKey() string {
	return s.stat.SessionId
}

func NewSubSession(req *http.Request, urlCtx base.UrlContext, hlsUrlPattern, sessionHashKey string, timeout time.Duration) *SubSession {
	if strings.HasPrefix(hlsUrlPattern, "/") {
		hlsUrlPattern = hlsUrlPattern[1:]
	}
	session := &SubSession{
		LastRequestTime: time.Now(),
		urlCtx:          urlCtx,
		hlsUrlPattern:   hlsUrlPattern,
		timeout:         timeout,
	}
	// stat
	session.stat.SessionId = base.GenUkHlsSubSession()
	session.stat.Protocol = base.SessionProtocolHlsStr
	session.stat.BaseType = base.SessionBaseTypeSubStr
	session.stat.RemoteAddr = req.RemoteAddr
	session.stat.StartTime = time.Now().String()

	// TODO(chef): [refactor] 也许后续可以弄短点，比如前8位或16位 202211
	session.sessionIdHash = nazamd5.Md5([]byte(session.stat.SessionId + sessionHashKey))
	return session
}

func (s *SubSession) Url() string {
	return s.urlCtx.Url
}

func (s *SubSession) AppName() string {
	if s.appName == "" {
		s.appName = GetAppNameFromUrlCtx(s.urlCtx, s.hlsUrlPattern)
	}
	return s.appName
}

func (s *SubSession) StreamName() string {
	return GetStreamNameFromUrlCtx(s.urlCtx)
}

func (s *SubSession) RawQuery() string {
	return s.urlCtx.RawQuery
}

func (s *SubSession) UpdateStat(intervalSec uint32) {
	wroteBytesSum := s.currStat.WroteBytesSum.Load()
	wDiff := wroteBytesSum - s.prevStat.WroteBytesSum
	s.stat.WriteBitrateKbits = int(wDiff * 8 / 1024 / uint64(intervalSec))
	s.stat.BitrateKbits = s.stat.WriteBitrateKbits
	s.prevStat.WroteBytesSum = wroteBytesSum
	return
}

func (s *SubSession) AddWroteBytesSum(wbs uint64) {
	s.currStat.WroteBytesSum.Add(wbs)
}

func (s *SubSession) GetStat() base.StatSession {
	s.stat.WroteBytesSum = s.currStat.WroteBytesSum.Load()
	return s.stat
}

func (s *SubSession) IsAlive() (readAlive, writeAlive bool) {
	if !s.IsExpired() {
		return true, true
	}
	return false, false
}

func (s *SubSession) IsExpired() bool {
	return s.LastRequestTime.Add(s.timeout).Before(time.Now())
}

func (s *SubSession) KeepAlive() {
	s.LastRequestTime = time.Now()
}

func GetAppNameFromUrlCtx(urlCtx base.UrlContext, hlsUrlPattern string) string {
	if hlsUrlPattern == "" {
		return urlCtx.PathWithoutLastItem
	} else {
		urlParts := strings.Split(urlCtx.PathWithoutLastItem, hlsUrlPattern)
		if len(urlParts) > 1 {
			return urlParts[1]
		}
	}
	return ""
}

func GetStreamNameFromUrlCtx(urlCtx base.UrlContext) string {
	return urlCtx.GetFilenameWithoutType()
}