添加设备视频流获取基本功能-测试
This commit is contained in:
lijun
commit
b16147d271
|
|
@ -0,0 +1,8 @@
|
||||||
|
# Default ignored files
|
||||||
|
/shelf/
|
||||||
|
/workspace.xml
|
||||||
|
# Editor-based HTTP Client requests
|
||||||
|
/httpRequests/
|
||||||
|
# Datasource local storage ignored files
|
||||||
|
/dataSources/
|
||||||
|
/dataSources.local.xml
|
||||||
|
|
@ -0,0 +1,9 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<module type="WEB_MODULE" version="4">
|
||||||
|
<component name="Go" enabled="true" />
|
||||||
|
<component name="NewModuleRootManager">
|
||||||
|
<content url="file://$MODULE_DIR$" />
|
||||||
|
<orderEntry type="inheritedJdk" />
|
||||||
|
<orderEntry type="sourceFolder" forTests="false" />
|
||||||
|
</component>
|
||||||
|
</module>
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="ProjectModuleManager">
|
||||||
|
<modules>
|
||||||
|
<module fileurl="file://$PROJECT_DIR$/.idea/VideoStream.iml" filepath="$PROJECT_DIR$/.idea/VideoStream.iml" />
|
||||||
|
</modules>
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
|
|
@ -0,0 +1,41 @@
|
||||||
|
module VideoStream
|
||||||
|
|
||||||
|
go 1.21
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/gin-gonic/gin v1.10.0
|
||||||
|
github.com/go-redis/redis/v8 v8.11.5
|
||||||
|
github.com/robfig/cron/v3 v3.0.1
|
||||||
|
gocv.io/x/gocv v0.39.0
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/bytedance/sonic v1.11.6 // indirect
|
||||||
|
github.com/bytedance/sonic/loader v0.1.1 // indirect
|
||||||
|
github.com/cespare/xxhash/v2 v2.1.2 // indirect
|
||||||
|
github.com/cloudwego/base64x v0.1.4 // indirect
|
||||||
|
github.com/cloudwego/iasm v0.2.0 // indirect
|
||||||
|
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.3 // indirect
|
||||||
|
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||||
|
github.com/go-playground/locales v0.14.1 // indirect
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||||
|
github.com/go-playground/validator/v10 v10.20.0 // indirect
|
||||||
|
github.com/goccy/go-json v0.10.2 // indirect
|
||||||
|
github.com/json-iterator/go v1.1.12 // indirect
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.7 // indirect
|
||||||
|
github.com/leodido/go-urn v1.4.0 // indirect
|
||||||
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.2 // indirect
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
|
||||||
|
github.com/ugorji/go/codec v1.2.12 // indirect
|
||||||
|
golang.org/x/arch v0.8.0 // indirect
|
||||||
|
golang.org/x/crypto v0.23.0 // indirect
|
||||||
|
golang.org/x/net v0.25.0 // indirect
|
||||||
|
golang.org/x/sys v0.20.0 // indirect
|
||||||
|
golang.org/x/text v0.15.0 // indirect
|
||||||
|
google.golang.org/protobuf v1.34.1 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,111 @@
|
||||||
|
github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=
|
||||||
|
github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=
|
||||||
|
github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=
|
||||||
|
github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
|
||||||
|
github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE=
|
||||||
|
github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
|
github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
|
||||||
|
github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
|
||||||
|
github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
|
||||||
|
github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
|
||||||
|
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
|
||||||
|
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
||||||
|
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||||
|
github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
|
||||||
|
github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
|
||||||
|
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
|
||||||
|
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
|
||||||
|
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
|
||||||
|
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
|
||||||
|
github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=
|
||||||
|
github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
|
||||||
|
github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=
|
||||||
|
github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
|
||||||
|
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
|
||||||
|
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
|
||||||
|
github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
|
||||||
|
github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
|
||||||
|
github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
|
||||||
|
github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM=
|
||||||
|
github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
|
||||||
|
github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
|
||||||
|
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
|
||||||
|
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
|
||||||
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
|
||||||
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
|
||||||
|
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
|
||||||
|
github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=
|
||||||
|
github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
|
||||||
|
github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=
|
||||||
|
github.com/onsi/ginkgo v1.16.5/go.mod h1:+E8gABHa3K6zRBolWtd+ROzc/U5bkGt0FwiG042wbpU=
|
||||||
|
github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=
|
||||||
|
github.com/onsi/gomega v1.18.1/go.mod h1:0q+aL8jAiMXy9hbwj2mr5GziHiwhAIQpFmmtT5hitRs=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
|
||||||
|
github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
|
||||||
|
github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
|
||||||
|
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
|
||||||
|
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
|
||||||
|
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
|
||||||
|
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
|
||||||
|
github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
|
||||||
|
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
|
||||||
|
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
|
||||||
|
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
|
||||||
|
github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
|
||||||
|
github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
|
||||||
|
github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
|
||||||
|
gocv.io/x/gocv v0.39.0 h1:vWHupDE22LebZW6id2mVeT767j1YS8WqGt+ZiV7XJXE=
|
||||||
|
gocv.io/x/gocv v0.39.0/go.mod h1:zYdWMj29WAEznM3Y8NsU3A0TRq/wR/cy75jeUypThqU=
|
||||||
|
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
|
||||||
|
golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
|
||||||
|
golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
|
||||||
|
golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI=
|
||||||
|
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
|
||||||
|
golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
|
||||||
|
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
|
||||||
|
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
|
golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=
|
||||||
|
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||||
|
golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=
|
||||||
|
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
|
||||||
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
|
||||||
|
google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
|
||||||
|
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||||
|
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
|
||||||
|
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
|
||||||
|
|
@ -0,0 +1,78 @@
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"VideoStream/proto"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"gocv.io/x/gocv"
|
||||||
|
"io"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SetUpToolGroup(router *gin.Engine) {
|
||||||
|
toolGroup := router.Group("/tool")
|
||||||
|
toolGroup.GET("/video_stream", GetVideoStream)
|
||||||
|
}
|
||||||
|
|
||||||
|
type videoStreamReq struct {
|
||||||
|
ID int `json:"id" form:"id"`
|
||||||
|
Key string `json:"key" form:"key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetVideoStream(c *gin.Context) {
|
||||||
|
var req videoStreamReq
|
||||||
|
if err := c.ShouldBind(&req); err != nil {
|
||||||
|
c.JSON(400, gin.H{"error": err.Error()})
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
//查看id是否存在
|
||||||
|
index := -1
|
||||||
|
for _, device := range proto.Config.DeviceInfo {
|
||||||
|
if device.ID == req.ID {
|
||||||
|
index = req.ID
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if index == -1 {
|
||||||
|
c.JSON(400, gin.H{"error": "id not exist"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
//查看key是否正确
|
||||||
|
if req.Key != "123456" {
|
||||||
|
c.JSON(400, gin.H{"error": "key error"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
//设备流
|
||||||
|
c.Stream(func(w io.Writer) bool {
|
||||||
|
var count int
|
||||||
|
for {
|
||||||
|
frame, cnt := GetDeviceCurrentFrame(req.ID)
|
||||||
|
if cnt == count {
|
||||||
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
//gocv.Matrix转为jpeg
|
||||||
|
img, err := gocv.IMEncode(".jpg", frame)
|
||||||
|
frame_ := img.GetBytes()
|
||||||
|
|
||||||
|
_, err = w.Write([]byte("--frame\r\nContent-Type: image/jpeg\r\n\r\n"))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("写入头部信息错误: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err = w.Write(frame_)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("写入帧数据错误: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err = w.Write([]byte("\r\n"))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("写入帧结束标记错误: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
time.Sleep(50 * time.Millisecond) // 控制帧率，模拟每秒约20帧，可按实际调整
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,199 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"VideoStream/handler"
|
||||||
|
"VideoStream/proto"
|
||||||
|
"VideoStream/worker"
|
||||||
|
"fmt"
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
"github.com/robfig/cron/v3"
|
||||||
|
"gocv.io/x/gocv"
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var DeviceRWMap = make(map[int]sync.RWMutex)
|
||||||
|
var DeviceCurrentFrameMap = make(map[int]gocv.Mat)
|
||||||
|
var DeviceFrameCount = make(map[int]int)
|
||||||
|
var DeviceIsStop = make(map[int]bool)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
gin.SetMode(gin.ReleaseMode)
|
||||||
|
r := gin.Default()
|
||||||
|
handler.SetUpToolGroup(r) // Tool
|
||||||
|
err := worker.InitRedis()
|
||||||
|
if err != nil {
|
||||||
|
panic("failed to connect redis:" + err.Error())
|
||||||
|
}
|
||||||
|
//定时任务
|
||||||
|
c := cron.New(cron.WithSeconds())
|
||||||
|
// 添加每 10 秒执行一次的任务
|
||||||
|
_, err2 := c.AddFunc("@every 10s", myTask)
|
||||||
|
if err2 != nil {
|
||||||
|
log.Fatal("添加定时任务失败: ", err2)
|
||||||
|
}
|
||||||
|
c.Start()
|
||||||
|
|
||||||
|
err3 := r.Run(":" + proto.Config.SERVER_PORT)
|
||||||
|
if err3 != nil {
|
||||||
|
panic("failed to run server:" + err3.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
//读取配置文件
|
||||||
|
//文件地址/home/videoplayer/vp.conf
|
||||||
|
configPath := "/home/videoplayer/vp_stream.conf"
|
||||||
|
//读取配置文件
|
||||||
|
err := proto.ReadConfig(configPath)
|
||||||
|
if err != nil {
|
||||||
|
panic("failed to read config file:" + err.Error())
|
||||||
|
}
|
||||||
|
for _, device := range proto.Config.DeviceInfo {
|
||||||
|
DeviceRWMap[device.ID] = sync.RWMutex{}
|
||||||
|
DeviceCurrentFrameMap[device.ID] = gocv.NewMat()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func myTask() {
|
||||||
|
log.Println("每10秒执行一次")
|
||||||
|
}
|
||||||
|
|
||||||
|
func ReadConfigAndSetSystem() {
|
||||||
|
configPath := "/home/videoplayer/vp_stream.conf"
|
||||||
|
//读取配置文件
|
||||||
|
err := proto.ReadConfig(configPath)
|
||||||
|
if err != nil {
|
||||||
|
panic("failed to read config file:" + err.Error())
|
||||||
|
}
|
||||||
|
//检测是否需要获取设备流
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetDeviceCurrentFrame(frame gocv.Mat, device_id int) error {
|
||||||
|
//获取读写锁
|
||||||
|
mutex, ok := DeviceRWMap[device_id]
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("设备:%s 读写锁不存在", device_id)
|
||||||
|
}
|
||||||
|
mutex.Lock()
|
||||||
|
defer mutex.Unlock()
|
||||||
|
//设置当前帧
|
||||||
|
DeviceCurrentFrameMap[device_id] = frame
|
||||||
|
DeviceFrameCount[device_id]++
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetDeviceCurrentFrame(device_id int) (gocv.Mat, int) {
|
||||||
|
//获取读写锁
|
||||||
|
mutex, ok := DeviceRWMap[device_id]
|
||||||
|
if !ok {
|
||||||
|
return gocv.NewMat(), -1
|
||||||
|
}
|
||||||
|
mutex.RLock()
|
||||||
|
defer mutex.RUnlock()
|
||||||
|
//获取当前帧
|
||||||
|
return DeviceCurrentFrameMap[device_id], DeviceFrameCount[device_id]
|
||||||
|
}
|
||||||
|
|
||||||
|
func getVideoFrame(device proto.DeviceInfo) {
|
||||||
|
webcam, err := gocv.OpenVideoCapture(device.Stream)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("设备:%s 错误: 无法打开视频流，err: %v\n", device.ID, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer webcam.Close()
|
||||||
|
// 字体相关设置，对应OpenCV默认字体等，这里简化处理，实际可按需求调整
|
||||||
|
font := gocv.FontHersheySimplex
|
||||||
|
fontScale := 0.5
|
||||||
|
fontColor := color.RGBA{G: 255}
|
||||||
|
lineType := 2
|
||||||
|
z := 0
|
||||||
|
for {
|
||||||
|
if device.LogFrame > 0 && z%device.LogFrame == 0 {
|
||||||
|
fmt.Printf("设备:%s 当前帧: %d\n", device.ID, z)
|
||||||
|
}
|
||||||
|
if device.NextStop {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
frame := gocv.NewMat()
|
||||||
|
ok := webcam.Read(&frame)
|
||||||
|
if !ok {
|
||||||
|
fmt.Printf("设备:%s 错误: 无法从视频流中读取帧\n", device)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if frame.Empty() {
|
||||||
|
fmt.Printf("设备:%s 错误: 无法从视频流中读取帧\n", device)
|
||||||
|
//等待50ms
|
||||||
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
height := frame.Rows()
|
||||||
|
width := frame.Cols()
|
||||||
|
if height < device.CheckFrameHeight || width < device.CheckFrameWidth {
|
||||||
|
fmt.Printf("设备:%s 帧尺寸已改变\n", device)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
currentTime := time.Now().Format("2006-01-02 15:04:05")
|
||||||
|
gocv.PutText(&frame, currentTime, image.Point{10, 20}, font, fontScale, fontColor, lineType)
|
||||||
|
//需要将帧付给全局变量
|
||||||
|
err := SetDeviceCurrentFrame(frame, device.ID)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("设备:%s 错误: 无法设置当前帧\n", device)
|
||||||
|
}
|
||||||
|
z++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发起get请求,返回响应状态码
|
||||||
|
func Get(url string) int {
|
||||||
|
req, err := http.NewRequest("GET", url, nil)
|
||||||
|
var client = &http.Client{}
|
||||||
|
if err != nil {
|
||||||
|
return 500
|
||||||
|
}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return 500
|
||||||
|
}
|
||||||
|
return resp.StatusCode
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetVideoStream(id int) {
|
||||||
|
for {
|
||||||
|
var device proto.DeviceInfo
|
||||||
|
var index int
|
||||||
|
//获取设备信息
|
||||||
|
for i, device1 := range proto.Config.DeviceInfo {
|
||||||
|
if device1.ID == id {
|
||||||
|
device = device1
|
||||||
|
break
|
||||||
|
}
|
||||||
|
index = i
|
||||||
|
}
|
||||||
|
if index == len(proto.Config.DeviceInfo) {
|
||||||
|
//设备不存在
|
||||||
|
log.Println("device:", id, " not found")
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if device.NextStop {
|
||||||
|
DeviceIsStop[id] = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
//设置设备控制信息
|
||||||
|
status := Get(device.Control)
|
||||||
|
DeviceIsStop[id] = false
|
||||||
|
log.Println("device:", device.ID, " set control info status:", status)
|
||||||
|
getVideoFrame(device)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func DoGetVideoStream() {
|
||||||
|
for _, device := range proto.Config.DeviceInfo {
|
||||||
|
go GetVideoStream(device.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,72 @@
|
||||||
|
package proto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
var Config ConfigStruct
|
||||||
|
|
||||||
|
type ConfigStruct struct {
|
||||||
|
REDIS_ADDR string `json:"redis_addr"`
|
||||||
|
TOKEN_USE_REDIS bool `json:"token_use_redis"`
|
||||||
|
REDIS_User_PW bool `json:"redis_user_pw"` // 是否使用密码
|
||||||
|
REDIS_PASSWORD string `json:"redis_password"`
|
||||||
|
REDIS_DB int `json:"redis_db"`
|
||||||
|
SERVER_PORT string `json:"SERVER_PORT" form:"SERVER_PORT"` // 服务器端口
|
||||||
|
DeviceInfo []DeviceInfo `json:"device_info" form:"device_info"` // 设备信息
|
||||||
|
}
|
||||||
|
|
||||||
|
type DeviceInfo struct {
|
||||||
|
ID int `json:"id" form:"id"`
|
||||||
|
Name string `json:"name" form:"name"`
|
||||||
|
IP string `json:"ip" form:"ip"`
|
||||||
|
Type string `json:"type" form:"type"`
|
||||||
|
Stream string `json:"stream" form:"stream"`
|
||||||
|
Control string `json:"control" form:"control"`
|
||||||
|
NextStop bool `json:"next_stop" form:"next_stop"` // 下一帧，是否停止
|
||||||
|
CheckFrameWidth int `json:"check_frame_width" form:"check_frame_width"` // 检测帧宽度
|
||||||
|
CheckFrameHeight int `json:"check_frame_height" form:"check_frame_height"` // 检测帧高度
|
||||||
|
LogFrame int `json:"log_frame" form:"log_frame"` // 日志帧间隔，默认值为0，即不输出日志
|
||||||
|
}
|
||||||
|
|
||||||
|
// 读取配置文件
|
||||||
|
func ReadConfig(path string) error {
|
||||||
|
//查看配置文件是否存在,不存在则创建
|
||||||
|
_, err := os.Stat(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Config file not found!")
|
||||||
|
//写入json文件
|
||||||
|
file, err := os.Create(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error creating config file")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
encoder := json.NewEncoder(file)
|
||||||
|
err = encoder.Encode(&Config)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error encoding config")
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
//读json文件
|
||||||
|
file, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error opening config file")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
decoder := json.NewDecoder(file)
|
||||||
|
err = decoder.Decode(&Config)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error decoding config")
|
||||||
|
} else {
|
||||||
|
if Config.SERVER_PORT == "" {
|
||||||
|
Config.SERVER_PORT = "5002" // 默认端口
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
package service
|
||||||
|
|
@ -0,0 +1,544 @@
|
||||||
|
package worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"VideoStream/proto"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"github.com/go-redis/redis/v8"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var RedisClient *redis.Client // Redis 客户端, 用于连接 Redis 服务器
|
||||||
|
func InitRedis() error {
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
if proto.Config.REDIS_User_PW == false {
|
||||||
|
// 连接redis
|
||||||
|
RedisClient = redis.NewClient(&redis.Options{
|
||||||
|
Addr: proto.Config.REDIS_ADDR, // Redis 服务器地址
|
||||||
|
DB: proto.Config.REDIS_DB, // 使用的数据库编号
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
// 连接redis
|
||||||
|
RedisClient = redis.NewClient(&redis.Options{
|
||||||
|
Addr: proto.Config.REDIS_ADDR, // Redis 服务器地址
|
||||||
|
Password: proto.Config.REDIS_PASSWORD, // 如果 Redis 设置了密码
|
||||||
|
DB: proto.Config.REDIS_DB, // 使用的数据库编号
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证 Redis 客户端是否可以正常工作
|
||||||
|
_, err := RedisClient.Ping(ctx).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error connecting to Redis: %v", err)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func CloseRedis() {
|
||||||
|
// 关闭 Redis 客户端
|
||||||
|
if err := RedisClient.Close(); err != nil {
|
||||||
|
fmt.Println("Error closing Redis client: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func IsContainKey(key string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.Exists(ctx, key).Result() // 检查键是否存在, 如果存在则返回 1, 否则返回 0
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if val == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置redis
|
||||||
|
func SetRedis(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
// 设置键值对, 0 表示不设置过期时间, 如果需要设置过期时间, 可以设置为 time.Second * 10 等
|
||||||
|
err := RedisClient.Set(ctx, key, value, time.Minute*30).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置redis,永久
|
||||||
|
func SetRedisForever(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.Set(ctx, key, value, 0).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置hash
|
||||||
|
func SetHashWithTime(key string, id int, name, email string, duration time.Duration) bool {
|
||||||
|
//捕获错误，如果错误返回
|
||||||
|
|
||||||
|
ctx := context.Background() // 创建一个上下文
|
||||||
|
fields := map[string]interface{}{
|
||||||
|
"id": strconv.Itoa(id),
|
||||||
|
"name": name,
|
||||||
|
"email": email,
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置哈希表的字段值, 0 表示不设置过期时间, 如果需要设置过期时间, 可以设置为 time.Second * 10 等
|
||||||
|
err := RedisClient.HSet(ctx, key, fields).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, time.Hour*10).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置redis hash，设置过期时间
|
||||||
|
func SetHash(key string, data map[string]interface{}) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.HSet(ctx, key, data).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("%v :Error setting hash: %v", key, err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, time.Minute*30).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("%v :Error setting expire: %v", key, err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func SetHashWithField(key string, field string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.HSet(ctx, key, field, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetHash(key string, field string) string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.HGet(ctx, key, field).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting hash: %v", err)
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetHashAll(key string) map[string]string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.HGetAll(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting hash: %v", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置redis
|
||||||
|
func SetRedisWithExpire(key string, value string, expire time.Duration) bool { // 设置键值对, 0 表示不设置过期时间, 如果需要设置过期时间, 可以设置为 time.Second * 10 等
|
||||||
|
ctx := context.Background()
|
||||||
|
// 设置键值对, 0 表示不设置过期时间, 如果需要设置过期时间, 可以设置为 time.Second * 10 等
|
||||||
|
err := RedisClient.Set(ctx, key, value, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取redis
|
||||||
|
func GetRedis(key string) string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.Get(ctx, key).Result() // 从 Redis 读取键值, 如果键不存在则返回空字符串, 如果出现错误则返回错误
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(key, " Error getting key: %v", err)
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// pop redis list from right,as stack
|
||||||
|
func PopRedisList(key string) string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.RPop(ctx, key).Result() // 从 Redis 读取键值, 如果键不存在则返回空字符串, 如果出现错误则返回错误
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(key, " Error reading from Redis: %v", err)
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// pop redis list from left,as queue
|
||||||
|
func PopRedisListLeft(key string) string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.LPop(ctx, key).Result() // 从 Redis 读取键值, 如果键不存在则返回空字符串, 如果出现错误则返回错误
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
func DelRedis(key string) {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.Del(ctx, key).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error deleting key: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// push redis list from right
|
||||||
|
func PushRedisList(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.RPush(ctx, key, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetRedisListLen(key string) int64 {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.LLen(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
func PushRedisListWithExpire(key string, value string, expire time.Duration) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.RPush(ctx, key, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// delete redis key
|
||||||
|
func delRedis(key string) {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.Del(ctx, key).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// User 用户,用于存入 Redis hash
|
||||||
|
type RUser struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Age int `json:"age"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u *RUser) toJSONString() string {
|
||||||
|
// 将User对象编码为JSON字符串
|
||||||
|
userJSON, err := json.Marshal(u)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Failed to marshal user: %v", err)
|
||||||
|
}
|
||||||
|
return string(userJSON)
|
||||||
|
}
|
||||||
|
|
||||||
|
// put hash to redis
|
||||||
|
func hSetRedis(key string, field string, value string) {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.HSet(ctx, key, field, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get hash from redis
|
||||||
|
func hGetRedis(key string, field string) string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.HGet(ctx, key, field).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置set，有过期时间
|
||||||
|
func SetRedisSet(key string, values []string, expire time.Duration) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SAdd(ctx, key, values).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置set,添加元素
|
||||||
|
func SetRedisSetAdd(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SAdd(ctx, key, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 批量添加元素
|
||||||
|
func SetRedisSetAddBatchWithExpire(key string, values []string, expire time.Duration) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SAdd(ctx, key, values).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetAddBatchWithExpire Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetAddBatchWithExpire Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置set,添加元素
|
||||||
|
func SetRedisSetAddWithExpire(key string, value string, expire time.Duration) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SAdd(ctx, key, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetAddWithExpire Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, key, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetAddWithExpire Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置set,删除元素
|
||||||
|
func SetRedisSetRemove(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SRem(ctx, key, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取两个set的交集
|
||||||
|
func GetRedisSetIntersect(key1 string, key2 string) []string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.SInter(ctx, key1, key2).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查看set是否包含元素
|
||||||
|
func IsContainSet(key string, value string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.SIsMember(ctx, key, value).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查看set的所有元素
|
||||||
|
func GetRedisSetMembers(key string) []string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.SMembers(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// BITMAP
|
||||||
|
func SetRedisBitmap(key string, offset int64, value int) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SetBit(ctx, key, offset, value).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// BITMAP获取
|
||||||
|
func GetRedisBitmap(key string, offset int64) int {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.GetBit(ctx, key, offset).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return int(val)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发布订阅者模式-发布消息
|
||||||
|
func Publish(channel string, message string, expire time.Duration) {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.Publish(ctx, channel, message).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error publishing message: %v", err)
|
||||||
|
}
|
||||||
|
err = RedisClient.Expire(ctx, channel, expire).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发布订阅者模式-订阅消息
|
||||||
|
func Subscribe(channel string) []string {
|
||||||
|
ctx := context.Background()
|
||||||
|
pubsub := RedisClient.Subscribe(ctx, channel)
|
||||||
|
ch := pubsub.Channel()
|
||||||
|
defer pubsub.Close()
|
||||||
|
var messages []string
|
||||||
|
for msg := range ch {
|
||||||
|
messages = append(messages, msg.Payload)
|
||||||
|
}
|
||||||
|
return messages
|
||||||
|
}
|
||||||
|
|
||||||
|
// redis两个set求差集存入第一个set
|
||||||
|
func SetRedisSetDiffAndStore(key1 string, key2 string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SDiffStore(ctx, key1, key1, key2).Err() //将key1和key2的差集存入key1
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetDiffAndStore Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// redis将第二个set存入第一个set
|
||||||
|
func SetRedisSetUnionAndStore(key1 string, key2 string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.SUnionStore(ctx, key1, key1, key2).Err() //将key1和key2的并集存入key1
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("SetRedisSetUnionAndStore Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// redis 清空set
|
||||||
|
func ClearRedisSet(key string) bool {
|
||||||
|
ctx := context.Background()
|
||||||
|
err := RedisClient.Del(ctx, key).Err()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error setting key: %v", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取两个集合的并集
|
||||||
|
func GetRedisSetUnion(key1 string, key2 string) []string {
|
||||||
|
ctx := context.Background()
|
||||||
|
val, err := RedisClient.SUnion(ctx, key1, key2).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
type RedisInfo struct {
|
||||||
|
Key string
|
||||||
|
Value string
|
||||||
|
Type string
|
||||||
|
Expire int // 过期时间, 单位: 秒
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取所有的key和value,及其对应的过期时间
|
||||||
|
func GetAllRedisInfo() ([]RedisInfo, error) {
|
||||||
|
ctx := context.Background()
|
||||||
|
keys, err := RedisClient.Keys(ctx, "*").Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var redisInfos []RedisInfo
|
||||||
|
for _, key := range keys {
|
||||||
|
//先查看key类型，再根据类型获取value
|
||||||
|
key_type, val, err := getKeyTypeAndData(key)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
expire, err := RedisClient.TTL(ctx, key).Result()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error getting key: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
redisInfo := RedisInfo{
|
||||||
|
Key: key,
|
||||||
|
Value: val,
|
||||||
|
Type: key_type,
|
||||||
|
Expire: int(expire.Seconds()),
|
||||||
|
}
|
||||||
|
redisInfos = append(redisInfos, redisInfo)
|
||||||
|
}
|
||||||
|
return redisInfos, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getKeyTypeAndData(key string) (string, string, error) {
|
||||||
|
ctx := context.Background()
|
||||||
|
key_type := RedisClient.Type(ctx, key).Val()
|
||||||
|
var val interface{}
|
||||||
|
var err error
|
||||||
|
switch key_type {
|
||||||
|
case "string":
|
||||||
|
val, err = RedisClient.Get(ctx, key).Result()
|
||||||
|
case "hash":
|
||||||
|
val, err = RedisClient.HGetAll(ctx, key).Result()
|
||||||
|
case "list":
|
||||||
|
val, err = RedisClient.LRange(ctx, key, 0, -1).Result()
|
||||||
|
case "set":
|
||||||
|
val, err = RedisClient.SMembers(ctx, key).Result()
|
||||||
|
case "zset":
|
||||||
|
val, err = RedisClient.ZRange(ctx, key, 0, -1).Result()
|
||||||
|
case "bitmap":
|
||||||
|
val, err = RedisClient.GetBit(ctx, key, 0).Result()
|
||||||
|
default:
|
||||||
|
val = "unknown type"
|
||||||
|
}
|
||||||
|
return key_type, fmt.Sprintf("%v", val), err
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue