Fix thread utilization on Windows (#1469)

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

Author: TimPansino

newrelic/core/_thread_utilization.c

@@ -16,10 +16,22 @@
 
 /* ------------------------------------------------------------------------- */
 
-#include <sys/time.h>
 #include <Python.h>
 #include <pythread.h>
 
+#ifdef _WIN32
+
+    #include <windows.h>
+    #include <minwinbase.h>
+    #include <sysinfoapi.h>
+    #include <winnt.h>
+
+#else
+
+    #include <sys/time.h>
+
+#endif
+
 #ifndef PyVarObject_HEAD_INIT
 #define PyVarObject_HEAD_INIT(type, size) PyObject_HEAD_INIT(type) size,
 #endif
@@ -34,20 +46,39 @@ typedef struct {
     long long time_last_fetched;
 } UtilizationCount;
 
-static void reset_utilization_count(UtilizationCount *self)
+static void get_current_time_usecs(long long *ts)
 {
-    struct timeval t;
+    #ifdef _WIN32
+        FILETIME ft;
+        ULARGE_INTEGER t;
+
+        // Most accurate timestamp available on Windows.
+        // Measured in 100ns intervals since January 1, 1601 (UTC).
+        GetSystemTimePreciseAsFileTime(&ft);
+        
+        // Convert FILETIME to a 64-bit integer
+        t.LowPart = ft.dwLowDateTime;
+        t.HighPart = ft.dwHighDateTime;
+        
+        // Convert from 100ns intervals to microseconds
+        *ts = (long long) (t.QuadPart / 10);
+    #else
+        struct timeval t;
+
+        gettimeofday(&t, NULL);
+        *ts = (t.tv_sec * 1000000) + t.tv_usec;
+    #endif
+}
+
 
+static void reset_utilization_count(UtilizationCount *self)
+{
     self->currently_active = 0;
 
     self->utilization_current = 0.0;
     self->utilization_previous = 0.0;
 
-    gettimeofday(&t, NULL);
-
-    self->time_last_updated = t.tv_sec * 1000000;
-    self->time_last_updated += t.tv_usec;
-
+    get_current_time_usecs(&self->time_last_updated);
     self->time_last_fetched = self->time_last_updated;
 }
 
@@ -56,12 +87,7 @@ static double adjust_utilization_count(UtilizationCount *self, int adjustment)
     long long current_time;
     double utilization = self->utilization_current;
 
-    struct timeval t;
-
-    gettimeofday(&t, NULL);
-
-    current_time = t.tv_sec * 1000000;
-    current_time += t.tv_usec;
+    get_current_time_usecs(&current_time);
 
     utilization = (current_time - self->time_last_updated) / 1000000.0;
 
@@ -104,8 +130,8 @@ static double fetch_utilization_count(UtilizationCount *self)
 
     time_last_updated = self->time_last_updated;
 
-    elapsed_time = self->time_last_updated - time_last_fetched;
-    elapsed_time /= 1000000.0;
+    // Convert to interval in seconds
+    elapsed_time = (self->time_last_updated - time_last_fetched) / 1000000.0;
 
     if (elapsed_time <= 0)
         return 0.0;
@@ -210,11 +236,7 @@ static double NRUtilization_adjust(NRUtilizationObject *self, int adjustment)
     long long now;
     double utilization = self->requests_utilization_count;
 
-    struct timeval t;
-
-    gettimeofday(&t, NULL);
-
-    now = ((long long)t.tv_sec) * 1000000 + ((long long)t.tv_usec);
+    get_current_time_usecs(&now);
 
     if (self->requests_utilization_last != 0.0) {
         utilization = (now - self->requests_utilization_last) / 1000000.0;

setup.py

@@ -209,9 +209,10 @@ def _run_setup():
         if with_extensions:
             kwargs_tmp["ext_modules"] = [
                 Extension("newrelic.packages.wrapt._wrappers", ["newrelic/packages/wrapt/_wrappers.c"]),
+                Extension("newrelic.core._thread_utilization", ["newrelic/core/_thread_utilization.c"]),
             ]
             if not is_windows:
-                # These extensions are only supported on POSIX platforms.
+                # This extension is only supported on POSIX platforms.
                 monotonic_libraries = []
                 if with_librt():
                     monotonic_libraries = ["rt"]
@@ -221,9 +222,6 @@ def _run_setup():
                         "newrelic.common._monotonic", ["newrelic/common/_monotonic.c"], libraries=monotonic_libraries
                     )
                 )
-                kwargs_tmp["ext_modules"].append(
-                    Extension("newrelic.core._thread_utilization", ["newrelic/core/_thread_utilization.c"])
-                )
 
             kwargs_tmp["cmdclass"] = dict(build_ext=optional_build_ext)