(35)定期拉取最新路由信息(V0.6)(LoadBalanceAgent部分)-【Lars-基于C++负载均衡远程服务器调度系统教程】

【Lars教程目录】

Lars源代码
https://github.com/aceld/Lars


【Lars系统概述】
第1章-概述
第2章-项目目录构建


【Lars系统之Reactor模型服务器框架模块】
第1章-项目结构与V0.1雏形
第2章-内存管理与Buffer封装
第3章-事件触发EventLoop
第4章-链接与消息封装
第5章-Client客户端模型
第6章-连接管理及限制
第7章-消息业务路由分发机制
第8章-链接创建/销毁Hook机制
第9章-消息任务队列与线程池
第10章-配置文件读写功能
第11章-udp服务与客户端
第12章-数据传输协议protocol buffer
第13章-QPS性能测试
第14章-异步消息任务机制
第15章-链接属性设置功能


【Lars系统之DNSService模块】
第1章-Lars-dns简介
第2章-数据库创建
第3章-项目目录结构及环境构建
第4章-Route结构的定义
第5章-获取Route信息
第6章-Route订阅模式
第7章-Backend Thread实时监控


【Lars系统之Report Service模块】
第1章-项目概述-数据表及proto3协议定义
第2章-获取report上报数据
第3章-存储线程池及消息队列


【Lars系统之LoadBalance Agent模块】
第1章-项目概述及构建
第2章-主模块业务结构搭建
第3章-Report与Dns Client设计与实现
第4章-负载均衡模块基础设计
第5章-负载均衡获取Host主机信息API
第6章-负载均衡上报Host主机信息API
第7章-过期窗口清理与过载超时(V0.5)
第8章-定期拉取最新路由信息(V0.6)
第9章-负载均衡获取Route信息API(0.7)
第10章-API初始化接口(V0.8)
第11章-Lars Agent性能测试工具
第12章- Lars启动工具脚本


9) 定期拉取最新路由信息(V0.6)

  1. 当agent首次连接到dns service时,将全部的load_balance均设置为NEW状态。如果dns service重新启动,或者断开链接重连,都会将之前的拉去中或者没拉取的load_balance状态都设置为NEW状态,只有NEW状态的load_balance才能定期自动拉取.

lars_loadbalance_agent/src/dns_client.cpp

//========================================================
static void conn_init(net_connection *conn, void *args)
{
    for (int i = 0; i < 3; i++) {
        r_lb[i]->reset_lb_status();
    }
}
//========================================================
void *dns_client_thread(void* args)
{
    printf("dns client thread start\n");
    event_loop loop;

    //1 加载配置文件得到dns service ip + port
    std::string ip = config_file::instance()->GetString("dnsserver", "ip", "");
    short port = config_file::instance()->GetNumber("dnsserver", "port", 0);

    //2 创建客户端
    tcp_client client(&loop, ip.c_str(), port, "dns client");


    //3 将thread_queue消息回调事件,绑定到loop中
    dns_queue->set_loop(&loop);
    dns_queue->set_callback(new_dns_request, &client);

    //4 设置当收到dns service回执的消息ID_GetRouteResponse处理函数
    client.add_msg_router(lars::ID_GetRouteResponse, deal_recv_route);

  //========================================================
    //5.设置链接成功/链接断开重连成功之后,通过conn_init来清理之前的任务
    client.set_conn_start(conn_init);
  //========================================================
  
    //启动事件监听
    loop.event_process(); 

    return NULL;
}

lars_loadbalance_agent/src/route_lb.cpp

//将全部的load_balance都重置为NEW状态
void route_lb::reset_lb_status()
{
    pthread_mutex_lock(&_mutex);
    for (route_map_it it = _route_lb_map.begin();
        it != _route_lb_map.end(); it++) {

        load_balance *lb = it->second;
        if (lb->status == load_balance::PULLING) {
            lb->status = load_balance::NEW;
        }
    }
    pthread_mutex_unlock(&_mutex);
}
  1. 增加配置文件参数update_timeout, 表示一个NEW状态的load_balance下的modid/cmdid节点应该经历多长时间进行一次刷新拉取。

