/*
 *
 * Copyright (c) 2006 The Apache Software Foundation
 *
 * Licensed 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 "qpid/broker/DeliveryRecord.h"
#include "qpid/broker/Channel.h"

using namespace qpid::broker;

DeliveryRecord::DeliveryRecord(Message::shared_ptr _msg, 
                               Queue::shared_ptr _queue, 
                               const string _consumerTag, 
                               const u_int64_t _deliveryTag) : msg(_msg), 
                                                               queue(_queue), 
                                                               consumerTag(_consumerTag),
                                                               deliveryTag(_deliveryTag),
                                                               pull(false){}

DeliveryRecord::DeliveryRecord(Message::shared_ptr _msg, 
                               Queue::shared_ptr _queue, 
                               const u_int64_t _deliveryTag) : msg(_msg), 
                                                               queue(_queue), 
                                                               consumerTag(""),
                                                               deliveryTag(_deliveryTag),
                                                               pull(true){}


void DeliveryRecord::discard(TransactionContext* ctxt) const{
    queue->dequeue(ctxt, msg, 0);
}

bool DeliveryRecord::matches(u_int64_t tag) const{
    return deliveryTag == tag;
}

bool DeliveryRecord::coveredBy(const AccumulatedAck* const range) const{
    return range->covers(deliveryTag);
}

void DeliveryRecord::redeliver(Channel* const channel) const{
    if(pull){
        //if message was originally sent as response to get, we must requeue it
        requeue();
    }else{
        channel->deliver(msg, consumerTag, deliveryTag);
    }
}

void DeliveryRecord::requeue() const{
    msg->redeliver();
    queue->deliver(msg);
}

void DeliveryRecord::addTo(Prefetch* const prefetch) const{
    if(!pull){
        //ignore 'pulled' messages (i.e. those that were sent in
        //response to get) when calculating prefetch
        prefetch->size += msg->contentSize();
        prefetch->count++;
    }    
}

void DeliveryRecord::subtractFrom(Prefetch* const prefetch) const{
    if(!pull){
        //ignore 'pulled' messages (i.e. those that were sent in
        //response to get) when calculating prefetch
        prefetch->size -= msg->contentSize();
        prefetch->count--;
    }
}
