com.moilioncircle.redis.replicator.RedisAofReplicator.java Source code

Introduction

Here is the source code for com.moilioncircle.redis.replicator.RedisAofReplicator.java

Source

/*
 * Copyright 2016 leon chen
 *
 * 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.moilioncircle.redis.replicator;

import com.moilioncircle.redis.replicator.cmd.Command;
import com.moilioncircle.redis.replicator.cmd.CommandName;
import com.moilioncircle.redis.replicator.cmd.CommandParser;
import com.moilioncircle.redis.replicator.cmd.ReplyParser;
import com.moilioncircle.redis.replicator.io.RedisInputStream;
import com.moilioncircle.redis.replicator.rdb.RdbFilter;
import com.moilioncircle.redis.replicator.rdb.RdbListener;
import com.moilioncircle.redis.replicator.rdb.datatype.KeyValuePair;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.io.*;
import java.util.Arrays;
import java.util.concurrent.ArrayBlockingQueue;

/**
 * Created by leon on 11/21/16.
 */
public class RedisAofReplicator extends AbstractReplicator {

    private static final Log logger = LogFactory.getLog(RedisAofReplicator.class);

    private final ReplyParser replyParser;

    public RedisAofReplicator(File file, Configuration configuration) throws FileNotFoundException {
        this(new FileInputStream(file), configuration);
    }

    public RedisAofReplicator(InputStream in, Configuration configuration) {
        this.configuration = configuration;
        this.inputStream = new RedisInputStream(in, this.configuration.getBufferSize());
        this.eventQueue = new ArrayBlockingQueue<>(this.configuration.getEventQueueSize());
        this.replyParser = new ReplyParser(inputStream);
        builtInCommandParserRegister();
        addExceptionListener(new DefaultExceptionListener());
    }

    @Override
    public void open() throws IOException {
        try {
            doOpen();
        } finally {
            close();
        }
    }

    private void doOpen() {
        worker.start();
        while (true) {
            try {
                Object obj = replyParser.parse();

                if (obj instanceof Object[]) {
                    if (configuration.isVerbose() && logger.isDebugEnabled())
                        logger.debug(Arrays.deepToString((Object[]) obj));

                    Object[] command = (Object[]) obj;
                    CommandName cmdName = CommandName.name((String) command[0]);
                    Object[] params = new Object[command.length - 1];
                    System.arraycopy(command, 1, params, 0, params.length);

                    final CommandParser<? extends Command> operations;
                    //if command do not register. ignore
                    if ((operations = commands.get(cmdName)) == null)
                        continue;

                    //do command replyParser
                    Command parsedCommand = operations.parse(cmdName, params);

                    //submit event
                    this.submitEvent(parsedCommand);
                } else {
                    logger.info("Redis reply:" + obj);
                }
            } catch (IOException | InterruptedException e) {
                break;
            }
        }
    }

    @Override
    public void close() throws IOException {
        doClose();
    }

    @Override
    public void doRdbHandler(KeyValuePair<?> kv) {
        throw new UnsupportedOperationException();
    }

    @Override
    public boolean doRdbFilter(KeyValuePair<?> kv) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void doPreFullSync() {
        throw new UnsupportedOperationException();
    }

    @Override
    public void doPostFullSync(final long checksum) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void addRdbFilter(RdbFilter filter) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void removeRdbFilter(RdbFilter filter) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void addRdbListener(RdbListener listener) {
        throw new UnsupportedOperationException();
    }

    @Override
    public void removeRdbListener(RdbListener listener) {
        throw new UnsupportedOperationException();
    }
}