Profiling with gperftools

NAMESPACE

@@ -68,6 +68,8 @@ export(nng_version)
 export(opt)
 export(parse_url)
 export(pipe_notify)
+export(profiler_start)
+export(profiler_stop)
 export(random)
 export(reap)
 export(recv)

R/profiler.R

@@ -0,0 +1,22 @@
+#' @title Start the `gperftools` profiler.
+#' @export
+#' @description Start the `gperftools` profiler.
+#' @return `NULL` (invisibly).
+#' @param path Character of length 1 with the file path to the profiling samples.
+profiler_start <- function(path) {
+  stopifnot(all(is.character(path)))
+  stopifnot(all(length(path) == 1L))
+  stopifnot(!anyNA(path))
+  stopifnot(all(nzchar(path)))
+  .Call(rnng_profiler_start, path = path, PACKAGE = "nanonext")
+  invisible()
+}
+
+#' @title Stop the `gperftools` profiler.
+#' @export
+#' @description Stop the `gperftools` profiler.
+#' @return `NULL` (invisibly).
+profiler_stop <- function() {
+  .Call(rnng_profiler_stop, PACKAGE = "nanonext")
+  invisible()
+}

configure

@@ -6,7 +6,7 @@ TLS_VER="edb8fec"
 
 # Initialise
 PKG_CFLAGS=""
-PKG_LIBS="-lnng -lmbedtls -lmbedx509 -lmbedcrypto"
+PKG_LIBS="-lnng -lmbedtls -lmbedx509 -lmbedcrypto -lprofiler"
 NNG_CFLAGS=""
 NNG_LIBS=""
 

src/init.c

@@ -167,6 +167,8 @@ static const R_CallMethodDef callMethods[] = {
   {"rnng_ncurl_transact", (DL_FUNC) &rnng_ncurl_transact, 1},
   {"rnng_next_config", (DL_FUNC) &rnng_next_config, 2},
   {"rnng_pipe_notify", (DL_FUNC) &rnng_pipe_notify, 6},
+  {"rnng_profiler_start", (DL_FUNC) &rnng_profiler_start, 1},
+  {"rnng_profiler_stop", (DL_FUNC) &rnng_profiler_stop, 0},
   {"rnng_protocol_open", (DL_FUNC) &rnng_protocol_open, 2},
   {"rnng_random", (DL_FUNC) &rnng_random, 2},
   {"rnng_reap", (DL_FUNC) &rnng_reap, 1},

src/nanonext.h

@@ -211,6 +211,8 @@ SEXP rnng_ncurl_session_close(SEXP);
 SEXP rnng_ncurl_transact(SEXP);
 SEXP rnng_next_config(SEXP, SEXP);
 SEXP rnng_pipe_notify(SEXP, SEXP, SEXP, SEXP, SEXP, SEXP);
+SEXP rnng_profiler_start(SEXP);
+SEXP rnng_profiler_stop(SEXP);
 SEXP rnng_protocol_open(SEXP, SEXP);
 SEXP rnng_random(SEXP, SEXP);
 SEXP rnng_reap(SEXP);

src/profiler.c

@@ -0,0 +1,13 @@
+#include <R.h>
+#include <Rinternals.h>
+#include "gperftools/profiler.h"
+
+SEXP rnng_profiler_start(SEXP path) {
+  ProfilerStart(CHAR(STRING_ELT(path, 0)));
+  return R_NilValue;
+}
+
+SEXP rnng_profiler_stop() {
+  ProfilerStop();
+  return R_NilValue;
+}