106 lines
2.1 KiB
C++
106 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2016 MariaDB Corporation Ab
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
|
*
|
|
* Change Date: 2022-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2 or later of the General
|
|
* Public License.
|
|
*/
|
|
|
|
#include <atomic>
|
|
#include <iostream>
|
|
#include <maxbase/log.hh>
|
|
#include <maxscale/housekeeper.h>
|
|
#include <maxscale/mainworker.hh>
|
|
#include "test_utils.hh"
|
|
|
|
using namespace std;
|
|
|
|
namespace
|
|
{
|
|
|
|
std::atomic<int> n_oneshot;
|
|
std::atomic<int> n_repeating;
|
|
|
|
const char* ZONESHOT_NAME = "OneShot";
|
|
const char* ZREPEATING_NAME = "Repeating";
|
|
|
|
bool oneshot(void*)
|
|
{
|
|
++n_oneshot;
|
|
return false; // Remove from housekeeper.
|
|
}
|
|
|
|
bool repeating(void*)
|
|
{
|
|
++n_repeating;
|
|
return true; // Continue calling.
|
|
}
|
|
|
|
int test()
|
|
{
|
|
int rc = EXIT_SUCCESS;
|
|
|
|
hktask_add(ZONESHOT_NAME, oneshot, nullptr, 1); // Call oneshot, once per second.
|
|
hktask_add(ZREPEATING_NAME, repeating, nullptr, 1); // Call repeating, once per second.
|
|
|
|
sleep(4); // Should get 1 oneshot call and ~4 repeating calls.
|
|
|
|
hktask_remove(ZREPEATING_NAME);
|
|
|
|
int n;
|
|
|
|
n = n_oneshot.load();
|
|
cout << "Oneshots: " << n << endl;
|
|
|
|
if (n != 1)
|
|
{
|
|
cerr << "Expected 1 oneshots, got " << n << "." << endl;
|
|
rc = EXIT_FAILURE;
|
|
}
|
|
|
|
n = n_repeating.load();
|
|
cout << "Repeating: " << n << endl;
|
|
|
|
// Let's check that the task removal really had an effect.
|
|
sleep(2);
|
|
|
|
if (n != n_repeating.load())
|
|
{
|
|
cerr << "Removed task was called." << endl;
|
|
rc = EXIT_FAILURE;
|
|
}
|
|
|
|
// Timing involved, so we allow for some non-determinism.
|
|
if (n < 3 || n > 5)
|
|
{
|
|
cerr << "Expected ~4 repeating, got " << n << "." << endl;
|
|
rc = EXIT_FAILURE;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
int rc = EXIT_FAILURE;
|
|
|
|
init_test_env();
|
|
|
|
maxscale::MainWorker mw;
|
|
mw.start();
|
|
|
|
rc = test();
|
|
|
|
mw.shutdown();
|
|
mw.join();
|
|
|
|
return rc;
|
|
}
|