Merge pull request #2 from RedTTGMoss/feature/metrics

Feature/metrics

Author: furesoft

src/Moss.NET.Sdk/Core/Extensions.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Diagnostics.Metrics;
 using System.Text.Json;
 using System.Text.Json.Serialization.Metadata;
 using Extism;
@@ -57,4 +58,15 @@ public static dynamic Get(this HoconRoot root, string path)
     {
         return new Options(root.GetObject(path));
     }
+
+    public static void Measure(string name, Action method)
+    {
+        var meter = MossExtension.Instance!.Meter;
+        var methodDuration = meter.CreateHistogram<double>(name + "_duration", "ms", "Method duration in milliseconds");
+
+        var stopwatch = System.Diagnostics.Stopwatch.StartNew();
+        method();
+        stopwatch.Stop();
+        methodDuration.Record(stopwatch.Elapsed.TotalMilliseconds);
+    }
 }

src/Moss.NET.Sdk/Core/Instrumentation/FileMeterListener.cs

@@ -0,0 +1,35 @@
+using System.Collections.Generic;
+
+namespace Moss.NET.Sdk.Core.Instrumentation;
+
+using System;
+using System.Diagnostics.Metrics;
+using System.IO;
+
+internal class FileMeterListener : MeterListener<double>
+{
+    private static string _filePath;
+    private static MeterListener _listener;
+
+    public FileMeterListener(string filePath)
+    {
+        _filePath = filePath;
+
+        if (!File.Exists(_filePath))
+        {
+            File.Create(filePath);
+        }
+    }
+
+    protected override void OnMeasurementRecorded(Instrument instrument, double measurement, ReadOnlySpan<KeyValuePair<string, object?>> tags, object? state)
+    {
+        var tagList = new List<string>();
+        foreach (var tag in tags)
+        {
+            tagList.Add($"{tag.Key}: {tag.Value}");
+        }
+
+        using var writer = new StreamWriter(_filePath, append: true);
+        writer.WriteLine($"{DateTime.UtcNow:O} - {instrument.Name}, Measurement: {measurement}{instrument.Unit}, Tags: {string.Join(", ", tagList)}");
+    }
+}

src/Moss.NET.Sdk/Core/Instrumentation/IMeterListener.cs

@@ -0,0 +1,26 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics.Metrics;
+
+namespace Moss.NET.Sdk.Core.Instrumentation;
+
+public interface IMeterListener
+{
+    internal void Init();
+}
+
+public abstract class MeterListener<T> : IMeterListener
+    where T : struct
+{
+    private readonly MeterListener _listener = new();
+
+    protected abstract void OnMeasurementRecorded(Instrument instrument, T measurement, ReadOnlySpan<KeyValuePair<string, object?>> tags,
+        object? state);
+
+    public void Init()
+    {
+        _listener.InstrumentPublished = (instrument, listener) => listener.EnableMeasurementEvents(instrument);
+        _listener.SetMeasurementEventCallback<T>(OnMeasurementRecorded);
+        _listener.Start();
+    }
+}

src/Moss.NET.Sdk/MossExtension.cs

@@ -1,8 +1,11 @@
 using System;
+using System.Diagnostics.Metrics;
 using System.Runtime.CompilerServices;
 using System.Text.Json;
 using Extism;
 using Hocon;
+using Moss.NET.Sdk.Core;
+using Moss.NET.Sdk.Core.Instrumentation;
 using Moss.NET.Sdk.FFI;
 using Moss.NET.Sdk.Scheduler;
 
