go语言并发获取hls,测试流媒体服务的拉流压力

前两天,我们的开发的兄弟们终于将我们的流媒体整体方案实现并部署了,虽然都是用的开源的项目,但是我们内部还是要测试服务表现的,这个任务就被扔给我了。

先不看内部各个链路的性能损耗,先测试单源,多路点播取流的场景。

使用golang,写程序如下:

package main

import (
	"bytes"
	"fmt"
	"io"
	"net/http"
	"os"
	"regexp"
	"strconv"
	"time"
)

/*this application is used to get rtsp stream data for me*/

func main() {
	var Host = ""
	var Tim = 1
	for _, a := range os.Args[1:] {
		if string(a) == "--help" {
			fmt.Println("用法为-g 输入goal")
			fmt.Println("用法为-t 输入threat num")
		}
		if string(a) == "/?" {
			fmt.Println("用法为-g 输入goal")
			fmt.Println("用法为-t 输入threat num")
		}
		m, err := regexp.MatchString("^-[g,t]", string(a))
		reg := regexp.MustCompile(`^-[g,t]`)
		if m {
			host := reg.FindAllString(string(a), -2)[0]
			if host == "-g" {
				pp := []rune(a)
				Host = string(pp[2:len(pp)])
			}
			if host == "-t" {
				pp := []rune(a)
				tim, err := strconv.Atoi(string(pp[2:len(pp)]))
				Tim = tim
				if err != nil {
					fmt.Printf("输入数字哦")
					return
				}
			}
		}
		if err != nil {
			fmt.Println(err)
			return
		}
	}

	if Host == "" {
		fmt.Println("please input /? for help")
		fmt.Println(string(Tim))
		return
	} else {
		for i := 1; i < Tim; i++ {
			fmt.Printf("拉起第%d个文件流\n", i)
			go getHttpHls(Host)
		}
		fmt.Println("拉起主线程文件流")
		getHttpHls(Host)
	}
}

func getHttpHls(h string) {
	req, _ := http.NewRequest("GET", h, nil)
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		// handle error
		fmt.Println(err)
		return
	}
	defer resp.Body.Close()

	//here we open a loop to read stream
	buffer := make([]byte, 8192)
	for {
		time.Sleep(500)
		_, _ = resp.Body.Read(buffer)
	}
}

  

 使用步骤:
1、使用命令:go build get_hls.go,编译
2、使用:./get_hls  -ghttp://127.0.0.1:8082/live/20_unv01_1.flv -t1000,通过这样的方式来获取1000个流,并将流直接抛弃
 

 

 

 

说明:

1、我这边自己到网上下载了一个流媒体拉流的工具,使用的时候,流媒体拉流工具的性能消耗比我的srs消耗还要大,而且一段时间会自动kill掉,所以写以上程序来埋坑

2、因为公司局域网是百兆局域网,是没有办法测试出我们服务的压力的,所以这里都是使用还回的方式在用,单路码率为4Mbps左右,拉22路码流就上不去了

 

上面这个程序很鄙陋,希望能够有高手予以指导,程序中,后面的buffer是读取到垃圾里面的,但是如果能够不读取,直接new处理一下,应该效果更好。

 

posted on 2019-05-10 14:35  &大飞  阅读(372)  评论(0编辑  收藏  举报

导航