#include "PolarisClient.h"
#include "PolarisTask.h"
#include "workflow/WFFacilities.h"
#include <signal.h>

using namespace polaris;
static WFFacilities::WaitGroup wait_group(1);
PolarisClient client;

void polaris_callback(PolarisTask *task) {
    int state = task->get_state();
    int error = task->get_error();
    if (state != WFT_STATE_SUCCESS) {
        fprintf(stderr, "Task error: %d\n", error);
        client.deinit();
        wait_group.done();
        return;
    }
    //get discover results and do something
    struct discover_result resp;
    if (!task->get_discover_result(&resp)) {
        fprintf(stderr, "get disocer_result error: %d\n", error);
        client.deinit();
        wait_group.done();
        return;
    }
    struct route_result route;
    if (!task->get_route_result(&route)) {
        fprintf(stderr, "get route_result error: %d\n", error);
        client.deinit();
        wait_group.done();
        return;
    }
    fprintf(stderr, "\nSuccess. Press Ctrl-C to exit.\n");
}

void sig_handler(int signo) { wait_group.done(); }

int main(int argc, char *argv[]) {
    PolarisTask *task;

    signal(SIGINT, sig_handler);
    std::string url = "http://your.polaris.cluster:8090";
    int ret = client.init(url);
    if (ret != 0) {
        client.deinit();
        exit(1);
    }
    task =
        client.create_discover_task("your.namespace", "your.service.name", 5, polaris_callback);

    PolarisConfig config;
    task->set_config(std::move(config));
    task->start();

    wait_group.wait();
    return 0;
}

#include "PolarisClient.h"
#include "PolarisTask.h"
#include "workflow/WFFacilities.h"
#include <signal.h>

using namespace polaris;
static WFFacilities::WaitGroup wait_group(1);
PolarisClient client;

void polaris_callback(PolarisTask *task) {
    int state = task->get_state();
    int error = task->get_error();
    if (state != WFT_STATE_SUCCESS) {
        fprintf(stderr, "Task error: %d\n", error);
        client.deinit();
        wait_group.done();
        return;
    }
    fprintf(stderr, "Task ok\n");
}

void sig_handler(int signo) { wait_group.done(); }

int main(int argc, char *argv[]) {
    if (argc < 2) {
        fprintf(stderr, "USAGE: %s [r/d]\n", argv[0]);
        exit(1);
    }
    PolarisTask *task;
    signal(SIGINT, sig_handler);
    std::string url = "http://your.polaris.cluster:8090";
    int ret = client.init(url);
    if (ret != 0) {
        exit(1);
    }
    PolarisConfig config;
    if (argv[1][0] == 'r') {
        task = client.create_register_task("your.namespace", "your.service_name", 5,
                                           polaris_callback);
        task->set_config(std::move(config));
        PolarisInstance instance;
        instance.set_host("your.instance.ip");
        instance.set_port(8080);
        std::map<std::string, std::string> meta = {{"key1", "value1"}};
        instance.set_metadata(meta);
        task->set_polaris_instance(std::move(instance));

    } else if (argv[1][0] == 'd') {
        task = client.create_deregister_task("your.namespace", "your.service_name", 5,
                                             polaris_callback);
        task->set_config(std::move(config));
        PolarisInstance instance;
        instance.set_host("your.instance.ip");
        instance.set_port(8080);
        task->set_polaris_instance(std::move(instance));

    } else {
        fprintf(stderr, "USAGE: %s [r/d]\n", argv[0]);
        exit(1);
    }
    task->start();
    wait_group.wait();
    return 0;
}