forked from all_user/stat_center
136 lines
3.7 KiB
Go
136 lines
3.7 KiB
Go
package m3u8
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"stat/common/enum/redisconst"
|
|
"stat/common/errors"
|
|
"stat/common/http"
|
|
"stat/common/log"
|
|
"stat/common/redis"
|
|
"stat/common/utils"
|
|
"strings"
|
|
|
|
"github.com/grafov/m3u8"
|
|
)
|
|
|
|
const (
|
|
keyUri = "/api/stat/media/enkey"
|
|
AuthKey = "fT5xSg4hltHpzVy6aV9rVECDJ1J1pN"
|
|
)
|
|
|
|
var headerMediaPlayList *m3u8.MediaPlaylist
|
|
|
|
// TransFormWithSafely 被检查权限的播放列表
|
|
// 主要检查用户是否有观看该视频的权限
|
|
// apiType(1:app,2:h5app,3:web)
|
|
func TransFormWithSafely(req *Req, cdnDomain, origDomain string, needHeader bool, apiType int) (*m3u8.MediaPlaylist, *bytes.Buffer, error) {
|
|
if req.Name == "" {
|
|
return nil, nil, errors.MakeWithMsg(errors.M3u8FileName, "file name is empty")
|
|
}
|
|
if cdnDomain == "" || origDomain == "" {
|
|
return nil, nil, errors.MakeWithMsg(errors.DomainErr, "cdnDomain or origDomain is empty")
|
|
}
|
|
|
|
return transform(req.Name, cdnDomain, origDomain, needHeader, apiType)
|
|
}
|
|
func AppendsMediaS3Version(name string) string {
|
|
// 默认使用V1
|
|
if name != "" && !strings.HasPrefix(name, "v2/") && !strings.HasPrefix(name, "v1/") && !strings.HasPrefix(name, "v3/") &&
|
|
!strings.HasPrefix(name, "/v2") && !strings.HasPrefix(name, "/v1") && !strings.HasPrefix(name, "/v3") {
|
|
return "/v1"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// transform m3u8文件转换，将不正确的m3u8文件，替换为正确的m3u8文件
|
|
func transform(name string, cdnDomain, origDomain string, needHeader bool, apiType int) (media *m3u8.MediaPlaylist, buff *bytes.Buffer, err error) {
|
|
m3u8reader, err := readFromRedis(name, origDomain)
|
|
if err != nil {
|
|
return
|
|
}
|
|
p, t, err := m3u8.DecodeFrom(m3u8reader, false)
|
|
if err != nil {
|
|
err = errors.MakeWithErr(errors.M3u8Read, err)
|
|
return
|
|
}
|
|
tmptKeyurl := keyUri
|
|
|
|
fmt.Println("<<<<<<<<<<<<<<<<<<<start get media list>>>>>>>>>>>>>>>")
|
|
switch t {
|
|
case m3u8.MEDIA:
|
|
pl := p.(*m3u8.MediaPlaylist)
|
|
dirPath := utils.PathForInfo(name).Dir
|
|
media, err = newMediaPlayList(pl, cdnDomain, origDomain, dirPath, tmptKeyurl, needHeader)
|
|
if err != nil {
|
|
return
|
|
}
|
|
buff = media.Encode()
|
|
case m3u8.MASTER:
|
|
mpl := p.(*m3u8.MasterPlaylist)
|
|
buff = mpl.Encode()
|
|
default:
|
|
err = errors.MakeWithMsg(errors.M3u8FileName, fmt.Sprintf("invalid m3u8 type %d", t))
|
|
}
|
|
return
|
|
}
|
|
|
|
//func readFromCDN(name, domain string) (reader io.Reader, err error) {
|
|
// requestUrl := utils.JoinUrlPath(domain, name)
|
|
// resp, err := http.GetLargeFile(requestUrl, nil)
|
|
// if err != nil {
|
|
// return
|
|
// }
|
|
// if resp.StatusCode != 200 {
|
|
// msg := fmt.Sprintf("read m3u8 fail, url: %s, status code %d", requestUrl, resp.StatusCode)
|
|
// err = errors.MakeWithMsg(errors.M3u8Read, msg)
|
|
// return
|
|
// }
|
|
// reader = bytes.NewReader(resp.Data)
|
|
// return
|
|
//}
|
|
|
|
// 从redis缓存中读取
|
|
func readFromRedis(name, domain string) (reader io.Reader, err error) {
|
|
m3u8Key := redisconst.MediaM3U8Key(name)
|
|
data, err := redis.StatRds.GetBytes(m3u8Key)
|
|
if err == nil && len(data) > 0 {
|
|
reader = bytes.NewReader(data)
|
|
return
|
|
}
|
|
requestUrl := utils.JoinUrlPath(domain, name)
|
|
resp, err := http.GetLargeFile(requestUrl, nil)
|
|
if err != nil {
|
|
//fmt.Println("requestUrl111", requestUrl)
|
|
return
|
|
}
|
|
|
|
if resp.StatusCode != 200 {
|
|
msg := fmt.Sprintf("read m3u8 fail, url: %s, status code %d", requestUrl, resp.StatusCode)
|
|
//fmt.Println(msg)
|
|
err = errors.MakeWithMsg(errors.M3u8Read, msg)
|
|
return
|
|
}
|
|
if err := redis.StatRds.Set(m3u8Key, resp.Data,
|
|
redisconst.MediaM3U8CacheExpire); err != nil {
|
|
log.Info("m3u8:readFromRedis redis set err", log.E(err))
|
|
}
|
|
reader = bytes.NewReader(resp.Data)
|
|
return
|
|
}
|
|
|
|
func GetHeaderMediaSegUrls() []string {
|
|
urls := make([]string, 0)
|
|
if headerMediaPlayList == nil {
|
|
return urls
|
|
}
|
|
for _, segment := range headerMediaPlayList.Segments {
|
|
if segment == nil {
|
|
continue
|
|
}
|
|
urls = append(urls, segment.URI)
|
|
}
|
|
return urls
|
|
}
|