Add performance monitoring API

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Features
+
+- Add performance monitoring API ([#470](https://github.com/getsentry/sentry-unreal/pull/470))
+
 ### Fixes
 
 - Fix Linux intermediates paths in `FilterPlugin.ini` ([#468](https://github.com/getsentry/sentry-unreal/pull/468))

plugin-dev/Source/Sentry/Private/Android/Infrastructure/SentryConvertorsAndroid.cpp

@@ -6,10 +6,14 @@
 
 #include "SentryScope.h"
 #include "SentryId.h"
+#include "SentryTransaction.h"
+#include "SentrySpan.h"
 #include "SentryDefines.h"
 
 #include "Android/SentryScopeAndroid.h"
 #include "Android/SentryIdAndroid.h"
+#include "Android/SentryTransactionAndroid.h"
+#include "Android/SentrySpanAndroid.h"
 
 #include "Android/AndroidApplication.h"
 #include "Android/AndroidJavaEnv.h"
@@ -149,6 +153,22 @@ USentryId* SentryConvertorsAndroid::SentryIdToUnreal(jobject id)
 	return unrealId;
 }
 
+USentryTransaction* SentryConvertorsAndroid::SentryTransactionToUnreal(jobject transaction)
+{
+	TSharedPtr<SentryTransactionAndroid> transactionNativeImpl = MakeShareable(new SentryTransactionAndroid(transaction));
+	USentryTransaction* unrealTransaction = NewObject<USentryTransaction>();
+	unrealTransaction->InitWithNativeImpl(transactionNativeImpl);
+	return unrealTransaction;
+}
+
+USentrySpan* SentryConvertorsAndroid::SentrySpanToUnreal(jobject span)
+{
+	TSharedPtr<SentrySpanAndroid> spanNativeImpl = MakeShareable(new SentrySpanAndroid(span));
+	USentrySpan* unrealSpan = NewObject<USentrySpan>();
+	unrealSpan->InitWithNativeImpl(spanNativeImpl);
+	return unrealSpan;
+}
+
 TMap<FString, FString> SentryConvertorsAndroid::StringMapToUnreal(jobject map)
 {
 	TMap<FString, FString> result;

plugin-dev/Source/Sentry/Private/Android/Infrastructure/SentryConvertorsAndroid.h

@@ -8,6 +8,8 @@
 
 class USentryScope;
 class USentryId;
+class USentryTransaction;
+class USentrySpan;
 class FSentryJavaObjectWrapper;
 class FJsonValue;
 
@@ -24,6 +26,8 @@ class SentryConvertorsAndroid
 	static ESentryLevel SentryLevelToUnreal(jobject level);
 	static USentryScope* SentryScopeToUnreal(jobject scope);
 	static USentryId* SentryIdToUnreal(jobject id);
+	static USentryTransaction* SentryTransactionToUnreal(jobject transaction);
+	static USentrySpan* SentrySpanToUnreal(jobject span);
 	static TMap<FString, FString> StringMapToUnreal(jobject stringMap);
 	static TArray<FString> StringListToUnreal(jobject stringList);
 	static TArray<uint8> ByteArrayToUnreal(jbyteArray byteArray);

plugin-dev/Source/Sentry/Private/Android/Infrastructure/SentryJavaClasses.cpp

@@ -15,6 +15,8 @@ const FSentryJavaClass SentryJavaClasses::UserFeedback		= FSentryJavaClass { "io
 const FSentryJavaClass SentryJavaClasses::Message			= FSentryJavaClass { "io/sentry/protocol/Message", ESentryJavaClassType::External };
 const FSentryJavaClass SentryJavaClasses::SentryLevel		= FSentryJavaClass { "io/sentry/SentryLevel", ESentryJavaClassType::External };
 const FSentryJavaClass SentryJavaClasses::SentryHint		= FSentryJavaClass { "io/sentry/Hint", ESentryJavaClassType::External };
+const FSentryJavaClass SentryJavaClasses::Transaction		= FSentryJavaClass { "io/sentry/ITransaction", ESentryJavaClassType::External };
+const FSentryJavaClass SentryJavaClasses::Span				= FSentryJavaClass { "io/sentry/ISpan", ESentryJavaClassType::External };
 
 // System Java classes definitions
 const FSentryJavaClass SentryJavaClasses::ArrayList			= FSentryJavaClass { "java/util/ArrayList", ESentryJavaClassType::System };

plugin-dev/Source/Sentry/Private/Android/Infrastructure/SentryJavaClasses.h

@@ -19,6 +19,8 @@ struct SentryJavaClasses
 	const static FSentryJavaClass Message;
 	const static FSentryJavaClass SentryLevel;
 	const static FSentryJavaClass SentryHint;
+	const static FSentryJavaClass Transaction;
+	const static FSentryJavaClass Span;
 
 	// System Java classes
 	const static FSentryJavaClass ArrayList;

plugin-dev/Source/Sentry/Private/Android/Java/SentryBridgeJava.java

@@ -17,6 +17,7 @@
 import io.sentry.Breadcrumb;
 import io.sentry.Hint;
 import io.sentry.IHub;
+import io.sentry.SamplingContext;
 import io.sentry.IScope;
 import io.sentry.ScopeCallback;
 import io.sentry.Sentry;
@@ -30,6 +31,7 @@
 public class SentryBridgeJava {
 	public static native void onConfigureScope(long callbackAddr, IScope scope);
 	public static native SentryEvent onBeforeSend(long handlerAddr, SentryEvent event, Hint hint);
+	public static native Double onTracesSampler(long samplerAddr, SamplingContext samplingContext);
 
 	public static void init(Activity activity, final String settingsJsonStr, final long beforeSendHandler) {
 		SentryAndroid.init(activity, new Sentry.OptionsConfiguration<SentryAndroidOptions>() {
@@ -62,6 +64,19 @@ public SentryEvent execute(SentryEvent event, Hint hint) {
 					for (int i = 0; i < Excludes.length(); i++) {
 						options.addInAppExclude(Excludes.getString(i));
 					}
+					options.setEnableTracing(settingJson.getBoolean("enableTracing"));
+					if(settingJson.has("tracesSampleRate")) {
+						options.setTracesSampleRate(settingJson.getDouble("tracesSampleRate"));
+					}
+					if(settingJson.has("tracesSampler")) {
+						final long samplerAddr = settingJson.getLong("tracesSampler");
+						options.setTracesSampler(new SentryOptions.TracesSamplerCallback() {
+							@Override
+							public Double sample(SamplingContext samplingContext) {
+								return onTracesSampler(samplerAddr, samplingContext);
+							}
+						});
+					}
 				} catch (JSONException e) {
 					throw new RuntimeException(e);
 				}

plugin-dev/Source/Sentry/Private/Android/Jni/SentryJniAndroid.cpp

@@ -31,4 +31,9 @@ JNI_METHOD jobject Java_io_sentry_unreal_SentryBridgeJava_onBeforeSend(JNIEnv* e
 	HintToProcess->InitWithNativeImpl(MakeShareable(new SentryHintAndroid(hint)));
 
 	return handler->HandleBeforeSend(EventToProcess, HintToProcess) ? event : nullptr;
+}
+
+JNI_METHOD jobject Java_io_sentry_unreal_SentryBridgeJava_onTracesSampler(JNIEnv* env, jclass clazz, jlong objAddr, jobject samplingContext)
+{
+	return nullptr;
 }

plugin-dev/Source/Sentry/Private/Android/SentrySpanAndroid.cpp

@@ -0,0 +1,50 @@
+// Copyright (c) 2023 Sentry. All Rights Reserved.
+
+#include "SentrySpanAndroid.h"
+
+#include "Infrastructure/SentryConvertorsAndroid.h"
+#include "Infrastructure/SentryJavaClasses.h"
+
+SentrySpanAndroid::SentrySpanAndroid(jobject span)
+	: FSentryJavaObjectWrapper(SentryJavaClasses::Span, span)
+{
+	SetupClassMethods();
+}
+
+void SentrySpanAndroid::SetupClassMethods()
+{
+	FinishMethod = GetMethod("finish", "()V");
+	IsFinishedMethod = GetMethod("isFinished", "()Z");
+	SetTagMethod = GetMethod("setTag", "(Ljava/lang/String;Ljava/lang/String;)V");
+	SetDataMethod = GetMethod("setData", "(Ljava/lang/String;Ljava/lang/Object;)V");
+}
+
+void SentrySpanAndroid::Finish()
+{
+	CallMethod<void>(FinishMethod);
+}
+
+bool SentrySpanAndroid::IsFinished() const
+{
+	return false;
+}
+
+void SentrySpanAndroid::SetTag(const FString& key, const FString& value)
+{
+	CallMethod<void>(SetTagMethod, *GetJString(key), *GetJString(value));
+}
+
+void SentrySpanAndroid::RemoveTag(const FString& key)
+{
+	SetTag(key, TEXT(""));
+}
+
+void SentrySpanAndroid::SetData(const FString& key, const TMap<FString, FString>& values)
+{
+	CallMethod<void>(SetDataMethod, *GetJString(key), SentryConvertorsAndroid::StringMapToNative(values)->GetJObject());
+}
+
+void SentrySpanAndroid::RemoveData(const FString& key)
+{
+	SetData(key, TMap<FString, FString>());
+}

plugin-dev/Source/Sentry/Private/Android/SentrySpanAndroid.h

@@ -0,0 +1,29 @@
+// Copyright (c) 2023 Sentry. All Rights Reserved.
+
+#pragma once
+
+#include "Interface/SentrySpanInterface.h"
+
+#include "Infrastructure/SentryJavaObjectWrapper.h"
+
+class SentrySpanAndroid : public ISentrySpan, public FSentryJavaObjectWrapper
+{
+public:
+	SentrySpanAndroid(jobject span);
+
+	void SetupClassMethods();
+
+	virtual void Finish() override;
+	virtual bool IsFinished() const override;
+	virtual void SetTag(const FString& key, const FString& value) override;
+	virtual void RemoveTag(const FString& key) override;
+	virtual void SetData(const FString& key, const TMap<FString, FString>& values) override;
+	virtual void RemoveData(const FString& key) override;
+
+
+private:
+	FSentryJavaMethod FinishMethod;
+	FSentryJavaMethod IsFinishedMethod;
+	FSentryJavaMethod SetTagMethod;
+	FSentryJavaMethod SetDataMethod;
+};

plugin-dev/Source/Sentry/Private/Android/SentrySubsystemAndroid.cpp

@@ -44,6 +44,15 @@ void SentrySubsystemAndroid::InitWithSettings(const USentrySettings* settings, U
 	SettingsJson->SetArrayField(TEXT("inAppInclude"), SentryConvertorsAndroid::StrinArrayToJsonArray(settings->InAppInclude));
 	SettingsJson->SetArrayField(TEXT("inAppExclude"), SentryConvertorsAndroid::StrinArrayToJsonArray(settings->InAppExclude));
 	SettingsJson->SetBoolField(TEXT("sendDefaultPii"), settings->SendDefaultPii);
+	SettingsJson->SetBoolField(TEXT("enableTracing"), settings->EnableTracing);
+	if(settings->EnableTracing && settings->SamplingType == ESentryTracesSamplingType::UniformSampleRate)
+	{
+		SettingsJson->SetNumberField(TEXT("tracesSampleRate"), settings->TracesSampleRate);
+	}
+	if(settings->EnableTracing && settings->SamplingType == ESentryTracesSamplingType::TracesSampler)
+	{
+		SettingsJson->SetNumberField(TEXT("tracesSampler"), (jlong)0);
+	}
 
 	FString SettingsJsonStr;
 	TSharedRef<TJsonWriter<>> JsonWriter = TJsonWriterFactory<>::Create(&SettingsJsonStr);
@@ -184,3 +193,11 @@ void SentrySubsystemAndroid::EndSession()
 {
 	FSentryJavaObjectWrapper::CallStaticMethod<void>(SentryJavaClasses::SentryBridgeJava, "endSession", "()V", nullptr);
 }
+
+USentryTransaction* SentrySubsystemAndroid::StartTransaction(const FString& name, const FString& operation)
+{
+	auto transaction = FSentryJavaObjectWrapper::CallStaticObjectMethod<jobject>(SentryJavaClasses::Sentry, "startTransaction", "(Ljava/lang/String;Ljava/lang/String;)Lio/sentry/ITransaction;",
+		*FSentryJavaObjectWrapper::GetJString(name), *FSentryJavaObjectWrapper::GetJString(operation));
+
+	return SentryConvertorsAndroid::SentryTransactionToUnreal(*transaction);
+}

plugin-dev/Source/Sentry/Private/Android/SentrySubsystemAndroid.h

@@ -26,4 +26,5 @@ class SentrySubsystemAndroid : public ISentrySubsystem
 	virtual void SetLevel(ESentryLevel level) override;
 	virtual void StartSession() override;
 	virtual void EndSession() override;
+	virtual USentryTransaction* StartTransaction(const FString& name, const FString& operation) override;
 };

plugin-dev/Source/Sentry/Private/Android/SentryTransactionAndroid.cpp

@@ -0,0 +1,63 @@
+// Copyright (c) 2023 Sentry. All Rights Reserved.
+
+#include "SentryTransactionAndroid.h"
+
+#include "Infrastructure/SentryConvertorsAndroid.h"
+#include "Infrastructure/SentryJavaClasses.h"
+
+SentryTransactionAndroid::SentryTransactionAndroid(jobject transaction)
+	: FSentryJavaObjectWrapper(SentryJavaClasses::Transaction, transaction)
+{
+	SetupClassMethods();
+}
+
+void SentryTransactionAndroid::SetupClassMethods()
+{
+	StartChildMethod = GetMethod("startChild", "(Ljava/lang/String;Ljava/lang/String;)Lio/sentry/ISpan;");
+	FinishMethod = GetMethod("finish", "()V");
+	IsFinishedMethod = GetMethod("isFinished", "()Z");
+	SetNameMethod = GetMethod("setName", "(Ljava/lang/String;)V");
+	SetTagMethod = GetMethod("setTag", "(Ljava/lang/String;Ljava/lang/String;)V");
+	SetDataMethod = GetMethod("setData", "(Ljava/lang/String;Ljava/lang/Object;)V");
+}
+
+USentrySpan* SentryTransactionAndroid::StartChild(const FString& operation, const FString& desctiption)
+{
+	auto span = CallObjectMethod<jobject>(StartChildMethod, *GetJString(operation), *GetJString(desctiption));
+	return SentryConvertorsAndroid::SentrySpanToUnreal(*span);
+}
+
+void SentryTransactionAndroid::Finish()
+{
+	CallMethod<void>(FinishMethod);
+}
+
+bool SentryTransactionAndroid::IsFinished() const
+{
+	return CallMethod<bool>(IsFinishedMethod);;
+}
+
+void SentryTransactionAndroid::SetName(const FString& name)
+{
+	CallMethod<void>(SetNameMethod, *GetJString(name));
+}
+
+void SentryTransactionAndroid::SetTag(const FString& key, const FString& value)
+{
+	CallMethod<void>(SetTagMethod, *GetJString(key), *GetJString(value));
+}
+
+void SentryTransactionAndroid::RemoveTag(const FString& key)
+{
+	SetTag(key, TEXT(""));
+}
+
+void SentryTransactionAndroid::SetData(const FString& key, const TMap<FString, FString>& values)
+{
+	CallMethod<void>(SetDataMethod, *GetJString(key), SentryConvertorsAndroid::StringMapToNative(values)->GetJObject());
+}
+
+void SentryTransactionAndroid::RemoveData(const FString& key)
+{
+	SetData(key, TMap<FString, FString>());
+}

plugin-dev/Source/Sentry/Private/Android/SentryTransactionAndroid.h

@@ -0,0 +1,32 @@
+// Copyright (c) 2023 Sentry. All Rights Reserved.
+
+#pragma once
+
+#include "Interface/SentryTransactionInterface.h"
+
+#include "Infrastructure/SentryJavaObjectWrapper.h"
+
+class SentryTransactionAndroid : public ISentryTransaction, public FSentryJavaObjectWrapper
+{
+public:
+	SentryTransactionAndroid(jobject transaction);
+
+	void SetupClassMethods();
+
+	virtual USentrySpan* StartChild(const FString& operation, const FString& desctiption) override;
+	virtual void Finish() override;
+	virtual bool IsFinished() const override;
+	virtual void SetName(const FString& name) override;
+	virtual void SetTag(const FString& key, const FString& value) override;
+	virtual void RemoveTag(const FString& key) override;
+	virtual void SetData(const FString& key, const TMap<FString, FString>& values) override;
+	virtual void RemoveData(const FString& key) override;
+
+private:
+	FSentryJavaMethod StartChildMethod;
+	FSentryJavaMethod FinishMethod;
+	FSentryJavaMethod IsFinishedMethod;
+	FSentryJavaMethod SetNameMethod;
+	FSentryJavaMethod SetTagMethod;
+	FSentryJavaMethod SetDataMethod;
+};

plugin-dev/Source/Sentry/Private/Apple/Infrastructure/SentryConvertorsApple.cpp

@@ -3,10 +3,14 @@
 #include "SentryConvertorsApple.h"
 #include "SentryScope.h"
 #include "SentryId.h"
+#include "SentryTransaction.h"
+#include "SentrySpan.h"
 #include "SentryDefines.h"
 
 #include "Apple/SentryScopeApple.h"
 #include "Apple/SentryIdApple.h"
+#include "Apple/SentryTransactionApple.h"
+#include "Apple/SentrySpanApple.h"
 
 SentryLevel SentryConvertorsApple::SentryLevelToNative(ESentryLevel level)
 {
@@ -147,6 +151,22 @@ USentryId* SentryConvertorsApple::SentryIdToUnreal(SentryId* id)
 	return unrealId;
 }
 
+USentryTransaction* SentryConvertorsApple::SentryTransactionToUnreal(id<SentrySpan> transaction)
+{
+	TSharedPtr<SentryTransactionApple> transactionNativeImpl = MakeShareable(new SentryTransactionApple(transaction));
+	USentryTransaction* unrealTransaction = NewObject<USentryTransaction>();
+	unrealTransaction->InitWithNativeImpl(transactionNativeImpl);
+	return unrealTransaction;
+}
+
+USentrySpan* SentryConvertorsApple::SentrySpanToUnreal(id<SentrySpan> span)
+{
+	TSharedPtr<SentrySpanApple> spanNativeImpl = MakeShareable(new SentrySpanApple(span));
+	USentrySpan* unrealSpan = NewObject<USentrySpan>();
+	unrealSpan->InitWithNativeImpl(spanNativeImpl);
+	return unrealSpan;
+}
+
 SentryLevel SentryConvertorsApple::StringToSentryLevel(NSString* string)
 {
 	SentryLevel nativeLevel = kSentryLevelDebug;

plugin-dev/Source/Sentry/Private/Apple/Infrastructure/SentryConvertorsApple.h

@@ -8,6 +8,7 @@
 
 class USentryScope;
 class USentryId;
+class USentryTransaction;
 
 class SentryConvertorsApple
 {
@@ -25,6 +26,8 @@ class SentryConvertorsApple
 	static TArray<uint8> ByteDataToUnreal(NSData* data);
 	static USentryScope* SentryScopeToUnreal(SentryScope* scope);
 	static USentryId* SentryIdToUnreal(SentryId* id);
+	static USentryTransaction* SentryTransactionToUnreal(id<SentrySpan> transaction);
+	static USentrySpan* SentrySpanToUnreal(id<SentrySpan> span);
 
 	/** Other conversions */
 	static SentryLevel StringToSentryLevel(NSString* string);