Method SetFlowLimit

src/dataserver/flow_controller_impl.cpp:370–421

}

bool FlowControllerImpl::SetFlowLimit(int ns, FlowType type, int64_t lower, int64_t upper) {
    log_debug("set(or get) flow limit bound, ns %d; type %s; lower %ld; upper %ld",
              ns, FlowTypeStr(type), lower, upper);
    if (lower < -1 || upper < -1) {
        log_error("invliad flow limit parameter, lower %ld; upper %ld", lower, upper);
        return false;
    }
    if (ns == -1) {
        bool success = true;
        for (size_t i = 0; i < MAXAREA; ++i)
            if (SetFlowLimit(i, type, lower, upper) == false)
                success = false;
        return success;
    } else {
        FlowLimit limit = GetFlowLimit(ns, type);
        if (upper != -1)
            limit.upper = upper;
        if (lower != -1)
            limit.lower = lower;

        if (limit.lower > limit.upper) {
            log_error("%ld > %ld is invalid", limit.lower, limit.upper);
            return false;
        }

        if (ns < 0 || ns > MAXAREA) ns = MAXAREA;
        switch (type) {
            case IN:
                if (lower >= 0)
                    flows_[ns].in.lower_bound = lower;
                if (upper >= 0)
                    flows_[ns].in.upper_bound = upper;
                break;
            case OUT:
                if (lower >= 0)
                    flows_[ns].out.lower_bound = lower;
                if (upper >= 0)
                    flows_[ns].out.upper_bound = upper;
                break;
            case OPS:
                if (lower >= 0)
                    flows_[ns].ops.lower_bound = lower;
                if (upper >= 0)
                    flows_[ns].ops.upper_bound = upper;
                break;
            default:
                log_error("unknow type %d", type);
                break;
        }
    }
    return true;
}

void FlowControllerImpl::SetAreaFlowPolicy(int ns, bool flag) {
    area_flow_policy_.set(ns, flag);