lars_loadbalance_agent/conf/lars_lb_agent.conf

;对于每个NEW状态的modid/cmdid,多久更新一下本地路由,秒
update_timeout=15

lars_loadbalance_agent/include/main_server.h

struct load_balance_config
{
    //...

    //对于每个NEW状态的modid/cmdid,多久更新一下本地路由,秒
    long update_timeout;
};

lars_loadbalance_agent/src/main_server.cpp

static void init_lb_agent()
{
    //1. 加载配置文件
    config_file::setPath("./conf/lars_lb_agent.conf");
    //...
    //...
  
    lb_config.update_timeout = config_file::instance()->GetNumber("loadbalance", "update_timeout", 15);

    //...
}
  1. load_balance设置最后update时间参数,及最后一次从dns service拉取下来更新host_map的时间.然后在route_lb每次执行get_host的时候,对每个已经存在的host节点做最后时间超时检查,如果超时,则重新从 dns service中拉取。

lars_loadbalance_agent/include/load_balance.h

/*
 * 负载均衡算法核心模块
 * 针对一组(modid/cmdid)下的全部host节点的负载规则
 */
class load_balance {
public:
    load_balance(int modid, int cmdid):
        status(PULLING),
        last_update_time(0),
        _modid(modid),
        _cmdid(cmdid)
    {
        //load_balance 初始化构造
    }

        // ...
    
    long last_update_time; //最后更新host_map时间戳
    
private:
        // ...
};

lars_loadbalance_agent/src/load_balance.cpp

//根据dns service远程返回的结果,更新_host_map
void load_balance::update(lars::GetRouteResponse &rsp)
{
    long current_time = time(NULL);

    //...
    //...

    //更新最后update时间
    last_update_time = current_time;
    //重置状态为NEW
    status = NEW;
}

load_balance每次调用update()都记录一次最后的更新时间,并标记为NEW表示当前modid/cmdid没有在PULLING,可以再更新。

lars_loadbalance_agent/src/route_lb.cpp

//agent获取一个host主机,将返回的主机结果存放在rsp中
int route_lb::get_host(int modid, int cmdid, lars::GetHostResponse &rsp)
{
    int ret = lars::RET_SUCC;

    //1. 得到key
    uint64_t key = ((uint64_t)modid << 32) + cmdid;

    pthread_mutex_lock(&_mutex);
    //2. 当前key已经存在_route_lb_map中
    if (_route_lb_map.find(key) != _route_lb_map.end()) {
        //2.1 取出对应的load_balance
        load_balance *lb = _route_lb_map[key];

        if (lb->empty() == true) {
            //存在lb 里面的host为空,说明正在pull()中,还没有从dns_service返回来,所以直接回复不存在
            assert(lb->status == load_balance::PULLING);
            rsp.set_retcode(lars::RET_NOEXIST);
        }
        else {
            ret = lb->choice_one_host(rsp);
            rsp.set_retcode(ret);
// =================================================
            //超时重拉路由
            //检查是否要重新拉路由信息
            //若路由并没有处于PULLING状态,且有效期已经超时,则重新拉取
            if (lb->status == load_balance::NEW && time(NULL) - lb->last_update_time > lb_config.update_timeout) {
                lb->pull();
            }
// =================================================
        }
    }
    //3. ...
    // ...
  
    pthread_mutex_unlock(&_mutex);

    return ret;
}

关于作者:

作者:Aceld(刘丹冰)

mail: danbing.at@gmail.com
github: https://github.com/aceld
原创书籍gitbook: http://legacy.gitbook.com/@aceld

原创声明:未经作者允许请勿转载, 如果转载请注明出处