com.moilioncircle.redis.replicator.rdb.skip.SkipRdbParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of redis-replicator Show documentation
Redis Replicator implement Redis Replication protocol written in java. It can parse,filter,broadcast the RDB and AOF events in a real time manner. It also can synchronize redis data to your local cache or to database.
There is a newer version: 3.8.1
Show newest version
/*
 * Copyright 2016-2018 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.rdb.skip;

import com.moilioncircle.redis.replicator.io.RedisInputStream;
import com.moilioncircle.redis.replicator.rdb.BaseRdbParser;

import java.io.IOException;

import static com.moilioncircle.redis.replicator.Constants.RDB_ENC_INT16;
import static com.moilioncircle.redis.replicator.Constants.RDB_ENC_INT32;
import static com.moilioncircle.redis.replicator.Constants.RDB_ENC_INT8;
import static com.moilioncircle.redis.replicator.Constants.RDB_ENC_LZF;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_DOUBLE;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_EOF;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_FLOAT;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_SINT;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_STRING;
import static com.moilioncircle.redis.replicator.Constants.RDB_MODULE_OPCODE_UINT;

/**
 * @author Leon Chen
 * @since 2.4.6
 */
public class SkipRdbParser {
    
    protected final RedisInputStream in;
    
    public SkipRdbParser(RedisInputStream in) {
        this.in = in;
    }
    
    public void rdbLoadTime() throws IOException {
        in.skip(4);
    }
    
    public void rdbLoadMillisecondTime() throws IOException {
        in.skip(8);
    }
    
    public BaseRdbParser.Len rdbLoadLen() throws IOException {
        return new BaseRdbParser(in).rdbLoadLen();
    }
    
    public void rdbLoadIntegerObject(int enctype) throws IOException {
        switch (enctype) {
            case RDB_ENC_INT8:
                in.skip(1);
                break;
            case RDB_ENC_INT16:
                in.skip(2);
                break;
            case RDB_ENC_INT32:
                in.skip(4);
                break;
            default:
                break;
        }
    }
    
    public void rdbLoadLzfStringObject() throws IOException {
        long clen = rdbLoadLen().len;
        rdbLoadLen();
        in.skip(clen);
    }
    
    public void rdbGenericLoadStringObject() throws IOException {
        BaseRdbParser.Len lenObj = rdbLoadLen();
        long len = (int) lenObj.len;
        boolean isencoded = lenObj.encoded;
        if (isencoded) {
            switch ((int) len) {
                case RDB_ENC_INT8:
                case RDB_ENC_INT16:
                case RDB_ENC_INT32:
                    rdbLoadIntegerObject((int) len);
                    return;
                case RDB_ENC_LZF:
                    rdbLoadLzfStringObject();
                    return;
                default:
                    throw new AssertionError("unknown RdbParser encoding type:" + len);
            }
        }
        in.skip(len);
    }
    
    public void rdbLoadPlainStringObject() throws IOException {
        rdbGenericLoadStringObject();
    }
    
    public void rdbLoadEncodedStringObject() throws IOException {
        rdbGenericLoadStringObject();
    }
    
    public void rdbLoadDoubleValue() throws IOException {
        int len = in.read();
        switch (len) {
            case 255:
            case 254:
            case 253:
                return;
            default:
                in.skip(len);
        }
    }
    
    public void rdbLoadBinaryDoubleValue() throws IOException {
        in.skip(8);
    }
    
    public float rdbLoadBinaryFloatValue() throws IOException {
        return in.skip(4);
    }
    
    public void rdbLoadCheckModuleValue() throws IOException {
        int opcode;
        while ((opcode = (int) rdbLoadLen().len) != RDB_MODULE_OPCODE_EOF) {
            if (opcode == RDB_MODULE_OPCODE_SINT || opcode == RDB_MODULE_OPCODE_UINT) {
                rdbLoadLen();
            } else if (opcode == RDB_MODULE_OPCODE_STRING) {
                rdbGenericLoadStringObject();
            } else if (opcode == RDB_MODULE_OPCODE_FLOAT) {
                rdbLoadBinaryFloatValue();
            } else if (opcode == RDB_MODULE_OPCODE_DOUBLE) {
                rdbLoadBinaryDoubleValue();
            }
        }
    }
}