Add more config options for disk buffering (signalfx#62)

Author: mhennoch

ios/SpanDb.swift

@@ -34,9 +34,13 @@ class SpanDb {
     private var sizeStmt_: OpaquePointer?
     private var initialized: Bool = false
 
-    init(path: String? = nil) {
-        self.databasePath = path ?? SpanDb.makeDatabasePath() ?? ":memory:"
-
+    init(enableDiskBuffering: Bool, path: String? = nil) {
+        if enableDiskBuffering {
+            self.databasePath = path ?? SpanDb.makeDatabasePath() ?? ":memory:"
+        } else {
+            self.databasePath = ":memory:"
+        }
+        
         var status = sqlite3_open(databasePath, &db_)
 
         if status != SQLITE_OK {

ios/SpanToDiskExporter.swift

@@ -42,9 +42,9 @@ class SpanToDiskExporter : SpanExporter {
     private var totalSpansInserted: Int64 = 0
     private var checkpointCounter: Int64 = 0
 
-    init(spanDb: SpanDb, maxFileSizeBytes: Int64 = 25 * 1024 * 1024, truncationCheckpoint: Int64 = 512) {
+    init(spanDb: SpanDb, limitDiskUsageMegabytes: Int64, truncationCheckpoint: Int64) {
         self.db = spanDb
-        self.maxFileSizeBytes = maxFileSizeBytes
+        self.maxFileSizeBytes = limitDiskUsageMegabytes * 1024 * 1024
         self.truncationCheckpoint = truncationCheckpoint
     }
 

ios/SplunkOtelReactNative.swift

@@ -21,8 +21,12 @@ class SplunkOtelReactNative: NSObject {
   private var appStartTime = Date()
   @objc(initialize:withResolver:withRejecter:)
   func initialize(config: Dictionary<String, Any>, resolve:RCTPromiseResolveBlock,reject:RCTPromiseRejectBlock) -> Void {
-      let db = SpanDb()
-      spanExporter = SpanToDiskExporter(spanDb: db)
+      let enableDiskBuffering = config["enableDiskBuffering"] as? Bool ?? true 
+      let limitDiskUsageMegabytes = config["limitDiskUsageMegabytes"] as? Int64 ?? 25
+      let truncationCheckpoint = config["truncationCheckpoint"] as? Int64 ?? 512
+        
+      let db = SpanDb(enableDiskBuffering: enableDiskBuffering)
+      spanExporter = SpanToDiskExporter(spanDb: db, limitDiskUsageMegabytes: limitDiskUsageMegabytes, truncationCheckpoint: truncationCheckpoint)
       initializeCrashReporting(exporter: spanExporter)
 
       do {

src/native.ts

@@ -40,6 +40,7 @@ export interface NativeSdKConfiguration {
   globalAttributes?: object;
   enableDiskBuffering?: boolean;
   limitDiskUsageMegabytes?: number;
+  truncationCheckpoint?: number;
 }
 
 export type AppStartInfo = {

src/splunkRum.ts

@@ -52,6 +52,7 @@ export interface ReactNativeConfiguration {
   appStartEnabled?: boolean;
   enableDiskBuffering?: boolean;
   limitDiskUsageMegabytes?: number;
+  truncationCheckpoint?: number;
   bufferTimeout?: number;
   bufferSize?: number;
   debug?: boolean;