varun-r-mallya/sysprof
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/sysprof.git synced 2026-02-12 16:10:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

libsysprof-profile: wait for recording and seutp ctrl+c

Browse Source
This commit is contained in:
Christian Hergert
2023-05-25 17:25:13 -07:00
parent ce4a4200f8
commit 731841a523
1 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
src/libsysprof-profile/tests/test-profiler.c
View File

@ -18,15 +18,34 @@
* SPDX-License-Identifier: GPL-3.0-or-later * SPDX-License-Identifier: GPL-3.0-or-later
*/ */
#include <glib-unix.h>
#include <sysprof-profile.h> #include <sysprof-profile.h>
static GMainLoop *main_loop; static GMainLoop *main_loop;
static char *capture_file; static char *capture_file;
static SysprofRecording *active_recording;
static const GOptionEntry entries[] = { static const GOptionEntry entries[] = {
{ "capture", 'c', 0, G_OPTION_ARG_FILENAME, &capture_file, "The file to capture into", "CAPTURE" }, { "capture", 'c', 0, G_OPTION_ARG_FILENAME, &capture_file, "The file to capture into", "CAPTURE" },
{ 0 } { 0 }
}; };
static void
wait_cb (GObject *object,
GAsyncResult *result,
gpointer user_data)
{
SysprofRecording *recording = (SysprofRecording *)object;
g_autoptr(GError) error = NULL;
gboolean r;
r = sysprof_recording_wait_finish (recording, result, &error);
g_assert_no_error (error);
g_assert_true (r);
g_main_loop_quit (main_loop);
}
static void static void
record_cb (GObject *object, record_cb (GObject *object,
GAsyncResult *result, GAsyncResult *result,
@ -38,6 +57,34 @@ record_cb (GObject *object,
g_assert_no_error (error); g_assert_no_error (error);
g_assert_nonnull (recording); g_assert_nonnull (recording);
g_assert_true (SYSPROF_IS_RECORDING (recording)); g_assert_true (SYSPROF_IS_RECORDING (recording));
sysprof_recording_wait_async (recording, NULL, wait_cb, NULL);
active_recording = recording;
}
static gboolean
sigint_handler (gpointer user_data)
{
static int count;
if (count >= 2)
{
g_main_loop_quit (main_loop);
return G_SOURCE_REMOVE;
}
g_printerr ("\n");
if (count == 0)
{
g_printerr ("%s\n", "Stopping profiler. Press twice more ^C to force exit.");
sysprof_recording_stop (active_recording);
}
count++;
return G_SOURCE_CONTINUE;
} }
int int
@ -68,6 +115,9 @@ main (int argc,
sysprof_profiler_record_async (profiler, writer, NULL, record_cb, NULL); sysprof_profiler_record_async (profiler, writer, NULL, record_cb, NULL);
g_unix_signal_add (SIGINT, sigint_handler, main_loop);
g_unix_signal_add (SIGTERM, sigint_handler, main_loop);
g_main_loop_run (main_loop); g_main_loop_run (main_loop);
g_clear_pointer (&writer, sysprof_capture_writer_unref); g_clear_pointer (&writer, sysprof_capture_writer_unref);