golang 创建一个简单的广播echo服务器

本文介绍了一个简单的广播服务器实现,包括客户端连接处理、消息广播等功能。通过Go语言实现了一个能够支持多个客户端连接并进行消息广播的服务器。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

package main;

import (
	"net"
	"fmt"
	"bufio"
)

//里面的代码部分参考cmu440课程
//https://github.com/cmu440/p0

//广播服务器接口
type MultiEchoServer interface {
	//开始
	Start(port int) error;
	//停止
	Close();
	//当前客户端连接数
	Count() int;
}

//广播服务器
type multiEchoServer struct {
	lis *net.TCPListener;
	//当前客户端ID
	curClientId int;
	//所有客户端
	clients map[int]*client;
	//广播消息
	broadcastMsg chan []byte;
}

//客户端
type client struct {
	//ID
	id int;
	//连接
	conn net.Conn;
	//接收消息
	recvMsg chan []byte;
	//发送消息
	sendMsg chan []byte;
	//接收消息是否关闭
	isRecvMsgClose chan bool;
	//发送消息是否关闭
	isSendMsgClose chan bool;
	//服务器
	mes *multiEchoServer;
}

//返回一个广播服务器
func New() *multiEchoServer {
	return &multiEchoServer{
		curClientId:  0,
		clients:      make(map[int]*client),
		broadcastMsg: make(chan []byte, 1),
	};
}

//启动服务器
func (m *multiEchoServer) Start(port int) error {
	//获取tcp地址
	addr, err := net.ResolveTCPAddr("tcp", fmt.Sprintf(":%d", port));
	if err != nil {
		return err;
	}
	//监听端口
	m.lis, err = net.ListenTCP("tcp", addr);
	if err != nil {
		return err;
	}
	//启一个goroutine处理广播
	go m.BroadcastLoop();
	//启一个goroutine处理客户端来的连接
	go func() {
		for {
			conn, err := m.lis.Accept();
			if err != nil {
				continue;
			}
			cli := &client{
				id:             m.curClientId,
				conn:           conn,
				recvMsg:        make(chan []byte, 1),
				sendMsg:        make(chan []byte, 1),
				isRecvMsgClose: make(chan bool, 1),
				isSendMsgClose: make(chan bool, 1),
				mes:            m,
			}
			//加客户端加入到服务器clients中
			clis := m.clients;
			clis[m.curClientId] = cli;
			m.clients = clis;
			m.curClientId++;

			//启两个goroutine分别处理客户端的接收与发送消息
			go cli.RecvLoop();
			go cli.SendLoop();
		}
	}();
	return nil;
}

//停止服务器
func (m *multiEchoServer) Close() {
	m.lis.Close();
	//循环关闭客户端
	for _, client := range m.clients {
		client.conn.Close();
		//这里只需给一个发送消息就好了
		client.isRecvMsgClose <- true;
	}
}

//返回当前客户端连接数
func (m *multiEchoServer) Count() int {
	return len(m.clients);
}

//处理广播
func (m *multiEchoServer) BroadcastLoop() {
	for {
		select {
		case data := <-m.broadcastMsg:
			{
				//遍历所有客户端,循环发送消息
				for _, client := range m.clients {
					client.sendMsg <- data;
				}
				break;
			}
		}
	}
}

//删除客户端
func (m *multiEchoServer) DelClient(c *client) error {
	c.conn.Close();
	clis := m.clients;
	delete(clis, c.id);
	m.clients = clis;
	return nil;
}

//处理客户端接收消息
func (c *client) RecvLoop() {
	defer func() {
		fmt.Println(c.conn.RemoteAddr().String() + " RecvLoop exit");
	}();
	for {
		read := bufio.NewReader(c.conn);
		data, err := read.ReadBytes('\n');
		if err != nil {
			c.isSendMsgClose <- true;
			//这里直接返回,如果不直接返回
			//当客户端退出时,这里会运行2次,导致c.isSendMsgClose<-true执行2次造成阻塞
			//没有机会运行后面的select,那么一直无法返回,不能回收。
			return;
		}

		select {
		//接收消息是否关闭
		case <-c.isRecvMsgClose:
			{
				c.isSendMsgClose <- true;
				return;
			}
		//广播消息
		case c.mes.broadcastMsg <- data:
			{
				break;
			}
		}
	}
}

//处理客户端发送消息
func (c *client) SendLoop() {
	defer func() {
		fmt.Println(c.conn.RemoteAddr().String() + " SendLoop exit");
	}();
	for {
		select {
		//发送消息关闭,则把客户端从服务中删除
		case <-c.isSendMsgClose:
			{
				c.mes.DelClient(c);
				return;
			}
		//向客户写入要发送的消息
		case data := <-c.sendMsg:
			{
				_, err := c.conn.Write(data);
				if err != nil {
					return;
				}
			}
		}
	}
}

func main() {
	mes := New();
	mes.Start(8888);

	//循环
	select {};
}

