vrf_assign_kstate.cc

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */

#include "kstate.h"
#include "vrf_assign_kstate.h"

VrfAssignKState::VrfAssignKState(KVrfAssignResp *obj,
                                 const std::string &resp_ctx,
                                 vr_vrf_assign_req &req, int id) :
    KState(resp_ctx, obj) {

    req.set_h_op(sandesh_op::DUMP);
    req.set_var_vif_index(id);
    req.set_var_marker(-1);
}

void VrfAssignKState::SendNextRequest() {
    VrfAssignContext *ctx = boost::any_cast<VrfAssignContext *>(more_context_);
    vr_vrf_assign_req req;

    req.set_h_op(sandesh_op::DUMP);
    req.set_var_vif_index(ctx->vif_index_);
    req.set_var_vlan_id(ctx->marker_);
    req.set_var_marker(ctx->marker_);
    EncodeAndSend(req);
}

void VrfAssignKState::Handler() {
    KVrfAssignResp *resp = static_cast<KVrfAssignResp *>(response_object_);
    if (resp) {
        if (MoreData()) {
            /* There are more labels in Kernel. We need to query them from
             * Kernel and send it to Sandesh.
             */
            SendResponse();
            SendNextRequest();
        } else {
            resp->set_context(response_context_);
            resp->Response();
            if (!more_context_.empty()) {
                VrfAssignContext *ctx =
                    boost::any_cast<VrfAssignContext *>(more_context_);
                if (ctx) {
                    delete ctx;
                    more_context_ = boost::any();
                }
            }
        }
    }
}

void VrfAssignKState::SendResponse() {
    KVrfAssignResp *resp = static_cast<KVrfAssignResp *>(response_object_);
    resp->set_context(response_context_);
    resp->set_more(true);
    resp->Response();

    response_object_ = new KVrfAssignResp();
}