increment_reset.cc 3.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
  2. *
  3. * Gearmand client and server library.
  4. *
  5. * Copyright (C) 2011 Data Differential, http://datadifferential.com/
  6. * All rights reserved.
  7. *
  8. * Redistribution and use in source and binary forms, with or without
  9. * modification, are permitted provided that the following conditions are
  10. * met:
  11. *
  12. * * Redistributions of source code must retain the above copyright
  13. * notice, this list of conditions and the following disclaimer.
  14. *
  15. * * Redistributions in binary form must reproduce the above
  16. * copyright notice, this list of conditions and the following disclaimer
  17. * in the documentation and/or other materials provided with the
  18. * distribution.
  19. *
  20. * * The names of its contributors may not be used to endorse or
  21. * promote products derived from this software without specific prior
  22. * written permission.
  23. *
  24. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27. * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28. * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35. *
  36. */
  37. #include <gear_config.h>
  38. #include <libtest/test.hpp>
  39. #include <libgearman-1.0/gearman.h>
  40. #include "tests/workers/v2/increment_reset.h"
  41. #include <climits>
  42. static pthread_mutex_t increment_reset_worker_mutex= PTHREAD_MUTEX_INITIALIZER;
  43. gearman_return_t increment_reset_worker_v2(gearman_job_st *job, void *)
  44. {
  45. static long counter= 0;
  46. long change= 0;
  47. const char *workload= (const char*)gearman_job_workload(job);
  48. if (gearman_job_workload_size(job) == test_literal_param_size("reset") and (not memcmp(workload, test_literal_param("reset"))))
  49. {
  50. pthread_mutex_lock(&increment_reset_worker_mutex);
  51. counter= 0;
  52. pthread_mutex_unlock(&increment_reset_worker_mutex);
  53. return GEARMAN_SUCCESS;
  54. }
  55. else if (workload and gearman_job_workload_size(job))
  56. {
  57. if (gearman_job_workload_size(job) > GEARMAN_MAXIMUM_INTEGER_DISPLAY_LENGTH)
  58. {
  59. return GEARMAN_FAIL;
  60. }
  61. char temp[GEARMAN_MAXIMUM_INTEGER_DISPLAY_LENGTH +1];
  62. memcpy(temp, workload, gearman_job_workload_size(job));
  63. temp[gearman_job_workload_size(job)]= 0;
  64. change= strtol(temp, (char **)NULL, 10);
  65. if (change == LONG_MIN or change == LONG_MAX or ( change == 0 and errno < 0))
  66. {
  67. gearman_job_send_exception(job, test_literal_param("strtol() failed"));
  68. return GEARMAN_FAIL;
  69. }
  70. }
  71. {
  72. pthread_mutex_lock(&increment_reset_worker_mutex);
  73. counter= counter +change;
  74. char result[GEARMAN_MAXIMUM_INTEGER_DISPLAY_LENGTH +1];
  75. size_t result_size= size_t(snprintf(result, sizeof(result), "%ld", counter));
  76. if (gearman_failed(gearman_job_send_data(job, result, result_size)))
  77. {
  78. pthread_mutex_unlock(&increment_reset_worker_mutex);
  79. return GEARMAN_FAIL;
  80. }
  81. pthread_mutex_unlock(&increment_reset_worker_mutex);
  82. }
  83. return GEARMAN_SUCCESS;
  84. }