<think>好的,我需要帮助用户了解如何用Golang构建视频播放服务器,特别是使用HLS协议。根据用户提供的引用资料,我需要整合这些信息,提供一个清晰的步骤指南。 首先,用户提到了几个相关的引用,其中引用[1]和[2]提到了基于Golang的HLS服务器实现,使用ffmpeg进行转码。引用[3]涉及RTMP协议,可能可以作为补充,但用户主要关注HLS。引用[4]指出HLS主要用于播放,发布可能需要其他协议如RTP,但用户的问题集中在构建播放服务器,所以暂时可以忽略发布部分。引用[5]提到了lal这个开源项目,可能需要介绍一下。 我应该从整体架构开始,分步骤说明。首先需要视频处理,使用ffmpeg将视频转码为HLS所需的TS片段和m3u8索引文件。然后搭建HTTP服务器,用Golang提供静态文件服务。可能需要动态处理转码请求,比如用户上传视频时自动转码。同时,可以提到现有的开源项目如Golang HLS Streamer和lal,供用户参考。 需要确保每个步骤都有具体的代码示例,比如使用Golang的http包创建静态服务器,调用ffmpeg的命令行参数。还要注意引用对应的资料,比如引用[1]和[2]中的项目,以及引用[5]中的lal。 可能的问题点是视频转码的参数设置和动态处理部分,需要详细说明ffmpeg的命令,以及如何在Golang中执行外部命令。此外,静态文件服务的路径设置和MIME类型可能需要配置,避免浏览器无法正确加载m3u8和TS文件。 最后,生成相关问题时要围绕HLS服务器构建、ffmpeg的使用、性能优化、与RTMP的比较以及开源项目选择等方面。</think>### 如何用 Golang 构建视频流媒体服务器(HLS 协议实现方案) #### 1. **核心架构设计** 基于 HLS 的视频流服务器通常分为以下模块: - **视频处理模块**:使用 `ffmpeg` 将原始视频转码为 HLS 格式(TS 分片 + m3u8 索引文件)[^1][^2] - **HTTP 静态服务模块**:通过 Golang 提供 m3u8 文件和 TS 分片的静态文件服务 - **动态转码模块**(可选):实时接收上传视频触发转码流程 #### 2. **具体实现步骤** ##### **2.1 视频预处理** 使用 ffmpeg 将视频转换为 HLS 格式: ```bash ffmpeg -i input.mp4 \ -c:v h264 -flags +cgop -g 30 -hls_time 10 \ -hls_list_size 0 -f hls output.m3u8 ``` 参数说明: - `-hls_time 10`:每个 TS 分片 10 秒 - `-hls_list_size 0`:m3u8 保留全部分片记录 ##### **2.2 基础静态服务器实现** ```go package main import ( "net/http" "path/filepath" ) func main() { fs := http.FileServer(http.Dir("hls_output")) http.Handle("/stream/", http.StripPrefix("/stream/", fs)) http.ListenAndServe(":8080", nil) } ``` 目录结构示例: ``` hls_output/ ├── video1/ │ ├── output.m3u8 │ └── segment0.ts └── video2/ ├── playlist.m3u8 └── seg_001.ts ``` ##### **2.3 动态转码集成** ```go func transcodeVideo(inputPath string) error { cmd := exec.Command("ffmpeg", "-i", inputPath, "-c:v", "libx264", "-hls_time", "10", "-hls_list_size", "0", filepath.Join("hls_output", "output.m3u8")) return cmd.Run() } ``` #### 3. **进阶优化方案** 1. **自适应码率**:生成多分辨率版本创建主 m3u8: ```bash ffmpeg -i input.mp4 \ -map 0:v:0 -map 0:a:0 -map 0:v:0 -map 0:a:0 \ -c:v:0 libx264 -b:v:0 2000k \ -c:v:1 libx264 -b:v:1 1000k \ -var_stream_map "v:0,a:0 v:1,a:1" \ -master_pl_name master.m3u8 \ -f hls out_%v.m3u8 ``` 2. **内存缓存**:使用 `sync.Pool` 缓存 TS 分片减少 IO 压力 3. **CDN 集成**:通过 `http.Redirect` 实现边缘节点调度 #### 4. **现成开源方案参考** - **lal 项目**:支持多种协议(RTMP/HLS/HTTP-FLV)的流媒体服务器[^5] - **Golang HLS Streamer**:轻量级 HLS 服务实现(支持目录监控自动转码)[^2] - **rtmp 库**:可用于扩展 RTMP 推流功能[^3] #### 5. **部署注意事项** 1. 设置正确的 MIME 类型: ```go mime.AddExtensionType(".m3u8", "application/vnd.apple.mpegurl") mime.AddExtensionType(".ts", "video/MP2T") ``` 2. 跨域配置: ```go fs := http.FileServer(http.Dir("hls_output")) handler := cors.Default().Handler(fs) ``` 3. 性能监控:集成 Prometheus 指标采集
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值