109 lines
3.4 KiB
Go
109 lines
3.4 KiB
Go
package host
|
|
|
|
import (
|
|
"easyvqd/internal/conf"
|
|
"easyvqd/pkg/pluginheart"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"git.lnton.com/lnton/pluginsdk/pkg/data"
|
|
"git.lnton.com/lnton/pluginsdk/pkg/plugin"
|
|
)
|
|
|
|
type IFrameResultCB func(string, []byte, int)
|
|
type Core struct {
|
|
Plugin *plugin.PluginSDK // 用于业务GRPC通信
|
|
HttpPlugin *pluginheart.Engine // 用于插件管理HTTP通信
|
|
RtspConfig *ConfigBaseOutput
|
|
CbIFrame IFrameResultCB
|
|
}
|
|
|
|
var StartOk chan struct{}
|
|
|
|
func NewCore(cfg *conf.Bootstrap) *Core {
|
|
cb := plugin.PluginCallback{
|
|
OnShutdown: func() { slog.Info("Plugin shutting down") },
|
|
}
|
|
sdk, err := plugin.NewPlugin(data.Plugin{
|
|
ID: "EasyVQD",
|
|
Name: "EasyVQD",
|
|
Version: "1.0.0",
|
|
ServerAddr: fmt.Sprintf("%s:%d", "localhost", cfg.Plugin.GrpcPort),
|
|
}, cb)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
engine := pluginheart.NewEngine("easyvqd", cfg.Plugin.HttpAPI)
|
|
|
|
// 后台等待启动完成信号，然后触发心跳（不阻塞主流程）
|
|
go func() {
|
|
StartOk = make(chan struct{})
|
|
<-StartOk
|
|
engine.AutoHeart(pluginheart.HeartInput{
|
|
Port: cfg.Server.HTTP.Port,
|
|
}, false)
|
|
}()
|
|
|
|
core := &Core{
|
|
Plugin: sdk,
|
|
HttpPlugin: engine,
|
|
}
|
|
sdk.AddResponseHandler("stop", core.stop)
|
|
sdk.AddResponseHandler("ping", core.ping)
|
|
|
|
// 这部分都是收到响应后的回调
|
|
sdk.AddResponseHandler("findDevices", core.findDevicesRespH)
|
|
sdk.AddResponseHandler("findChannels", core.findChannelsRespH)
|
|
sdk.AddResponseHandler("getBaseConfig", core.getBaseConfigRespH)
|
|
sdk.AddResponseHandler("findTalkUrl", core.findTalkUrlRespH)
|
|
sdk.AddResponseHandler("iframeData", core.iframeDataRespH)
|
|
// 响应主机下发的 "iframeData" 命令
|
|
sdk.AddRequestHandler("iframeData", core.iframeDataReqH)
|
|
// defer sdk.Close()
|
|
return core
|
|
}
|
|
|
|
func (c *Core) iframeDataReqH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
//slog.Debug("Received 'iframeData' from host", "request_id", requestID, "args", args)
|
|
out := IframeOutput{}
|
|
if err := json.Unmarshal(args, &out); err != nil {
|
|
return nil, nil
|
|
}
|
|
c.CbIFrame(out.StreamName, out.Data, out.Codes)
|
|
return nil, nil
|
|
}
|
|
|
|
func (c Core) stop(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Info("Received 'stop' from host", "request_id", requestID)
|
|
return map[string]interface{}{"status": "stopped"}, nil
|
|
}
|
|
|
|
func (c Core) ping(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'ping' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|
|
|
|
func (c Core) findDevicesRespH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'findDeviceList' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|
|
|
|
func (c Core) findChannelsRespH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'findChannels' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|
|
func (c Core) getBaseConfigRespH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'getBaseConfig' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|
|
func (c Core) findTalkUrlRespH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'findTalkUrl' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|
|
func (c Core) iframeDataRespH(requestID string, args json.RawMessage) (interface{}, error) {
|
|
slog.Debug("Received 'iframeData' from host", "request_id", requestID, "args", args)
|
|
return nil, nil
|
|
}
|