com.alibaba.napoli.metamorphosis.client.consumer.FetchRequest.java Source code

Introduction

Here is the source code for com.alibaba.napoli.metamorphosis.client.consumer.FetchRequest.java

Source

/*
 * (C) 2007-2012 Alibaba Group Holding Limited.
 * 
 * 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.
 * Authors:
 *   wuhua <wq163@163.com> , boyan <killme2008@gmail.com>
 */
package com.alibaba.napoli.metamorphosis.client.consumer;

import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.alibaba.napoli.metamorphosis.cluster.Broker;
import com.alibaba.napoli.metamorphosis.cluster.Partition;
import com.alibaba.napoli.metamorphosis.utils.MessageUtils;

/**
 * ?
 * 
 * @author boyan
 * @Date 2011-4-25
 * 
 */
public class FetchRequest implements Delayed {
    // ?
    private long delayTimeStamp;
    private long delay;
    private TopicPartitionRegInfo topicPartitionRegInfo;
    private int maxSize;
    private int originalMaxSize;
    private Broker broker;
    private int retries = 0;
    private long tmpOffset;

    /**
     * Just for test
     * 
     * @param delay
     */
    FetchRequest(final long delay) {
        super();
        this.delay = delay;
        this.delayTimeStamp = System.currentTimeMillis() + delay;
    }

    public int getRetries() {
        return this.retries;
    }

    static final Log log = LogFactory.getLog(FetchRequest.class);

    public void increaseMaxSize() {
        if (this.maxSize > MessageUtils.MAX_READ_BUFFER_SIZE) {
            log.warn("maxSize?" + MessageUtils.MAX_READ_BUFFER_SIZE
                    + "Bytes??-Dnotify.remoting.max_read_buffer_size?");
            return;
        }
        this.maxSize = 2 * this.maxSize;
    }

    public void decreaseMaxSize() {
        if (this.maxSize < this.originalMaxSize) {
            this.maxSize = this.originalMaxSize;
        } else if (this.maxSize == this.originalMaxSize) {
            return;
        } else {
            this.maxSize = this.maxSize / 2;
        }
    }

    public void resetRetries() {
        this.retries = 0;
    }

    public int incrementRetriesAndGet() {
        return ++this.retries;
    }

    public FetchRequest(final Broker broker, final long delay, final TopicPartitionRegInfo topicPartitionRegInfo,
            final int maxSize) {
        super();
        this.broker = broker;
        this.delay = delay;
        if (delay >= 0) {
            this.delayTimeStamp = System.currentTimeMillis() + delay;
        }
        this.topicPartitionRegInfo = topicPartitionRegInfo;
        this.maxSize = maxSize;
        this.originalMaxSize = maxSize;
        if (this.maxSize <= 0) {
            throw new IllegalArgumentException("maxSize <=0");
        }
    }

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + (this.broker == null ? 0 : this.broker.hashCode());
        result = prime * result + (int) (this.delay ^ this.delay >>> 32);
        result = prime * result + this.maxSize;
        result = prime * result + this.retries;
        result = prime * result + (this.topicPartitionRegInfo == null ? 0 : this.topicPartitionRegInfo.hashCode());
        return result;
    }

    @Override
    public boolean equals(final Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (this.getClass() != obj.getClass()) {
            return false;
        }
        final FetchRequest other = (FetchRequest) obj;
        if (this.broker == null) {
            if (other.broker != null) {
                return false;
            }
        } else if (!this.broker.equals(other.broker)) {
            return false;
        }
        if (this.delay != other.delay) {
            return false;
        }
        if (this.maxSize != other.maxSize) {
            return false;
        }
        if (this.retries != other.retries) {
            return false;
        }
        if (this.topicPartitionRegInfo == null) {
            if (other.topicPartitionRegInfo != null) {
                return false;
            }
        } else if (!this.topicPartitionRegInfo.equals(other.topicPartitionRegInfo)) {
            return false;
        }
        return true;
    }

    public Broker getBroker() {
        return this.broker;
    }

    public void setBroker(final Broker broker) {
        this.broker = broker;
    }

    /**
     * ???
     * 
     * @param delay
     */
    public void setDelay(final long delay) {
        this.delay = delay;
        this.delayTimeStamp = System.currentTimeMillis() + delay;
    }

    @Override
    public int compareTo(final Delayed o) {
        if (o == this) {
            return 0;
        }
        final FetchRequest other = (FetchRequest) o;
        final long sub = this.delayTimeStamp - other.delayTimeStamp;
        return sub <= 0 ? -1 : 1;
    }

    public long getDelay() {
        return this.delay;
    }

    @Override
    public long getDelay(final TimeUnit unit) {
        return unit.convert(this.delayTimeStamp - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
    }

    public int getMaxSize() {
        return this.maxSize;
    }

    /**
     * offsetacktrueoffsetoffset?1,?offset
     * 
     * @param offset
     * @param ack
     */
    public void setOffset(final long offset, final long msgId, final boolean ack) {
        if (ack) {
            // topicPartitionRegInfo???zk?
            synchronized (this.topicPartitionRegInfo) {
                this.topicPartitionRegInfo.getOffset().set(offset);
                if (msgId > 0) {
                    this.topicPartitionRegInfo.setMessageId(msgId);
                }
                // ??storage
                this.topicPartitionRegInfo.setModified(true);
            }
            this.rollbackOffset();
        } else {
            this.tmpOffset = offset;
        }
    }

    public TopicPartitionRegInfo getTopicPartitionRegInfo() {
        return this.topicPartitionRegInfo;
    }

    public String getTopic() {
        return this.topicPartitionRegInfo.getTopic();
    }

    public int getPartition() {
        return this.topicPartitionRegInfo.getPartition().getPartition();
    }

    public Partition getPartitionObject() {
        return this.topicPartitionRegInfo.getPartition();

    }

    public void rollbackOffset() {
        this.tmpOffset = -1;
    }

    public long getLastMessageId() {
        return this.topicPartitionRegInfo.getMessageId();
    }

    /**
     * ?offsetoffsetoffset
     * 
     * @return
     */
    public long getOffset() {
        if (this.tmpOffset > 0) {
            return this.tmpOffset;
        } else {
            return this.topicPartitionRegInfo.getOffset().get();
        }
    }

}