Golang原生解析与解复用MPEG传输流(.ts)插件库go-astits的使用

这是一个用于在Go中原生解复用和复用MPEG传输流(ts)的库。

警告：该库尚未准备好用于生产环境，请自行承担使用风险！

安装

要安装库，请使用以下命令：

go get -u github.com/asticode/go-astits/...

要安装可执行文件，请使用以下命令：

go install github.com/asticode/go-astits/cmd

使用说明

传输流由数据包组成。每个数据包都有一个头部、一个可选的适配字段和一个有效载荷。多个有效载荷可以附加并解析为数据。

                                           TRANSPORT STREAM
 +--------------------------------------------------------------------------------------------------+
 |                                                                                                  |

                       PACKET                                         PACKET
 +----------------------------------------------+----------------------------------------------+----
 |                                              |                                              |

 +--------+---------------------------+---------+--------+---------------------------+---------+
 | HEADER | OPTIONAL ADAPTATION FIELD | PAYLOAD | HEADER | OPTIONAL ADAPTATION FIELD | PAYLOAD | ...
 +--------+---------------------------+---------+--------+---------------------------+---------+

                                      |         |                                    |         |
                                      +---------+                                    +---------+
                                           |                                              |
                                           +----------------------------------------------+
                                                                DATA

代码示例

解复用(Demux)

// 创建一个可取消的上下文，以便随时停止读取数据包/数据
ctx, cancel := context.WithCancel(context.Background())

// 处理SIGTERM信号
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGTERM)
go func() {
    <-ch
    cancel()
}()

// 打开文件或初始化任何类型的io.Reader
// 建议使用bufio.Reader进行缓冲以提高性能
f, _ := os.Open("/path/to/file.ts")
defer f.Close()

// 创建解复用器
dmx := astits.NewDemuxer(ctx, f)
for {
    // 获取下一个数据
    d, _ := dmx.NextData()
    
    // 如果是PMT数据
    if d.PMT != nil {
        // 遍历基本流
        for _, es := range d.PMT.ElementaryStreams {
                fmt.Printf("Stream detected: %d\n", es.ElementaryPID)
        }
        return
    }
}

复用(Mux)

// 创建一个可取消的上下文，以便随时停止写入数据包/数据
ctx, cancel := context.WithCancel(context.Background())

// 处理SIGTERM信号
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGTERM)
go func() {
    <-ch
    cancel()
}()

// 创建文件或初始化任何类型的io.Writer
// 建议使用bufio.Writer进行缓冲以提高性能
f, _ := os.Create("/path/to/file.ts")
defer f.Close()

// 创建复用器
mx := astits.NewMuxer(ctx, f)

// 添加基本流
mx.AddElementaryStream(astits.PMTElementaryStream{
    ElementaryPID: 1,
    StreamType:    astits.StreamTypeMetadata,
})

// 写入表
// 使用该函数不是强制性的，WriteData会不时重新传输表
mx.WriteTables()

// 写入数据
mx.WriteData(&astits.MuxerData{
    PES: &astits.PESData{
        Data: []byte("test"),
    },
    PID: 1,
})

选项

要为解复用器或复用器传递选项，请查找前缀为"DemuxerOpt"或"MuxerOpt"的方法，并在调用"NewDemuxer"或"NewMuxer"时添加它们：

// 这是自定义的数据包解析器
p := func(ps []*astits.Packet) (ds []*astits.Data, skip bool, err error) {
        // 这里是你的逻辑
        skip = true
        return
}

// 现在你可以创建带有适当选项的解复用器
dmx := NewDemuxer(ctx, f, DemuxerOptPacketSize(192), DemuxerOptPacketsParser(p))

CLI

该库提供了2个CLI工具，它们会在执行"go get"时自动安装在"GOPATH/bin"中。

astits-probe

列出流

$ astits-probe -i <path to your file> -f <format: text|json (default: text)>

列出数据包

$ astits-probe packets -i <path to your file>

列出数据

$ astits-probe data -i <path to your file> -d <data type: eit|nit|... (repeatable argument | if empty, all data types are shown)>

astits-es-split

将流拆分为单独的.ts文件

$ astits-es-split <path to your file> -o <path to output dir>

特性和路线图

• [x] 添加解复用器
• [x] 添加复用器
• [x] 解复用PES数据包
• [x] 复用PES数据包
• [x] 解复用PAT数据包
• [x] 复用PAT数据包
• [x] 解复用PMT数据包
• [x] 复用PMT数据包
• [x] 解复用EIT数据包
• [ ] 复用EIT数据包
• [x] 解复用NIT数据包
• [ ] 复用NIT数据包
• [x] 解复用SDT数据包
• [ ] 复用SDT数据包
• [x] 解复用TOT数据包
• [ ] 复用TOT数据包
• [ ] 解复用BAT数据包
• [ ] 复用BAT数据包
• [ ] 解复用DIT数据包
• [ ] 复用DIT数据包
• [ ] 解复用RST数据包
• [ ] 复用RST数据包
• [ ] 解复用SIT数据包
• [ ] 复用SIT数据包
• [ ] 复用ST数据包
• [ ] 解复用TDT数据包
• [ ] 复用TDT数据包
• [ ] 解复用TSDT数据包
• [ ] 复用TSDT数据包