Dtrace probes in your Golang code

probes.c

/*
  Compile probes.c as a shared library
  In Mac OS X (http://www.finkproject.org/doc/porting/shared.php#build-lib):
  cc -fno-common -c probes.c
  cc -dynamiclib -install_name /usr/local/lib/libprobes.dylib -o libprobes.dylib probes.o
  cp libprobes.dylib /usr/local/lib/
*/
#include "probes.h"

void Ping() {
  GOLANG_PING();
}

int Ping_enabled() {
  return GOLANG_PING_ENABLED();
}

void Pong() {
  GOLANG_PONG();
}

int Pong_enabled() {
  return GOLANG_PONG_ENABLED();
}

probes.d

/* 
  Generate probes.h header file with:
  dtrace -o probes.h -h -s probes.d
*/
provider golang {
  probe ping();
  probe pong();
};

probes.go

package main

/*
#cgo LDFLAGS: -lprobes -L/usr/local/lib
#include "probes.h"
*/
import "C"

import (
  "time"
  "math/rand"
)

func ping(c chan int, q chan int) {
  for ;; {
    time.Sleep(time.Duration(rand.Intn(5)) * time.Second)
    n := <- c
    n += 1

    if n == 10 {
      q <- 0
    }

    c <- n
    C.Ping(C.int(n))
  }
}

func pong(c chan int, q chan int) {
  for ;; {
    time.Sleep(time.Duration(rand.Intn(5)) * time.Second)
    n := <- c
    n -= 1

    if n == 0 {
      q <- 1
    }

    c <- n
    C.Pong(C.int(n))
  }
}

func main() {
  c := make(chan int, 2)
  q := make(chan int)

  go ping(c, q)
  go pong(c, q)

  c <- 5

  // Wait for quit
  winner := <- q

  if winner == 0 {
    println("Ping wins!")
  } else {
    println("Pong wins!")
  }
}

probes.h

/* Header file generated by dtrace */
/* ... */

/* probes.c functions */
void Ping();
int  Ping_enabled();
void Pong();
int  Pong_enabled();

Demo of static Dtrace probes in Golang. Run with go run goarena.go, spectate with sudo dtrace -s spectator.d.

I failed at making Golang's cgo work nicely with a static library, which is why probes.c needs to be compiled as a shared library. Otherwise, dtrace's probe.h file could have been used directly.

Golang bindings for Chris Andrew's libusdt would probably allow more user-friendly dtrace probes in similar programs.