plugin-onvif/device_list.go

package onvif

import (
	"strings"

	. "github.com/Monibuca/engine/v3"
	rtsp "github.com/Monibuca/plugin-rtsp/v3"
	. "github.com/Monibuca/utils/v3"
	"github.com/aler9/gortsplib"
	lonvif "github.com/liyanhui1998/go-onvif"
)

type DeviceList struct {
	Data map[string]map[string]*DeviceStatus
}

func (dl *DeviceList) discoveryDevice() {
	for _, i := range config.Interfaces {
		deviceParams := WsDiscover(i.InterfaceName, authCfg)

		devsMap, ok := dl.Data[i.InterfaceName]
		if !ok {
			devsMap = make(map[string]*DeviceStatus)
			dl.Data[i.InterfaceName] = devsMap
		}

		for _, dParam := range deviceParams {
			// 如果已经存在，则不再添加
			if _, ok := devsMap[dParam.Ipddr]; ok {
				continue
			}
			var dev *lonvif.Device
			devStatus := &DeviceStatus{}
			dev, err := lonvif.NewDevice(dParam)
			if err == nil {
				devStatus.Device = dev
				devStatus.Status = StatusInitOk
			} else {
				devStatus.Status = StatusInitError
				devStatus.Device = &lonvif.Device{
					Params: dParam,
				}
			}
			devsMap[dParam.Ipddr] = devStatus
		}
	}
}

func (dl *DeviceList) pullStream() {
	for _, devicesMap := range dl.Data {
		for _, d := range devicesMap {
			streamPath := strings.ReplaceAll(d.Device.Params.Ipddr, ".", "_")
			streamPath = "onvif/" + strings.ReplaceAll(streamPath, ":", "_")
			//避免重复拉流
			if FindStream(streamPath) != nil {
				continue
			}
			rtspUrl, err := GetStreamUri(d.Device)
			if err != nil {
				Printf("[ONVIF] get stream err:", err)
				d.Status = StatusGetStreamUriError
				continue
			}
			d.Status = StatusGetStreamUriOk
			go func(targetURL string, streamPath string, d *DeviceStatus) {
				err := (&rtsp.RTSPClient{Transport: gortsplib.TransportTCP}).PullStream(streamPath, targetURL)
				if err == nil {
					d.Status = StatusPullRtspOk
				} else {
					d.Status = StatusPullRtspError
				}
			}(rtspUrl, streamPath, d)
		}
	}
}