ex_mpool.c
上传用户:tsgydb
上传日期:2007-04-14
资源大小:10674k
文件大小:6k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /*-
  2.  * See the file LICENSE for redistribution information.
  3.  *
  4.  * Copyright (c) 1997, 1998, 1999, 2000
  5.  * Sleepycat Software.  All rights reserved.
  6.  *
  7.  * $Id: ex_mpool.c,v 11.13 2000/10/27 20:32:00 dda Exp $
  8.  */
  9. #include "db_config.h"
  10. #ifndef NO_SYSTEM_INCLUDES
  11. #include <sys/types.h>
  12. #if TIME_WITH_SYS_TIME
  13. #include <sys/time.h>
  14. #include <time.h>
  15. #else
  16. #if HAVE_SYS_TIME_H
  17. #include <sys/time.h>
  18. #else
  19. #include <time.h>
  20. #endif
  21. #endif
  22. #include <errno.h>
  23. #include <fcntl.h>
  24. #include <stdlib.h>
  25. #include <string.h>
  26. #include <unistd.h>
  27. #endif
  28. #include <db.h>
  29. int init __P((char *, int, int, char *));
  30. int run __P((int, int, int, int, char *));
  31. int run_mpool __P((int, int, int, int, char *));
  32. #ifdef HAVE_VXWORKS
  33. int ex_mpool __P((void));
  34. #define MPOOL "/vxtmp/vxtmp/mpool" /* File. */
  35. #define ERROR_RETURN ERROR
  36. #define VXSHM_KEY 12
  37. #else
  38. int main __P((int, char *[]));
  39. void usage __P((char *));
  40. #define MPOOL "mpool" /* File. */
  41. #define ERROR_RETURN 1
  42. #endif
  43. #ifndef HAVE_VXWORKS
  44. int
  45. main(argc, argv)
  46. int argc;
  47. char *argv[];
  48. {
  49. extern char *optarg;
  50. extern int optind;
  51. int cachesize, ch, hits, npages, pagesize;
  52. char *progname;
  53. cachesize = 20 * 1024;
  54. hits = 1000;
  55. npages = 50;
  56. pagesize = 1024;
  57. progname = argv[0];
  58. while ((ch = getopt(argc, argv, "c:h:n:p:")) != EOF)
  59. switch (ch) {
  60. case 'c':
  61. if ((cachesize = atoi(optarg)) < 20 * 1024)
  62. usage(progname);
  63. break;
  64. case 'h':
  65. if ((hits = atoi(optarg)) <= 0)
  66. usage(progname);
  67. break;
  68. case 'n':
  69. if ((npages = atoi(optarg)) <= 0)
  70. usage(progname);
  71. break;
  72. case 'p':
  73. if ((pagesize = atoi(optarg)) <= 0)
  74. usage(progname);
  75. break;
  76. case '?':
  77. default:
  78. usage(progname);
  79. }
  80. argc -= optind;
  81. argv += optind;
  82. return (run_mpool(pagesize, cachesize, hits, npages, progname));
  83. }
  84. void
  85. usage(progname)
  86. char *progname;
  87. {
  88. (void)fprintf(stderr,
  89.     "usage: %s [-c cachesize] [-h hits] [-n npages] [-p pagesize]n",
  90.     progname);
  91. exit(1);
  92. }
  93. #else
  94. int
  95. ex_mpool()
  96. {
  97. char *progname = "ex_mpool"; /* Program name. */
  98. int cachesize, ch, hits, npages, pagesize;
  99. cachesize = 20 * 1024;
  100. hits = 1000;
  101. npages = 50;
  102. pagesize = 1024;
  103. return (run_mpool(pagesize, cachesize, hits, npages, progname));
  104. }
  105. #endif
  106. int
  107. run_mpool(pagesize, cachesize, hits, npages, progname)
  108. int pagesize, cachesize, hits, npages;
  109. char *progname;
  110. {
  111. int ret;
  112. /* Initialize the file. */
  113. if ((ret = init(MPOOL, pagesize, npages, progname)) != 0)
  114. return (ret);
  115. /* Get the pages. */
  116. if ((ret = run(hits, cachesize, pagesize, npages, progname)) != 0)
  117. return (ret);
  118. return (0);
  119. }
  120. /*
  121.  * init --
  122.  * Create a backing file.
  123.  */
  124. int
  125. init(file, pagesize, npages, progname)
  126. char *file, *progname;
  127. int pagesize, npages;
  128. {
  129. int cnt, flags, fd;
  130. char *p;
  131. /*
  132.  * Create a file with the right number of pages, and store a page
  133.  * number on each page.
  134.  */
  135. flags = O_CREAT | O_RDWR | O_TRUNC;
  136. #ifdef DB_WIN32
  137. flags |= O_BINARY;
  138. #endif
  139. if ((fd = open(file, flags, 0666)) < 0) {
  140. fprintf(stderr,
  141.     "%s: %s: %sn", progname, file, strerror(errno));
  142. return (ERROR_RETURN);
  143. }
  144. if ((p = (char *)malloc(pagesize)) == NULL) {
  145. fprintf(stderr, "%s: %sn", progname, strerror(ENOMEM));
  146. return (ERROR_RETURN);
  147. }
  148. /* The pages are numbered from 0. */
  149. for (cnt = 0; cnt <= npages; ++cnt) {
  150. *(int *)p = cnt;
  151. if (write(fd, p, pagesize) != pagesize) {
  152. fprintf(stderr,
  153.     "%s: %s: %sn", progname, file, strerror(errno));
  154. return (ERROR_RETURN);
  155. }
  156. }
  157. (void)close(fd);
  158. free(p);
  159. return (0);
  160. }
  161. /*
  162.  * run --
  163.  * Get a set of pages.
  164.  */
  165. int
  166. run(hits, cachesize, pagesize, npages, progname)
  167. int hits, cachesize, pagesize, npages;
  168. char *progname;
  169. {
  170. DB_ENV *dbenv;
  171. DB_MPOOLFILE *dbmfp;
  172. db_pgno_t pageno;
  173. int cnt, ret;
  174. void *p;
  175. printf("%s: cachesize: %d; pagesize: %d; N pages: %dn",
  176.     progname, cachesize, pagesize, npages);
  177. /*
  178.  * Open a memory pool, specify a cachesize, output error messages
  179.  * to stderr.
  180.  */
  181. if ((ret = db_env_create(&dbenv, 0)) != 0) {
  182. fprintf(stderr,
  183.     "%s: db_env_create: %sn", progname, db_strerror(ret));
  184. return (ERROR_RETURN);
  185. }
  186. dbenv->set_errfile(dbenv, stderr);
  187. dbenv->set_errpfx(dbenv, progname);
  188. #ifdef HAVE_VXWORKS
  189. if ((ret = dbenv->set_shm_key(dbenv, VXSHM_KEY)) != 0) {
  190. dbenv->err(dbenv, ret, "set_shm_key");
  191. return (ERROR_RETURN);
  192. }
  193. #endif
  194. /* Set the cachesize. */
  195. if ((ret = dbenv->set_cachesize(dbenv, 0, cachesize, 0)) != 0) {
  196. dbenv->err(dbenv, ret, "set_cachesize");
  197. goto err1;
  198. }
  199. /* Open the environment. */
  200. if ((ret = dbenv->open(
  201.     dbenv, NULL, DB_CREATE | DB_INIT_MPOOL, 0)) != 0) {
  202. dbenv->err(dbenv, ret, "open");
  203. goto err1;
  204. }
  205. /* Open the file in the environment. */
  206. if ((ret =
  207.     memp_fopen(dbenv, MPOOL, 0, 0, pagesize, NULL, &dbmfp)) != 0) {
  208. dbenv->err(dbenv, ret, "memp_fopen: %s", MPOOL);
  209. goto err1;
  210. }
  211. printf("retrieve %d random pages... ", hits);
  212. srand((u_int)time(NULL));
  213. for (cnt = 0; cnt < hits; ++cnt) {
  214. pageno = (rand() % npages) + 1;
  215. if ((ret = memp_fget(dbmfp, &pageno, 0, &p)) != 0) {
  216. dbenv->err(dbenv, ret,
  217.     "unable to retrieve page %lu", (u_long)pageno);
  218. goto err2;
  219. }
  220. if (*(db_pgno_t *)p != pageno) {
  221. dbenv->errx(dbenv,
  222.     "wrong page retrieved (%lu != %d)",
  223.     (u_long)pageno, *(int *)p);
  224. goto err2;
  225. }
  226. if ((ret = memp_fput(dbmfp, p, 0)) != 0) {
  227. dbenv->err(dbenv, ret,
  228.     "unable to return page %lu", (u_long)pageno);
  229. goto err2;
  230. }
  231. }
  232. printf("successful.n");
  233. /* Close the file. */
  234. if ((ret = memp_fclose(dbmfp)) != 0) {
  235. dbenv->err(dbenv, ret, "memp_fclose");
  236. goto err1;
  237. }
  238. /* Close the pool. */
  239. if ((ret = dbenv->close(dbenv, 0)) != 0) {
  240. fprintf(stderr,
  241.     "%s: db_env_create: %sn", progname, db_strerror(ret));
  242. return (ERROR_RETURN);
  243. }
  244. return (0);
  245. err2: (void)memp_fclose(dbmfp);
  246. err1: (void)dbenv->close(dbenv, 0);
  247. return (ERROR_RETURN);
  248. }