function_v2.cc 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  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 <libgearman/common.h>
  39. #include <libgearman/packet.hpp>
  40. #include <libgearman/function/base.hpp>
  41. #include <libgearman/function/function_v2.hpp>
  42. /*
  43. FunctionV2 function
  44. */
  45. gearman_function_error_t FunctionV2::callback(gearman_job_st* job_shell, void *context_arg)
  46. {
  47. Job* job= job_shell->impl();
  48. if (gearman_job_is_map(job))
  49. {
  50. gearman_job_build_reducer(job, NULL);
  51. }
  52. gearman_return_t error= _function(job_shell, context_arg);
  53. switch (error)
  54. {
  55. case GEARMAN_SHUTDOWN:
  56. job->_error_code= GEARMAN_SUCCESS;
  57. return GEARMAN_FUNCTION_SHUTDOWN;
  58. case GEARMAN_WORK_EXCEPTION:
  59. case GEARMAN_FATAL:
  60. job->_error_code= GEARMAN_FATAL;
  61. return GEARMAN_FUNCTION_FATAL;
  62. case GEARMAN_ERROR:
  63. job->_error_code= GEARMAN_ERROR;
  64. return GEARMAN_FUNCTION_ERROR;
  65. case GEARMAN_SUCCESS:
  66. job->_error_code= GEARMAN_SUCCESS;
  67. return GEARMAN_FUNCTION_SUCCESS;
  68. case GEARMAN_IO_WAIT:
  69. case GEARMAN_SHUTDOWN_GRACEFUL:
  70. case GEARMAN_ERRNO:
  71. case GEARMAN_EVENT:
  72. case GEARMAN_TOO_MANY_ARGS:
  73. case GEARMAN_NO_ACTIVE_FDS:
  74. case GEARMAN_INVALID_MAGIC:
  75. case GEARMAN_INVALID_COMMAND:
  76. case GEARMAN_INVALID_PACKET:
  77. case GEARMAN_UNEXPECTED_PACKET:
  78. case GEARMAN_GETADDRINFO:
  79. case GEARMAN_NO_SERVERS:
  80. case GEARMAN_LOST_CONNECTION:
  81. case GEARMAN_MEMORY_ALLOCATION_FAILURE:
  82. case GEARMAN_JOB_EXISTS:
  83. case GEARMAN_JOB_QUEUE_FULL:
  84. case GEARMAN_SERVER_ERROR:
  85. case GEARMAN_WORK_DATA:
  86. case GEARMAN_WORK_WARNING:
  87. case GEARMAN_WORK_STATUS:
  88. case GEARMAN_NOT_CONNECTED:
  89. case GEARMAN_COULD_NOT_CONNECT:
  90. case GEARMAN_SEND_IN_PROGRESS:
  91. case GEARMAN_RECV_IN_PROGRESS:
  92. case GEARMAN_NOT_FLUSHING:
  93. case GEARMAN_DATA_TOO_LARGE:
  94. case GEARMAN_INVALID_FUNCTION_NAME:
  95. case GEARMAN_INVALID_WORKER_FUNCTION:
  96. case GEARMAN_NO_REGISTERED_FUNCTION:
  97. case GEARMAN_NO_REGISTERED_FUNCTIONS:
  98. case GEARMAN_NO_JOBS:
  99. case GEARMAN_ECHO_DATA_CORRUPTION:
  100. case GEARMAN_NEED_WORKLOAD_FN:
  101. case GEARMAN_PAUSE:
  102. case GEARMAN_UNKNOWN_STATE:
  103. case GEARMAN_PTHREAD:
  104. case GEARMAN_PIPE_EOF:
  105. case GEARMAN_QUEUE_ERROR:
  106. case GEARMAN_FLUSH_DATA:
  107. case GEARMAN_SEND_BUFFER_TOO_SMALL:
  108. case GEARMAN_IGNORE_PACKET:
  109. case GEARMAN_UNKNOWN_OPTION:
  110. case GEARMAN_TIMEOUT:
  111. case GEARMAN_ARGUMENT_TOO_LARGE:
  112. case GEARMAN_INVALID_ARGUMENT:
  113. case GEARMAN_IN_PROGRESS:
  114. case GEARMAN_INVALID_SERVER_OPTION:
  115. case GEARMAN_JOB_NOT_FOUND:
  116. case GEARMAN_MAX_RETURN:
  117. break;
  118. }
  119. gearman_universal_set_error(job->universal(), GEARMAN_INVALID_ARGUMENT, GEARMAN_AT,
  120. "Worker returned invalid gearman_return_t: %s",
  121. gearman_strerror(error));
  122. return GEARMAN_FUNCTION_ERROR;
  123. }