From f4203087de090fb4e84c7001d48460a7c548f94c Mon Sep 17 00:00:00 2001
From: =?utf8?q?Milo=C5=A1=20Mula=C4=8D?= <mulac@civ.zcu.cz>
Date: Thu, 12 Oct 2006 14:31:18 +0000
Subject: [PATCH] enable to trim number of threads and random start delay

---
 org.glite.lb.client/examples/user_jobs_threaded.c | 118 ++++++++++++++--------
 1 file changed, 74 insertions(+), 44 deletions(-)

diff --git a/org.glite.lb.client/examples/user_jobs_threaded.c b/org.glite.lb.client/examples/user_jobs_threaded.c
index ccb6d3f..b529f77 100644
--- a/org.glite.lb.client/examples/user_jobs_threaded.c
+++ b/org.glite.lb.client/examples/user_jobs_threaded.c
@@ -3,6 +3,8 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
+#include <getopt.h>
 
 #include <expat.h>
 
@@ -18,15 +20,39 @@ int use_proxy = 0;
 int (*user_jobs)(edg_wll_Context, edg_wlc_JobId **, edg_wll_JobStat **);
 
 
-void
-usage(char *me)
+
+static const char *get_opt_string = "hpj:t:s:w:";
+
+static struct option opts[] = {
+	{"help",	0, NULL,	'h'},
+	{"proxy",       0, NULL,	'x'},
+	{"owner",	1, NULL,	'o'},
+	{"num-threads",	1, NULL,	't'},
+	{"rand-start",	1, NULL,	's'},
+//	{"rand-work",	1, NULL,	'w'},
+	{NULL,		0, NULL,	0}
+};
+
+
+
+static void usage(char *me) 
 {
-	fprintf(stderr,"usage: %s [-h] [-x] [userid]\n", me);
+	fprintf(stderr,"usage: %s [option]\n"
+		"\t-h, --help\t show this help\n"
+		"\t-p, --proxy\t contact proxy (not implemented yet)\n"
+		"\t-o, --owner DN\t show jobs of user with this DN\n"
+		"\t-t, --num-threads N\t number fo threads to create\n"
+		"\t-s, --rand-start N\t start threads in random interval <0,N> sec\n"
+//		"\t-w, --rand-work N\t simulate random server respose time <0,N> sec\n"
+		"\n"
+	,me);
 }
 
+
 typedef struct {
     char *owner;
     int proxy;
+    int rand_start;
     char *argv_0;} thread_code_args;
 
 void *thread_code(thread_code_args *arguments) {
@@ -36,10 +62,16 @@ void *thread_code(thread_code_args *arguments) {
 	edg_wlc_JobId		*jobs = NULL;
 	edg_wll_JobStat		*states = NULL;
 	int		i,j;
+	long		sl;
+
+
+	sl = (unsigned long) ((double) random()/ RAND_MAX * arguments->rand_start * 1000000);
+	printf("Thread [%d] - sleeping for %ld us\n",pthread_self(),sl);
+	usleep( sl );
 
 	user_jobs = edg_wll_UserJobs;
 
-       if (arguments->proxy) {
+	if (arguments->proxy) {
             user_jobs = edg_wll_UserJobsProxy;
         }
 
@@ -110,60 +142,58 @@ err:
 
 int main(int argc,char **argv)
 {
+        thread_code_args arguments = { NULL, 0, 0, NULL };
+	int	i,rc,status,opt;
+	int	thr_num = 10;	// default
 
-        #define NUM_THREADS 60
 
-        thread_code_args arguments = { NULL , 0 , NULL };
-	int	i,rc,status;
-        pthread_t threads[NUM_THREADS];
-	pthread_attr_t attr;
+        while ((opt = getopt_long(argc,argv,get_opt_string,opts,NULL)) != EOF) switch (opt) {
+                case 'x': arguments.proxy = 1; break;
+                case 'o': arguments.owner = optarg; break;
+                case 't': thr_num = atoi(optarg); break;
+                case 's': arguments.rand_start = atoi(optarg); break;
+                default : usage(argv[0]); exit(0); break;
+        }
 
-	if (globus_module_activate(GLOBUS_COMMON_MODULE) != GLOBUS_SUCCESS)   {
-		fputs("globus_module_activate()\n",stderr);
-		return 1;
-	}
+        arguments.argv_0 = argv[0];
 
-        pthread_attr_init(&attr);
-        pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
+	/* Do a thready work */
+	{
+        	pthread_t threads[thr_num];
+		pthread_attr_t attr;
 
-	for ( i = 1; i < argc; i++ ) {
-		if ( !strcmp(argv[i], "-h") || !strcmp(argv[i], "--help") ) {
-			usage(argv[0]);
-			exit(0);
+		if (globus_module_activate(GLOBUS_COMMON_MODULE) != GLOBUS_SUCCESS)   {
+			fputs("globus_module_activate()\n",stderr);
+			return 1;
 		}
-                else if ( !strcmp(argv[i], "-x") ) {
-                    arguments.proxy = 1;
-         	}
 
-		arguments.owner = strdup(argv[i]);
-	}
-
-        arguments.argv_0 = argv[0];
-
-
-        for (i=0;i<NUM_THREADS;i++) {
-            printf("Running thread %d\n",i);
-            rc = pthread_create(&threads[i], &attr, (void*)thread_code, &arguments );
-            if (rc){
-                printf("ERROR; return code from pthread_create() is %d\n", rc);
-                exit(-1);
-            }
-        }
+        	pthread_attr_init(&attr);
+	        pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
+	
+		for (i=0;i<thr_num;i++) {
+		    printf("Running thread %d\n",i);
+		    rc = pthread_create(&threads[i], &attr, (void*)thread_code, &arguments );
+		    if (rc){
+			printf("ERROR; return code from pthread_create() is %d\n", rc);
+			exit(-1);
+		    }
+		}
 
 
 
-//        thread_code(&arguments);
+		//        thread_code(&arguments);
 
-//        pthread_exit(NULL)
+		//        pthread_exit(NULL)
 
-        pthread_attr_destroy(&attr);
+		pthread_attr_destroy(&attr);
 
-        for (i=0;i<NUM_THREADS;i++) {
-//            printf("*** Joining thread %d.\n",i);
-            rc=pthread_join(threads[i],(void **)&status);
-//            printf("*** Thread %d joined. (status %d ret. code %d)\n",i,status,rc);
+		for (i=0;i<thr_num;i++) {
+		//            printf("*** Joining thread %d.\n",i);
+		    rc=pthread_join(threads[i],(void **)&status);
+		//            printf("*** Thread %d joined. (status %d ret. code %d)\n",i,status,rc);
 
-        }
+		}
+	}
 
         printf("Threaded example main() exitting\n");
 
-- 
1.8.2.3