OBS源码学习（六）-将当前视频画面推流到两个不同的RTMP服务器上

一、OBS本身的推流方法
在OBS设置好推流地址后保存并应用地址
点击保存之后会调用SaveStream1Settings函数

void OBSBasicSettings::SaveStream1Settings()
{
	bool customServer = IsCustomService();
	const char *service_id = customServer ? "rtmp_custom" : "rtmp_common";

	obs_service_t *oldService = main->GetService();
	OBSData hotkeyData = obs_hotkeys_save_service(oldService);
	obs_data_release(hotkeyData);

	OBSData settings = obs_data_create();
	obs_data_release(settings);

	if (!customServer) {
		obs_data_set_string(settings, "service",
				    QT_TO_UTF8(ui->service->currentText()));
		obs_data_set_string(
			settings, "server",
			QT_TO_UTF8(ui->server->currentData().toString()));
	} else {
		obs_data_set_string(settings, "server",
				    QT_TO_UTF8(ui->customServer->text()));
		obs_data_set_bool(settings, "use_auth",
				  ui->useAuth->isChecked());
		if (ui->useAuth->isChecked()) {
			obs_data_set_string(
				settings, "username",
				QT_TO_UTF8(ui->authUsername->text()));
			obs_data_set_string(settings, "password",
					    QT_TO_UTF8(ui->authPw->text()));
		}
	}

	obs_data_set_bool(settings, "bwtest",
			  ui->bandwidthTestEnable->isChecked());

	if (!!auth && strcmp(auth->service(), "Twitch") == 0) {
		bool choiceExists = config_has_user_value(
			main->Config(), "Twitch", "AddonChoice");
		int currentChoice =
			config_get_int(main->Config(), "Twitch", "AddonChoice");
		int newChoice = ui->twitchAddonDropdown->currentIndex();

		config_set_int(main->Config(), "Twitch", "AddonChoice",
			       newChoice);

		if (choiceExists && currentChoice != newChoice)
			forceAuthReload = true;
	}
	if (!!auth && strcmp(auth->service(), "Mixer") == 0) {
		bool choiceExists = config_has_user_value(
			main->Config(), "Mixer", "AddonChoice");
		int currentChoice =
			config_get_int(main->Config(), "Mixer", "AddonChoice");
		int newChoice = ui->mixerAddonDropdown->currentIndex();

		config_set_int(main->Config(), "Mixer", "AddonChoice",
			       newChoice);

		if (choiceExists && currentChoice != newChoice)
			forceAuthReload = true;
	}

	obs_data_set_string(settings, "key", QT_TO_UTF8(ui->key->text()));

	OBSService newService = obs_service_create(
		service_id, "default_service", settings, hotkeyData);
	obs_service_release(newService);

	if (!newService)
		return;

	main->SetService(newService);
	main->SaveService();
	main->auth = auth;
	if (!!main->auth)
		main->auth->LoadUI();
}

然后会调用SetService函数将newService保存在全局变量service中

void OBSBasic::SetService(obs_service_t *newService)
{
	if (newService)
		service = newService;
}

最后调用SaveService函数将推流地址保存在service.josn中

void OBSBasic::SaveService()
{
	if (!service)
		return;

	char serviceJsonPath[512];
	int ret = GetProfilePath(serviceJsonPath, sizeof(serviceJsonPath),
				 SERVICE_PATH);
	if (ret <= 0)
		return;

	obs_data_t *data = obs_data_create();
	obs_data_t *settings = obs_service_get_settings(service);

	obs_data_set_string(data, "type", obs_service_get_type(service));
	obs_data_set_obj(data, "settings", settings);

	if (!obs_data_save_json_safe(data, serviceJsonPath, "tmp", "bak"))
		blog(LOG_WARNING, "Failed to save service");

	obs_data_release(settings);
	obs_data_release(data);
}

当点击推流时调用StartStreaming函数

void OBSBasic::StartStreaming()
{
	if (outputHandler->StreamingActive())
		return;
	if (disableOutputsRef)
		return;

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTING);

	SaveProject();

	ui->streamButton->setEnabled(false);
	ui->streamButton->setChecked(false);
	ui->streamButton->setText(QTStr("Basic.Main.Connecting"));

	if (sysTrayStream) {
		sysTrayStream->setEnabled(false);
		sysTrayStream->setText(ui->streamButton->text());
	}

	if (!outputHandler->StartStreaming(service)) {
		QString message =
			!outputHandler->lastError.empty()
				? QTStr(outputHandler->lastError.c_str())
				: QTStr("Output.StartFailedGeneric");
		ui->streamButton->setText(QTStr("Basic.Main.StartStreaming"));
		ui->streamButton->setEnabled(true);
		ui->streamButton->setChecked(false);

		if (sysTrayStream) {
			sysTrayStream->setText(ui->streamButton->text());
			sysTrayStream->setEnabled(true);
		}

		QMessageBox::critical(this, QTStr("Output.StartStreamFailed"),
				      message);
		return;
	}

	bool recordWhenStreaming = config_get_bool(
		GetGlobalConfig(), "BasicWindow", "RecordWhenStreaming");
	if (recordWhenStreaming)
		StartRecording();

	bool replayBufferWhileStreaming = config_get_bool(
		GetGlobalConfig(), "BasicWindow", "ReplayBufferWhileStreaming");
	if (replayBufferWhileStreaming)
		StartReplayBuffer();
}

