~drizzle-trunk/drizzle/development

1963.2.1 by Brian Aker
Fixing up the kill function.
1
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 *
4
 *  Copyright (C) 2010 Brian Aker
5
 *
6
 *  This program is free software; you can redistribute it and/or modify
7
 *  it under the terms of the GNU General Public License as published by
8
 *  the Free Software Foundation; either version 2 of the License, or
9
 *  (at your option) any later version.
10
 *
11
 *  This program is distributed in the hope that it will be useful,
12
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 *  GNU General Public License for more details.
15
 *
16
 *  You should have received a copy of the GNU General Public License
17
 *  along with this program; if not, write to the Free Software
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
 */
20
2173.2.1 by Monty Taylor
Fixes incorrect usage of include
21
#include <config.h>
1963.2.1 by Brian Aker
Fixing up the kill function.
22
23
#include <drizzled/session.h>
1966.2.5 by Brian Aker
Style change around session list.
24
#include <drizzled/session/cache.h>
2173.2.1 by Monty Taylor
Fixes incorrect usage of include
25
#include <plugin/utility_functions/functions.h>
1963.2.1 by Brian Aker
Fixing up the kill function.
26
2282.1.2 by Olaf van der Spek
Session Cache
27
namespace drizzled {
28
namespace utility_functions {
1963.2.1 by Brian Aker
Fixing up the kill function.
29
30
int64_t Kill::val_int()
31
{
32
  int64_t session_id_for_kill= args[0]->val_int();
33
34
  if (getSession().getSessionId() == session_id_for_kill)
35
  {
36
    my_error(drizzled::ER_KILL_DENY_SELF_ERROR, MYF(0));
37
    return 0;
38
  }
39
2282.1.2 by Olaf van der Spek
Session Cache
40
  Session::shared_ptr die_session(session::Cache::find(session_id_for_kill));
1963.2.1 by Brian Aker
Fixing up the kill function.
41
42
  if (not die_session)
43
  {
44
    my_error(drizzled::ER_NO_SUCH_THREAD, session_id_for_kill, MYF(0));
45
    return 0;
46
  }
47
2015.3.1 by Brian Aker
Encapsulate client call. Also remove the need to call current_session when
48
  if (not die_session->isViewable(*getSession().user()))
1963.2.1 by Brian Aker
Fixing up the kill function.
49
  {
50
    my_error(drizzled::ER_KILL_DENIED_ERROR, session_id_for_kill, MYF(0));
51
    return 0;
52
  }
53
54
  die_session->awake(arg_count == 2 ? Session::KILL_QUERY : Session::KILL_CONNECTION);
55
56
  return session_id_for_kill;
57
}
58
59
} /* namespace utility_functions */
60
} /* namespace drizzled */