org.tinygroup.tinypc.impl.WorkQueueImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.pc Show documentation
There is a newer version: 2.2.3
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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 org.tinygroup.tinypc.impl;

import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;
import org.tinygroup.tinypc.*;
import java.io.IOException;
import java.rmi.RemoteException;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by luoguo on 14-1-23.
 */
public class WorkQueueImpl implements WorkQueue {
    /**
     *
     */
    private static final long serialVersionUID = -6270528078815279562L;
    private static final transient Logger LOGGER = LoggerFactory.getLogger(WorkQueueImpl.class);
    private List workList = new ArrayList();
    private transient ObjectStorage objectStorage = null;
  

    public WorkQueueImpl() throws RemoteException {

    }

    public synchronized ObjectStorage getObjectStorage() {
        return objectStorage;
    }

    public synchronized void setObjectStorage(ObjectStorage objectStorage) {
        this.objectStorage = objectStorage;
        try {
            List works = objectStorage.loadObjects("Work");
            this.workList.addAll(works);
        } catch (Exception e) {
            LOGGER.error(e);
        }
    }

    public synchronized void add(Work work) throws RemoteException {
        if (work.isNeedSerialize()) {
            if (objectStorage == null) {
                throw new PCRuntimeException("没有对象仓库对象实例存在！");
            }
            try {
                objectStorage.saveObject(work, "Work");
            } catch (IOException e) {
                LOGGER.error(e);
                throw new PCRuntimeException(String.format("序列化Work:%s %s时出现异常", work.getType(), work.getId()), e);
            }
        }
        workList.add(work);
    }

    public synchronized void updateWorkStatus(Work work, WorkStatus workStatus) throws RemoteException {
        for (Work w : workList) {
            if (w.getId().equals(work.getId())) {
                w.setWorkStatus(workStatus);
            }
        }
    }

    public synchronized WorkStatus getWorkStatus(Work work) throws RemoteException {
        for (Work w : workList) {
            if (w.getId().equals(work.getId())) {
                return work.getWorkStatus();
            }
        }
        throw new PCRuntimeException(String.format("找不到查询的工作:%s-%s！", work.getType(), work.getId()));
    }

    public synchronized void remove(Work work) throws RemoteException {
        workList.remove(work);
    }

    public synchronized int size() throws RemoteException {
        return workList.size();
    }

    public synchronized List getWorkList() throws RemoteException {
        return workList;
    }

    public synchronized Work getIdleWork() throws RemoteException {
        for (Work work : workList) {
            if (work.getWorkStatus().equals(WorkStatus.WAITING)) {
                return work;
            }
        }
        return null;
    }

    public synchronized List getWorkList(String type, WorkStatus workStatus) {
        List list = new ArrayList();
        for (Work work : workList) {
            if (work.getType().equals(type) && work.getWorkStatus().equals(workStatus)) {
                list.add(work);
            }
        }
        return list;
    }

    public synchronized List getWorkList(WorkStatus workStatus) {
        List list = new ArrayList();
        for (Work work : workList) {
            if (work.getWorkStatus().equals(workStatus)) {
                list.add(work);
            }
        }
        return list;
    }

    public synchronized void replace(Work oldWork, Work newWork) throws RemoteException {
        remove(oldWork);
        add(newWork);
    }

    public void moveToLast(Work work) {
        workList.remove(work);
        workList.add(work);
    }
}