@@ -12,6 +15,9 @@ public class MossExtension
 {
     private static MossExtension? _instance;
 
+    public Meter Meter;
+    public IMeterListener MeterListener;
+
     internal static MossExtension? Instance
     {
         get
@@ -38,13 +44,17 @@ public virtual void ExtensionLoop(MossState state)
 
     public static void PreInitExtension()
     {
-        var configSource = System.IO.File.ReadAllText("extension/automation.conf");
-        Config = HoconParser.Parse(configSource);
+        var configPath = "extension/automation.conf";
+
+        if (System.IO.File.Exists(configPath)) {
+            var configSource = System.IO.File.ReadAllText(configPath);
+            Config = HoconParser.Parse(configSource);
+        }
 
         TaskScheduler.Init();
     }
 
-    public static HoconRoot Config { get; set; }
+    public static HoconRoot? Config { get; set; }
 
     public static void Init<T>() where T : MossExtension, new()
     {
@@ -54,7 +64,21 @@ public static void PreInitExtension()
         _instance = Activator.CreateInstance<T>();
 
         var input = Pdk.GetInputJson(JsonContext.Default.MossState);
-        _instance!.Register(input!);
+
+#if DEBUG
+        _instance.Meter = new Meter("Moss.NET.Sdk");
+        _instance.MeterListener = new FileMeterListener("extension/instrumentation.txt");
+        _instance.MeterListener.Init();
+
+        Extensions.Measure("extension.register", () =>
+        {
+#endif
+            _instance!.Register(input!);
+#if DEBUG
+        });
+#endif
+
+
         var extensionInfo = new ExtensionInfo
         {
             Files = Assets.Expose()

src/Moss.NET.Sdk/Scheduler/TaskScheduler.cs

@@ -68,6 +68,16 @@ public static void Init()
             // ignored
         }
 
+        ReadJobConfig(json);
+    }
+
+    private static void ReadJobConfig(string json)
+    {
+        if (MossExtension.Config is null)
+        {
+            return;
+        }
+
         _config = MossExtension.Config.GetObject("scheduler");
 
         foreach (var jobInfo in _config.GetObject("jobs"))

src/SdkTesterPlugin/SampleExtension.cs

@@ -60,7 +60,7 @@ public override void ExtensionLoop(MossState state)
 
         GUI.InvertIcon("swap", "swap_inverted");
 
-        RunOnce.Execute("test", async () =>
+        RunOnce.Execute("test", void () =>
         {
             cm.Open(10, 10);
 
@@ -89,11 +89,11 @@ public override void ExtensionLoop(MossState state)
             var epub = new EpubNotebook("test ebook", "extension/Assets/test.epub");
             InternalFunctions.NewContentEpub();
 
-            await pdf.UploadAsync();
+            pdf.Upload();
 
             InternalFunctions.ExportDocument("0ba3df9c-8ca0-4347-8d7c-07471101baad");
 
-            await quickSheets.EnsureDownloadAsync();
+            quickSheets.EnsureDownload();
             quickSheets.LoadFilesFromCache();
             quickSheets.UnloadFiles();
 
@@ -115,14 +115,14 @@ public override void ExtensionLoop(MossState state)
                 Uuid = quickSheets.Metadata.Accessor.Uuid,
             });*/
 
-            await duplicate.EnsureDownloadAsync();
+            duplicate.EnsureDownload();
             duplicate.Metadata.Get<string>("visible_name");
 
             duplicate.Metadata.Set("visible_name", "Duplicated QuickSheet");
             duplicate.Metadata.Set("parent", "4dba1a54-93b8-4992-886f-08c0b17f93da");
 
             var k = duplicate.Duplicate();
-            await duplicate.UploadAsync();
+            duplicate.Upload();
             k.Delete(() => _logger.Info("Deleted"));
 
             var docs = Enumerable.Repeat(0, 2)

src/SdkTesterPlugin/SdkTesterPlugin.csproj

@@ -8,6 +8,7 @@
     </PropertyGroup>
 
     <ItemGroup>
+        <PackageReference Include="System.Diagnostics.DiagnosticSource" Version="9.0.3" />
         <ProjectReference Include="..\Moss.NET.Sdk\Moss.NET.Sdk.csproj"/>
         <PackageReference Include="Extism.runtime.win-x64" Version="1.9.1"/>
         <PackageReference Include="Extism.Pdk" Version="1.1.1" />