com.baidu.jprotobuf.pbrpc.registry.AsyncRegistryService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jprotobuf-rpc-core Show documentation

The core module of jprotobuf rpc socket implementation

There is a newer version: 4.2.2

/*
 * Copyright 2002-2014 the original author or authors.
 *
 * 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.
 */
package com.baidu.jprotobuf.pbrpc.registry;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingDeque;

import com.baidu.jprotobuf.pbrpc.registry.RegisterInfo;
import com.baidu.jprotobuf.pbrpc.registry.RegistryCenterService;

/**
 * Asynchronous register supports for {@link RegistryCenterService}
 *
 * @author xiemalin
 * @since 2.27
 */
public abstract class AsyncRegistryService implements RegistryCenterService {
    
    private LinkedBlockingDeque asyncQueue = new LinkedBlockingDeque(100);
    
    private boolean stop;
    
    private ExecutorService es = Executors.newFixedThreadPool(1);
    
    /**
     * get the stop
     * @return the stop
     */
    public boolean isStop() {
        return stop;
    }
    
    public void stop() {
        stop = true;
        es.shutdown();
    }

    /**
     * set stop value to stop
     * @param stop the stop to set
     */
    public void setStop(boolean stop) {
        this.stop = stop;
    }

    /**
     * 
     */
    public AsyncRegistryService() {
        
        es.execute(new Runnable() {
            
            @Override
            public void run() {
                while (!stop) {
                    RegisterInfo registerInfo  = null;
                    try {
                        registerInfo = asyncQueue.take();
                        doRegister(registerInfo);
                    } catch (Exception e) {
                        // any exception do retry
                        if (registerInfo != null) {
                            asyncQueue.addLast(registerInfo);
                        }
                    }
                }
            }
        });
        
    }

    /* (non-Javadoc)
     * @see com.baidu.jprotobuf.pbrpc.registry.RegistryCenterService#register(com.baidu.jprotobuf.pbrpc.registry.RegisterInfo)
     */
    @Override
    public final void register(RegisterInfo url) {
        asyncQueue.add(url);
    }


    protected abstract void doRegister(RegisterInfo url);

}