Go语言利用原生库实现屏幕截取和实时推流功能

更新时间：2025年06月03日 10:21:06 作者：GetcharZp

这篇文章主要为大家详细介绍了Go语言利用原生库实现一个高效的屏幕实时截图推送服务,文中的示例代码讲解详细,有需要的小伙伴可以了解下

核心功能

屏幕截取：使用Go捕获当前屏幕内容
实时推流：将视频流通过HTTP协议推送
Web播放：简单HTML页面验证流媒体

Go后端实现

1. 全局帧缓存模块 (frame_cache.go)

package main

import (
	"sync"
	"time"
)

var (
	frameCache     []byte
	frameCacheTime time.Time
	cacheMutex     sync.RWMutex
)

// 更新帧缓存
func updateFrameCache(frame []byte) {
	cacheMutex.Lock()
	defer cacheMutex.Unlock()
	frameCache = frame
	frameCacheTime = time.Now()
}

// 获取帧缓存
func getFrameCache() ([]byte, time.Time) {
	cacheMutex.RLock()
	defer cacheMutex.RUnlock()
	return frameCache, frameCacheTime
}

2. 截屏服务模块 (capture_service.go)

package main

import (
	"bytes"
	"image/jpeg"
	"log"
	"time"

	"github.com/kbinani/screenshot"
)

func startCaptureService() {
	ticker := time.NewTicker(time.Second / time.Duration(frameRate))
	defer ticker.Stop()

	for range ticker.C {
		img, err := screenshot.CaptureRect(screenBounds)
		if err != nil {
			log.Printf("截屏失败: %v", err)
			continue
		}

		buf := new(bytes.Buffer)
		if err := jpeg.Encode(buf, img, &jpeg.Options{Quality: quality}); err != nil {
			log.Printf("JPEG编码失败: %v", err)
			continue
		}

		updateFrameCache(buf.Bytes())
	}
}

3. 流处理模块 (stream_handler.go)

package main

import (
	"fmt"
	"net/http"
	"time"
)

func streamHandler(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Content-Type", "multipart/x-mixed-replace; boundary=frame")
	w.Header().Set("Cache-Control", "no-cache")
	w.Header().Set("Connection", "keep-alive")

	// 发送初始帧分隔符
	if _, err := w.Write([]byte("--frame\r\n")); err != nil {
		return
	}

	lastSent := time.Now()
	for {
		frame, frameTime := getFrameCache()

		// 只发送新帧
		if frameTime.After(lastSent) {
			if _, err := fmt.Fprintf(w,
				"Content-Type: image/jpeg\r\nContent-Length: %d\r\n\r\n",
				len(frame)); err != nil {
				return
			}

			if _, err := w.Write(frame); err != nil {
				return
			}

			if _, err := w.Write([]byte("\r\n--frame\r\n")); err != nil {
				return
			}

			if f, ok := w.(http.Flusher); ok {
				f.Flush()
			}

			lastSent = frameTime
		}

		// 防止CPU空转
		time.Sleep(time.Second / time.Duration(frameRate*2))

		select {
		case <-r.Context().Done():
			return
		default:
		}
	}
}

4. 主程序模块 (main.go)

package main

import (
	"image"
	"log"
	"net/http"

	"github.com/kbinani/screenshot"
)

var (
	quality      = 50                     // JPEG质量(1-100)
	frameRate    = 10                     // 帧率(FPS)
	screenBounds = image.Rect(0, 0, 0, 0) // 屏幕尺寸
)

func main() {
	// 初始化截屏区域
	if n := screenshot.NumActiveDisplays(); n <= 0 {
		log.Fatal("未检测到活动显示器")
	}
	screenBounds = screenshot.GetDisplayBounds(0)

	// 启动截屏服务
	go startCaptureService()

	// 设置路由
	http.HandleFunc("/stream", streamHandler)
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		http.ServeFile(w, r, "index.html")
	})

	// 启动服务器
	log.Println("服务启动: http://localhost:8080")
	log.Fatal(http.ListenAndServe(":8080", nil))
}

三、Web前端实现

创建一个简单的index.html文件，前端直接用 image 标签进行显示即可：

<!DOCTYPE html>
<html>
<head>
    <title>Go屏幕流测试</title>
    <style>
        body { font-family: Arial, sans-serif; text-align: center; margin-top: 50px; }
        #video { max-width: 90%; border: 1px solid #ccc; margin: 20px auto; }
    </style>
</head>
<body>
    <h1>Go屏幕流测试播放器</h1>
    <img id="video" src="/stream" alt="视频流">
    
    <div>
        <button onclick="window.location.reload()">重新加载</button>
    </div>
    
    <script>
        // 自动重连逻辑
        const video = document.getElementById('video');
        video.onerror = function() {
            setTimeout(function() {
                video.src = '/stream?t=' + new Date().getTime();
            }, 1000);
        };
    </script>
</body>
</html>

四、运行步骤

1.安装依赖库：

go get github.com/kbinani/screenshot

2.运行Go程序：

go run main.go

3.打开浏览器访问： http://localhost:8080

五、技术要点说明

使用screenshot库捕获屏幕内容

采用MJPEG(Motion-JPEG)格式推送视频流

前端通过简单的<img>标签即可播放视频流

自动重连机制确保网络不稳定时的观看体验

到此这篇关于Go语言利用原生库实现屏幕截取和实时推流功能的文章就介绍到这了,更多相关Go屏幕截取内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

go
截屏

go语言中if语句用法实例
这篇文章主要介绍了go语言中if语句用法,以实例形式分析了if语句的定义及使用技巧,非常具有实用价值,需要的朋友可以参考下
2015-02-02
Golang throttled基于GCRA速率限制库使用探索
这篇文章主要为大家介绍了Golang throttled基于GCRA速率限制库使用实例探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
盘点几种Go语言开发的IDE
Go语言作为一种新兴的编程语言,近年来受到了越来越多的关注,它以其简洁、高效和并发性能而闻名,被广泛应用于各种软件开发项目中,本文将介绍几种常用的Go语言IDE,并对它们进行比较,帮助开发者根据自己的需求选择合适的工具,需要的朋友可以参考下
2023-11-11
golang创建文件目录os.Mkdir,os.MkdirAll的区别说明
本文主要讲述os.Mkdir、os.MkdirAll区别以及在创建文件目录过程中的一些其他技巧，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
GO语言类型转换和类型断言实例分析
这篇文章主要介绍了GO语言类型转换和类型断言,以实例形式详细分析了类型转换和类型断言的概念与使用技巧,需要的朋友可以参考下
2015-01-01
Go 语言json.Unmarshal 遇到的小问题(推荐)
这篇文章主要介绍了 Go 语言json.Unmarshal 遇到的小问题，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2020-07-07
golang bad file descriptor问题的解决方法
这篇文章主要给大家介绍了golang bad file descriptor问题的解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
2019-02-02
golang语言中for循环语句用法实例
这篇文章主要介绍了golang语言中for循环语句用法,实例分析了for循环遍历的使用技巧,具有一定参考借鉴价值,需要的朋友可以参考下
2015-01-01
Go之集合slice的实现
本文主要介绍了Go之集合slice的实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-08-08
利用rpm打包上线部署golang代码的方法教程
RPM是RPM Package Manager（RPM软件包管理器）的缩写，这篇文章主要给大家介绍了关于利用rpm打包上线部署golang代码的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。
2018-01-01