go-m3u8 库使用指南

go-m3u8 是一个用于解析和生成 Apple HLS (HTTP Live Streaming) m3u8 播放列表的 Go 语言库。下面我将详细介绍如何使用这个库。

安装

go get github.com/quangngotan95/go-m3u8/m3u8

基本概念

m3u8 是 HLS 协议使用的播放列表格式，主要有两种类型：

• 主播放列表 (Master Playlist)
• 媒体播放列表 (Media Playlist)

解析 m3u8 文件

解析主播放列表

package main

import (
	"fmt"
	"os"

	"github.com/quangngotan95/go-m3u8/m3u8"
)

func main() {
	// 读取m3u8文件内容
	data, err := os.ReadFile("master.m3u8")
	if err != nil {
		panic(err)
	}

	// 解析播放列表
	playlist, err := m3u8.Parse(string(data))
	if err != nil {
		panic(err)
	}

	// 判断是否是主播放列表
	if playlist.IsMaster() {
		master := playlist.(*m3u8.MasterPlaylist)
		// 遍历所有变体流
		for _, variant := range master.Variants {
			fmt.Printf("Bandwidth: %d, Resolution: %s, URI: %s\n",
				variant.Bandwidth,
				variant.Resolution,
				variant.URI)
		}
	}
}

解析媒体播放列表

func parseMediaPlaylist() {
	data, err := os.ReadFile("media.m3u8")
	if err != nil {
		panic(err)
	}

	playlist, err := m3u8.Parse(string(data))
	if err != nil {
		panic(err)
	}

	if !playlist.IsMaster() {
		media := playlist.(*m3u8.MediaPlaylist)
		fmt.Printf("Target Duration: %f\n", media.TargetDuration)
		fmt.Printf("Sequence: %d\n", media.Sequence)
		fmt.Printf("Segments Count: %d\n", len(media.Segments))
		
		for _, seg := range media.Segments {
			fmt.Printf("Duration: %f, URI: %s\n", seg.Duration, seg.URI)
		}
	}
}

生成 m3u8 文件

创建主播放列表

func createMasterPlaylist() {
	playlist := m3u8.NewMasterPlaylist()
	
	// 添加变体流
	playlist.Append(
		"video-800k.m3u8",
		800000,
		"640x360",
		"avc1.42001e,mp4a.40.2")
	
	playlist.Append(
		"video-1200k.m3u8",
		1200000,
		"854x480",
		"avc1.42001f,mp4a.40.2")
	
	// 设置独立属性
	playlist.IndependentSegments = true
	
	// 输出播放列表
	fmt.Println(playlist.String())
}

创建媒体播放列表

func createMediaPlaylist() {
	// 创建媒体播放列表，设置容量为10个片段
	playlist, err := m3u8.NewMediaPlaylist(10, 10)
	if err != nil {
		panic(err)
	}
	
	// 设置播放列表属性
	playlist.TargetDuration = 6
	playlist.Sequence = 1
	playlist.MediaType = m3u8.VOD // 点播类型
	
	// 添加媒体片段
	playlist.Append("segment1.ts", 5.0, "")
	playlist.Append("segment2.ts", 5.0, "")
	playlist.Append("segment3.ts", 5.0, "")
	
	// 关闭播放列表(对于VOD很重要)
	playlist.Close()
	
	// 输出播放列表
	fmt.Println(playlist.String())
}

高级功能

添加自定义标签

func addCustomTags() {
	playlist := m3u8.NewMasterPlaylist()
	
	// 添加EXT-X-VERSION标签
	playlist.SetVersion(3)
	
	// 添加自定义注释
	playlist.SetComments([]string{"Generated by go-m3u8"})
	
	// 添加其他自定义标签
	playlist.SetCustomTag(m3u8.CustomTag{
		Name:  "EXT-X-MY-TAG",
		Value: "custom-value",
	})
	
	fmt.Println(playlist.String())
}

处理加密片段

func handleEncryptedSegments() {
	playlist, _ := m3u8.NewMediaPlaylist(5, 5)
	
	// 添加加密信息
	key := &m3u8.Key{
		Method:            "AES-128",
		URI:               "https://example.com/key.bin",
		IV:                "0x1234567890ABCDEF1234567890ABCDEF",
		KeyFormat:         "identity",
		KeyFormatVersions: "1",
	}
	
	playlist.Key = key
	
	// 添加加密片段
	playlist.Append("encrypted1.ts", 5.0, "")
	playlist.Append("encrypted2.ts", 5.0, "")
	
	fmt.Println(playlist.String())
}

注意事项

1. 对于媒体播放列表，如果是VOD类型，必须调用Close()方法
2. 创建媒体播放列表时需要指定初始容量和最大容量
3. 解析时注意检查播放列表类型(IsMaster())
4. 处理大播放列表时注意内存使用

总结

go-m3u8 库提供了完整的 m3u8 播放列表解析和生成功能，支持 HLS 协议的各种特性。通过这个库，你可以轻松地在 Go 应用中集成 HLS 流媒体处理能力。

如果需要更高级的功能，可以查看库的源代码，它提供了丰富的 API 来处理各种 m3u8 特性和自定义需求。