← Back to branch summary

~clint-fewbar/ubuntu/precise/gearmand/drop-unneeded-patches

« back to all changes in this revision

Viewing changes to tests/context.h

  • Committer: Bazaar Package Importer
  • Author(s): Monty Taylor
  • Date: 2009-08-11 10:06:22 UTC
  • mto: (1.2.3 upstream) (6.1.1 sid)
  • mto: This revision was merged to the branch mainline in revision 3.
  • Revision ID: james.westby@ubuntu.com-20090811100622-6ig4iknanc73olum
Import upstream version 0.9

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/context.h
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
 
 
38
 
 
39
 
#pragma once
40
 
 
41
 
#include <string>
42
 
#include <unistd.h>
43
 
 
44
 
class Context
45
 
{
46
 
public:
47
 
  gearman_worker_st *worker;
48
 
  in_port_t _port;
49
 
  std::string _worker_function_name;
50
 
  server_startup_st &_servers;
51
 
  bool run_worker;
52
 
 
53
 
  Context(in_port_t port_arg, server_startup_st &server_arg):
54
 
    worker(NULL),
55
 
    _port(port_arg),
56
 
    _worker_function_name("queue_test"),
57
 
    _servers(server_arg),
58
 
    run_worker(false)
59
 
  {
60
 
  }
61
 
 
62
 
  const char *worker_function_name() const
63
 
  {
64
 
    return _worker_function_name.c_str();
65
 
  }
66
 
 
67
 
  in_port_t port() const
68
 
  {
69
 
    return _port;
70
 
  }
71
 
 
72
 
  bool initialize(int argc, const char *argv[])
73
 
  {
74
 
    if (server_startup(_servers, "gearmand", _port, argc, argv) == false)
75
 
    {
76
 
      return NULL;
77
 
    }
78
 
 
79
 
    if ((worker= gearman_worker_create(NULL)) == NULL)
80
 
    {
81
 
      return false;
82
 
    }
83
 
 
84
 
    if (gearman_failed(gearman_worker_add_server(worker, NULL, _port)))
85
 
    {
86
 
      return false;
87
 
    }
88
 
 
89
 
    return true;
90
 
  }
91
 
 
92
 
  void reset()
93
 
  {
94
 
    _servers.shutdown_and_remove();
95
 
    gearman_worker_free(worker);
96
 
 
97
 
    worker= NULL;
98
 
  }
99
 
};