Home | History | Annotate | Line # | Download | only in Python 
      1 #!/usr/sbin/dtrace -Zs
      2 /*
      3  * py_cputime.d - measure Python on-CPU times for functions.
      4  *                Written for the Python DTrace provider.
      5  *
      6  * $Id: py_cputime.d,v 1.1.1.1 2015/09/30 22:01:09 christos Exp $
      7  *
      8  * This traces Python activity from all programs running on the system with
      9  * Python provider support.
     10  *
     11  * USAGE: py_cputime.d 		# hit Ctrl-C to end
     12  *
     13  * FIELDS:
     14  *		FILE		Filename of the Python program
     15  *		TYPE		Type of call (func/total)
     16  *		NAME		Name of call (function name)
     17  *		TOTAL		Total on-CPU time for calls (us)
     18  *
     19  * Filename and function names are printed if available.
     20  *
     21  * COPYRIGHT: Copyright (c) 2007 Brendan Gregg.
     22  *
     23  * CDDL HEADER START
     24  *
     25  *  The contents of this file are subject to the terms of the
     26  *  Common Development and Distribution License, Version 1.0 only
     27  *  (the "License").  You may not use this file except in compliance
     28  *  with the License.
     29  *
     30  *  You can obtain a copy of the license at Docs/cddl1.txt
     31  *  or http://www.opensolaris.org/os/licensing.
     32  *  See the License for the specific language governing permissions
     33  *  and limitations under the License.
     34  *
     35  * CDDL HEADER END
     36  *
     37  * 09-Sep-2007	Brendan Gregg	Created this.
     38  */
     39 
     40 #pragma D option quiet
     41 
     42 dtrace:::BEGIN
     43 {
     44 	printf("Tracing... Hit Ctrl-C to end.\n");
     45 }
     46 
     47 python*:::function-entry
     48 {
     49 	self->depth++;
     50 	self->exclude[self->depth] = 0;
     51 	self->function[self->depth] = vtimestamp;
     52 }
     53 
     54 python*:::function-return
     55 /self->function[self->depth]/
     56 {
     57 	this->oncpu_incl = vtimestamp - self->function[self->depth];
     58 	this->oncpu_excl = this->oncpu_incl - self->exclude[self->depth];
     59 	self->function[self->depth] = 0;
     60 	self->exclude[self->depth] = 0;
     61 	this->file = basename(copyinstr(arg0));
     62 	this->name = copyinstr(arg1);
     63 
     64 	@num[this->file, "func", this->name] = count();
     65 	@num["-", "total", "-"] = count();
     66 	@types_incl[this->file, "func", this->name] = sum(this->oncpu_incl);
     67 	@types_excl[this->file, "func", this->name] = sum(this->oncpu_excl);
     68 	@types_excl["-", "total", "-"] = sum(this->oncpu_excl);
     69 
     70 	self->depth--;
     71 	self->exclude[self->depth] += this->oncpu_incl;
     72 }
     73 
     74 dtrace:::END
     75 {
     76 	printf("\nCount,\n");
     77 	printf("   %-20s %-10s %-32s %8s\n", "FILE", "TYPE", "NAME", "COUNT");
     78 	printa("   %-20s %-10s %-32s %@8d\n", @num);
     79 
     80 	normalize(@types_excl, 1000);
     81 	printf("\nExclusive function on-CPU times (us),\n");
     82 	printf("   %-20s %-10s %-32s %8s\n", "FILE", "TYPE", "NAME", "TOTAL");
     83 	printa("   %-20s %-10s %-32s %@8d\n", @types_excl);
     84 
     85 	normalize(@types_incl, 1000);
     86 	printf("\nInclusive function on-CPU times (us),\n");
     87 	printf("   %-20s %-10s %-32s %8s\n", "FILE", "TYPE", "NAME", "TOTAL");
     88 	printa("   %-20s %-10s %-32s %@8d\n", @types_incl);
     89 }
     90