目录
问题
解决
问题
如何根据一个流媒体地址URL判断对应的流媒体协议，比如RTMP、RTSP协议等。
其本质就是一个简单的字符串操作，通过这个需求，我们来了解一下Golang是如何处理字符串的。
解决
这里提供一个产品级方法，可以直接拿来用。
（已经经过项目验证，质量嘛，有保证）
func getProtocol(url string) (string, error) {
// golang的判断语句没有括号，一开始挺不适应的
if url != "" {
// 获取冒号索引位置
index := strings.Index(url, ":")
// 找到了说明URL正确，否则报错
if index > 0 {
// 这里需要注意两点：
// 1. golang字符串的索引原则是左闭右开
// 2. 协议类型最好进行格式化处理再输出返回，比如这里是“最大化”（大写格式）
return strings.ToUpper(url[0:index]), nil
} else {
// 注意提前定义ErrorInvalidURL类型
return "", ErrorInvalidURL
}
}
return "", ErrorNullURL
}
上面的 getProtocol() 方法会输出流媒体协议的大写字符串标识。其实，任何开发语言都一样，我们在进行字符串比较的时候都要主要先进行大写或者小写的统一，再进行比较。
最后，给出一个完整的代码实例，大家也可以自己用下面的代码测试，验证是否真的好用：
// 声明包
package main
// 导入依赖包
import(
"fmt"
"strings"
)
// 定义宏变量类型，这里定义了两种错误信息
var (
ErrorNullURL = fmt.Errorf("url is null")
ErrorInvalidURL = fmt.Errorf("url is not valid")
)
// 判断流媒体协议类型的核心方法
func getProtocol(url string) (string, error) {
if url != "" {
index := strings.Index(url, ":")
if index > 0 {
return strings.ToUpper(url[0:index]), nil
} else {
return "", ErrorInvalidURL
}
}
return "", ErrorNullURL
}
// 主方法
func main() {
// demo实例
url1 := "rtmp://172.0.0.1:1935/test/live"
url2 := "rtsp://172.0.0.1:1554/live"
// 测试1
if pro, err := getProtocol(url1); err == nil {
fmt.Println("url1 protocol: ", pro)
}
// 测试2
if pro, err := getProtocol(url2); err == nil {
fmt.Println("url2 protocol: ", pro)
}
}
将上述代码保存为 url-check.go 文件。
执行上述程序，运行如下命令：
go run url-check.go
上面程序的运行结果如下：
url1 protocol: RTMP
url2 protocol: RTSP
整体运行过程的截图如下：
推荐阅读：
