com.chinamobile.bcbsp.bspcontroller.HRNQueue.java Source code

Introduction

Here is the source code for com.chinamobile.bcbsp.bspcontroller.HRNQueue.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

package com.chinamobile.bcbsp.bspcontroller;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;

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

/**
 * High response ratio job waitQueues
 * @author hadoop
 *
 */
public class HRNQueue implements Queue<JobInProgress> {
    /**handle log file in the class*/
    private static final Log LOG = LogFactory.getLog(HRNQueue.class);
    /**Queue name*/
    private final String name;
    /**jobinprogress queue*/
    private BlockingQueue<JobInProgress> queue = new LinkedBlockingQueue<JobInProgress>();
    /**tmp for queue resort*/
    private List<JobInProgress> resort_tmp = new ArrayList<JobInProgress>();
    /**jobs num counter*/
    private long counter;

    /**
     * HRNQueue construct method
     * @param name
     *        queue name
     */
    public HRNQueue(String name) {
        this.name = name;
        this.counter = 0;
    }

    @Override
    public String getName() {
        // TODO Auto-generated method stub
        return this.name;
    }

    @Override
    public void addJob(JobInProgress job) {
        // TODO Auto-generated method stub
        try {
            queue.put(job);
            this.counter++;
        } catch (InterruptedException ie) {
            //LOG.error("Fail to add a job to the " + this.name + " queue.", ie);
            throw new RuntimeException("Fail to add a job to the " + this.name + " queue.", ie);
        }
    }

    @Override
    public void resortQueue() {
        // TODO Auto-generated method stub
        Comparator<JobInProgress> comp = new Comparator<JobInProgress>() {
            @Override
            public int compare(JobInProgress o1, JobInProgress o2) {
                int res;
                if (o1.getHRN() > o2.getHRN()) {
                    res = -1;
                } else if (o1.getHRN() == o2.getHRN()) {
                    res = 0;
                } else {
                    res = 1;
                }
                return res;
            }
        };
        synchronized (queue) {
            try {
                resort_tmp.clear();
                int wait_count = queue.size();
                int i = 0;
                for (i = 0; i < wait_count; i++) {
                    resort_tmp.add(queue.take());
                }
                Collections.sort(resort_tmp, comp);
                for (i = 0; i < wait_count; i++) {
                    queue.put(resort_tmp.get(i));
                }
            } catch (Exception e) {
                //LOG.error("resort error: " + e.toString());
                throw new RuntimeException("resort error: ", e);
            }
        }
    }

    @Override
    public void removeJob(JobInProgress job) {
        // TODO Auto-generated method stub
        queue.remove(job);
        this.counter--;
    }

    @Override
    public JobInProgress removeJob() {
        // TODO Auto-generated method stub
        try {
            JobInProgress jip = queue.take();
            this.counter--;
            return jip;
        } catch (InterruptedException ie) {
            LOG.error("Fail to remove a job from the " + this.name + " queue.", ie);
        } catch (Exception e) {
            // TODO : Here is Exception when stop the JobProcessor Thread.
            LOG.error("Fail to remove a job.", e);
        }
        return null;
    }

    @Override
    public Collection<JobInProgress> getJobs() {
        // TODO Auto-generated method stub
        return queue;
    }

    /**
     * get the waitQueue size
     * @return num of jobs in the waitQueue
     */
    @Override
    public long getSize() {
        // TODO Auto-generated method stub
        return this.counter;
    }

    /**
     * if the waitQueue is empty
     * @return true empty,false not empty
     */
    @Override
    public boolean isEmpty() {
        // TODO Auto-generated method stub
        if (this.counter == 0L) {
            return true;
        } else {
            return false;
        }
    }

    /**
     * if the waitQueue contains a specific job
     * @param job
     *           specific job to tell
     * @return search result.
     */
    @Override
    public boolean contains(JobInProgress job) {
        // TODO Auto-generated method stub
        return queue.contains(job);
    }

    @Override
    public JobInProgress getFront() {
        // TODO Auto-generated method stub
        return queue.peek();
    }
}