← Back to branch summary

~ubuntu-branches/ubuntu/lucid/postgresql-8.4/lucid-proposed

« back to all changes in this revision

Viewing changes to src/interfaces/ecpg/test/thread/thread_implicit.pgc

  • Committer: Bazaar Package Importer
  • Author(s): Martin Pitt
  • Date: 2009-03-20 12:00:13 UTC
  • Revision ID: james.westby@ubuntu.com-20090320120013-hogj7egc5mjncc5g
Tags: upstream-8.4~0cvs20090328
Import upstream version 8.4~0cvs20090328

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/interfaces/ecpg/test/thread/thread_implicit.pgc
 
1
/*
 
2
 *      Thread test program
 
3
 *      by Lee Kindness.
 
4
 */
 
5
 
 
6
#include <stdlib.h>
 
7
#include "ecpg_config.h"
 
8
 
 
9
#ifndef ENABLE_THREAD_SAFETY
 
10
int
 
11
main(void)
 
12
{
 
13
        printf("No threading enabled.\n");
 
14
        return 0;
 
15
}
 
16
#else
 
17
#ifndef WIN32
 
18
#include <pthread.h>
 
19
#else
 
20
#include <windows.h>
 
21
#endif
 
22
 
 
23
exec sql include ../regression;
 
24
 
 
25
void *test_thread(void *arg);
 
26
 
 
27
int nthreads   = 10;
 
28
int iterations = 20;
 
29
 
 
30
int main(int argc, char *argv[])
 
31
{
 
32
#ifndef WIN32
 
33
  pthread_t *threads;
 
34
#else
 
35
  HANDLE *threads;
 
36
#endif
 
37
  int n;
 
38
  EXEC SQL BEGIN DECLARE SECTION;
 
39
  int l_rows;
 
40
  EXEC SQL END DECLARE SECTION;
 
41
 
 
42
 /* Do not switch on debug output for regression tests. The threads get executed in
 
43
  * more or less random order */
 
44
 /* ECPGdebug(1, stderr); */
 
45
 
 
46
  /* setup test_thread table */
 
47
  EXEC SQL CONNECT TO REGRESSDB1;
 
48
  EXEC SQL DROP TABLE test_thread; /* DROP might fail */
 
49
  EXEC SQL COMMIT;
 
50
  EXEC SQL CREATE TABLE
 
51
    test_thread(tstamp    TIMESTAMP NOT NULL DEFAULT CAST(timeofday() AS TIMESTAMP),
 
52
                thread    TEXT      NOT NULL,
 
53
                iteration INTEGER   NOT NULL,
 
54
                PRIMARY KEY(thread, iteration));
 
55
  EXEC SQL COMMIT;
 
56
  EXEC SQL DISCONNECT;
 
57
 
 
58
  /* create, and start, threads */
 
59
  threads = calloc(nthreads, sizeof(threads[0]));
 
60
  if( threads == NULL )
 
61
    {
 
62
      fprintf(stderr, "Cannot alloc memory\n");
 
63
      return( 1 );
 
64
    }
 
65
  for( n = 0; n < nthreads; n++ )
 
66
    {
 
67
#ifndef WIN32
 
68
      pthread_create(&threads[n], NULL, test_thread, (void *) (long) (n + 1));
 
69
#else
 
70
      threads[n] = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) test_thread, (void *) (n+1), 0, NULL);
 
71
#endif
 
72
    }
 
73
 
 
74
  /* wait for thread completion */
 
75
#ifndef WIN32
 
76
  for( n = 0; n < nthreads; n++ )
 
77
    {
 
78
      pthread_join(threads[n], NULL);
 
79
    }
 
80
#else
 
81
  WaitForMultipleObjects(nthreads, threads, TRUE, INFINITE);
 
82
#endif
 
83
  free(threads);
 
84
 
 
85
  /* and check results */
 
86
  EXEC SQL CONNECT TO REGRESSDB1;
 
87
  EXEC SQL SELECT COUNT(*) INTO :l_rows FROM test_thread;
 
88
  EXEC SQL COMMIT;
 
89
  EXEC SQL DISCONNECT;
 
90
  if( l_rows == (nthreads * iterations) )
 
91
    printf("Success.\n");
 
92
  else
 
93
    printf("ERROR: Failure - expecting %d rows, got %d.\n", nthreads * iterations, l_rows);
 
94
 
 
95
  return( 0 );
 
96
}
 
97
 
 
98
void *test_thread(void *arg)
 
99
{
 
100
  long threadnum = (long)arg;
 
101
  EXEC SQL BEGIN DECLARE SECTION;
 
102
  int  l_i;
 
103
  char l_connection[128];
 
104
  EXEC SQL END DECLARE SECTION;
 
105
 
 
106
  /* build up connection name, and connect to database */
 
107
#ifndef WIN32_ONLY_COMPILER
 
108
  snprintf(l_connection, sizeof(l_connection), "thread_%03ld", threadnum);
 
109
#else
 
110
  _snprintf(l_connection, sizeof(l_connection), "thread_%03ld", threadnum);
 
111
#endif
 
112
  EXEC SQL WHENEVER sqlerror sqlprint;
 
113
  EXEC SQL CONNECT TO REGRESSDB1 AS :l_connection;
 
114
  if( sqlca.sqlcode != 0 )
 
115
    {
 
116
      printf("%s: ERROR: cannot connect to database!\n", l_connection);
 
117
      return( NULL );
 
118
    }
 
119
  EXEC SQL BEGIN;
 
120
 
 
121
  /* insert into test_thread table */
 
122
  for( l_i = 1; l_i <= iterations; l_i++ )
 
123
    {
 
124
      EXEC SQL INSERT INTO test_thread(thread, iteration) VALUES(:l_connection, :l_i);
 
125
      if( sqlca.sqlcode != 0 )
 
126
        printf("%s: ERROR: insert failed!\n", l_connection);
 
127
    }
 
128
 
 
129
  /* all done */
 
130
  EXEC SQL COMMIT;
 
131
  EXEC SQL DISCONNECT :l_connection;
 
132
  return( NULL );
 
133
}
 
134
#endif /* ENABLE_THREAD_SAFETY */