// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License..

#include "NetworkManagerServer.h"

NetworkManagerServer* NetworkManagerServer::instance = NULL;

NetworkManagerServer::NetworkManagerServer() {}


void NetworkManagerServer::Init() {
    this->server = new Server(this->io_service, this->port);
}


NetworkManagerServer* NetworkManagerServer::getInstance(int port) {
    if (instance == NULL) {
        instance = new NetworkManagerServer();
        instance->setPort(port);
    }

    return instance;
}


void NetworkManagerServer::startService() {
    this->server->start_accept();
    this->io_service.run();
}


void NetworkManagerServer::connectCallbackHandler(CallbackHandler cb) {
    this->server->connectCallbackHandler(cb);
}