关键函数是outputHandler->StartStreaming(service)；service就是保存的推流地址outputHandler是输出句柄。之后的流程大家可以自己看看。
二、我们公司业务要是我会从公司后台得到两个推流地址，以下函数是为了构造两个OBSService对象

void OBSBasic::setServiceData(QString Url)
{
	//Url = "rtmp://living.test.cbct.cn/live 36353_dev_swqepr?bizid=36353&txSecret=c44ae5d5d9aa0feb46b8c194bb6c7503&txTime=60029539";
	QString strRtmp = Url.section("/", -5, -2);
	QString strkey = Url.section("/", -1);
	QByteArray RtmpData = strRtmp.toLatin1();
	QByteArray KeyData = strkey.toLatin1();
	const char *val = RtmpData.data();
	const char *key = KeyData.data();
	const char *service_id = "rtmp_custom";
	qDebug() << "OBSBasic::setServiceData" << val << key;
	OBSData settings = obs_data_create();
	obs_data_release(settings);

	obs_data_set_string(settings, "server", val);
	obs_data_set_bool(settings, "use_auth", false);
	obs_data_set_string(settings, "key", key);
	OBSService newService = obs_service_create(
		service_id, "default_service", settings, NULL);
	obs_service_release(newService);

	if (!newService)
		return;

	SetService(newService);
	SaveService();
}

void OBSBasic::setService1Data(QString Url)
{
	QString strRtmp = Url.section("/", -5, -2);
	QString strkey = Url.section("/", -1);
	QByteArray RtmpData = strRtmp.toLatin1();
	QByteArray KeyData = strkey.toLatin1();
	const char *val = RtmpData.data();
	const char *key = KeyData.data();
	const char *service_id = "rtmp_custom";
	qDebug() << "OBSBasic::setService1Data" << val << key;

	OBSData settings = obs_data_create();
	obs_data_release(settings);

	obs_data_set_string(settings, "server", val);
	obs_data_set_bool(settings, "use_auth", false);
	obs_data_set_string(settings, "key", key);

	OBSService newService = obs_service_create(
		service_id, "default_service", settings, NULL);
	obs_service_release(newService);

	if (!newService)
		return;

	SetService1(newService);
}

setServiceData和setServiceData1分别构造了OBSService对象service和service1，现在已经有了两个OBSService对象，然后构造一个输出句柄outputHandler1

void OBSBasic::ResetOutputs()
{
	ProfileScope("OBSBasic::ResetOutputs");

	const char *mode = config_get_string(basicConfig, "Output", "Mode");
	bool advOut = astrcmpi(mode, "Advanced") == 0;

	if (!outputHandler || !outputHandler->Active()) {
		outputHandler.reset();
		outputHandler.reset(advOut ? CreateAdvancedOutputHandler(this)
					   : CreateSimpleOutputHandler(this));

		outputHandler1.reset();
		outputHandler1.reset(advOut ? CreateAdvancedOutputHandler(this)
					    : CreateSimpleOutputHandler(this));

最后写一个推流函数StartStreaming1（）

void OBSBasic::StartStreaming1()
{
	if (outputHandler1->StreamingActive())
		return;
	if (disableOutputsRef)
		return;

	if (api)
		api->on_event(OBS_FRONTEND_EVENT_STREAMING_STARTING);

	SaveProject();

	//ui->BtnStartTimeLive->setEnabled(false);
	//ui->BtnStartTimeLive->setChecked(false);
	//ui->BtnStartTimeLive->setText(QTStr("Basic.Main.Connecting"));

	if (sysTrayStream) {
		sysTrayStream->setEnabled(false);
		sysTrayStream->setText(ui->BtnStartTimeLive->text());
	}

	if (!outputHandler1->StartStreaming(service1, true)) {
		QString message =
			!outputHandler1->lastError.empty()
				? QTStr(outputHandler1->lastError.c_str())
				: QTStr("Output.StartFailedGeneric");
		ui->BtnStartTimeLive->setText(
			QTStr("Basic.Main.StartStreaming"));
		ui->BtnStartTimeLive->setEnabled(true);
		ui->BtnStartTimeLive->setChecked(false);

		if (sysTrayStream) {
			sysTrayStream->setText(ui->streamButton->text());
			sysTrayStream->setEnabled(true);
		}

		QMessageBox::critical(this, QTStr("Output.StartStreamFailed"),
				      message);
		return;
	}

	bool recordWhenStreaming = config_get_bool(
		GetGlobalConfig(), "BasicWindow", "RecordWhenStreaming");
	if (recordWhenStreaming)
		StartRecording();

	bool replayBufferWhileStreaming = config_get_bool(
		GetGlobalConfig(), "BasicWindow", "ReplayBufferWhileStreaming");
	if (replayBufferWhileStreaming)
		StartReplayBuffer();
}

之后的逻辑都是和原来推流一样的。

代码写的有点乱，又看不懂的同学可以私信我。