You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
4.9 KiB
138 lines
4.9 KiB
/* Copyright (c) 2015, 2019, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License, version 2.0, for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
|
|
#include "my_inttypes.h"
|
|
#include "my_macros.h"
|
|
#include "mysql/service_locking.h"
|
|
#include "mysql/udf_registration_types.h"
|
|
#include "sql/locking_service.h" // acquire_locking_service_locks
|
|
|
|
/*
|
|
These functions are provided as UDFs rather than built-in SQL functions
|
|
to improve flexibility - it is easier to change the functionality of
|
|
UDFs than built-in functions as UDFs are a weaker contract with the
|
|
user about their functionality.
|
|
|
|
Note that these UDFs does not use the locking service plugin API as this
|
|
is not possible with the current UDF framework implementation.
|
|
*/
|
|
|
|
// Common initialization code for get_read_lock and get_write_lock
|
|
static inline bool init_acquire(UDF_INIT *initid, UDF_ARGS *args,
|
|
char *message) {
|
|
initid->maybe_null = false;
|
|
initid->decimals = 0;
|
|
initid->max_length = 1;
|
|
initid->ptr = NULL;
|
|
initid->const_item = 0;
|
|
initid->extension = NULL;
|
|
|
|
// At least three arguments - namespace, lock, timeout
|
|
if (args->arg_count < 3) {
|
|
strcpy(message,
|
|
"Requires at least three arguments: (namespace,lock(...),timeout).");
|
|
return true;
|
|
}
|
|
|
|
// Timeout is the last argument, should be INT
|
|
if (args->arg_type[args->arg_count - 1] != INT_RESULT) {
|
|
strcpy(message, "Wrong argument type - expected integer.");
|
|
return true;
|
|
}
|
|
|
|
// All other arguments should be strings
|
|
for (size_t i = 0; i < (args->arg_count - 1); i++) {
|
|
if (args->arg_type[i] != STRING_RESULT) {
|
|
strcpy(message, "Wrong argument type - expected string.");
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
extern "C" {
|
|
|
|
bool service_get_read_locks_init(UDF_INIT *initid, UDF_ARGS *args,
|
|
char *message) {
|
|
return init_acquire(initid, args, message);
|
|
}
|
|
|
|
long long service_get_read_locks(UDF_INIT *, UDF_ARGS *args, unsigned char *,
|
|
unsigned char *) {
|
|
const char *lock_namespace = args->args[0];
|
|
long long timeout = *((long long *)args->args[args->arg_count - 1]);
|
|
// For the UDF 1 == success, 0 == failure.
|
|
return !acquire_locking_service_locks(
|
|
NULL, lock_namespace, const_cast<const char **>(&args->args[1]),
|
|
args->arg_count - 2, LOCKING_SERVICE_READ, static_cast<ulong>(timeout));
|
|
}
|
|
|
|
bool service_get_write_locks_init(UDF_INIT *initid, UDF_ARGS *args,
|
|
char *message) {
|
|
return init_acquire(initid, args, message);
|
|
}
|
|
|
|
long long service_get_write_locks(UDF_INIT *, UDF_ARGS *args, unsigned char *,
|
|
unsigned char *) {
|
|
const char *lock_namespace = args->args[0];
|
|
long long timeout = *((long long *)args->args[args->arg_count - 1]);
|
|
// For the UDF 1 == success, 0 == failure.
|
|
return !acquire_locking_service_locks(
|
|
NULL, lock_namespace, const_cast<const char **>(&args->args[1]),
|
|
args->arg_count - 2, LOCKING_SERVICE_WRITE,
|
|
(timeout == -1 ? TIMEOUT_INF : static_cast<Timeout_type>(timeout)));
|
|
}
|
|
|
|
bool service_release_locks_init(UDF_INIT *initid, UDF_ARGS *args,
|
|
char *message) {
|
|
initid->maybe_null = false;
|
|
initid->decimals = 0;
|
|
initid->max_length = 1;
|
|
initid->ptr = NULL;
|
|
initid->const_item = 0;
|
|
initid->extension = NULL;
|
|
|
|
// Only one argument - lock_namespace (string)
|
|
if (args->arg_count != 1) {
|
|
strcpy(message, "Requires one argument: (lock_namespace).");
|
|
return true;
|
|
}
|
|
if (args->arg_type[0] != STRING_RESULT) {
|
|
strcpy(message, "Wrong argument type - expected string.");
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
long long service_release_locks(UDF_INIT *, UDF_ARGS *args, unsigned char *,
|
|
unsigned char *) {
|
|
const char *lock_namespace = args->args[0];
|
|
// For the UDF 1 == success, 0 == failure.
|
|
return !release_locking_service_locks(NULL, lock_namespace);
|
|
}
|
|
|
|
} // extern "C"
|
|
|