io.journalkeeper.core.state.InternalState Maven / Gradle / Ivy

Go to download
/**
 * 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.
 * 

 * 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.
 */
/**
 * 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 io.journalkeeper.core.state;

import io.journalkeeper.core.journal.JournalSnapshot;
import io.journalkeeper.utils.ThreadSafeFormat;

import java.net.URI;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * JournalKeeper保留状态
 * @author LiYue
 * Date: 2019/11/20
 */
public class InternalState implements JournalSnapshot {
    private ConfigState configState;
    private URI preferredLeader = null;
    private Map partitionIndices;
    private long lastIncludedIndex;
    private int lastIncludedTerm;
    private long minOffset;
    private long snapshotTimestamp = System.currentTimeMillis();


    public InternalState() {
    }

    public InternalState(ConfigState configState, Set partitions, URI preferredLeader) {
        this.configState = configState;
        this.partitionIndices = new HashMap<>(partitions.size());
        for (Integer partition : partitions) {
            partitionIndices.put(partition, 0L);
        }
        this.preferredLeader = preferredLeader;
        this.lastIncludedIndex = -1L;
        this.lastIncludedTerm = -1;
        this.minOffset = 0;

    }

    public URI getPreferredLeader() {
        return preferredLeader;
    }

    public void setPreferredLeader(URI preferredLeader) {
        this.preferredLeader = preferredLeader;
    }

    public Set getPartitions() {
        return Collections.unmodifiableSet(partitionIndices.keySet());
    }

    public void setPartitions(Set partitions) {
        Map copyOnWriteMap = new HashMap<>();
        for (Integer partition : partitions) {
            copyOnWriteMap.put(partition, partitionIndices.getOrDefault(partition, 0L));
        }
        partitionIndices = copyOnWriteMap;
    }

    public ConfigState getConfigState() {
        return configState;
    }

    public void setConfigState(ConfigState configState) {
        this.configState = configState;
    }

    public long getLastIncludedIndex() {
        return lastIncludedIndex;
    }

    public void setLastIncludedIndex(long lastIncludedIndex) {
        this.lastIncludedIndex = lastIncludedIndex;
    }

    public int getLastIncludedTerm() {
        return lastIncludedTerm;
    }

    public void setLastIncludedTerm(int lastIncludedTerm) {
        this.lastIncludedTerm = lastIncludedTerm;
    }

    public void next() {
        lastIncludedIndex++;
    }

    @Override
    public Map partitionMinIndices() {
        return Collections.unmodifiableMap(partitionIndices);
    }

    @Override
    public long minIndex() {
        return lastIncludedIndex + 1;
    }

    @Override
    public long minOffset() {
        return minOffset;
    }

    public long getSnapshotTimestamp() {
        return snapshotTimestamp;
    }

    public void setSnapshotTimestamp(long snapshotTimestamp) {
        this.snapshotTimestamp = snapshotTimestamp;
    }

    public Map getPartitionIndices() {
        return Collections.unmodifiableMap(partitionIndices);
    }

    public void setPartitionIndices(Map partitionIndices) {
        this.partitionIndices = new HashMap<>(partitionIndices);
    }

    public long getMinOffset() {
        return minOffset;
    }

    public void setMinOffset(long minOffset) {
        this.minOffset = minOffset;
    }

    @Override
    public String toString() {
        return "InternalState{" +
                "configState=" + configState +
                ", preferredLeader=" + preferredLeader +
                ", partitionIndices=" + partitionIndices +
                ", lastIncludedIndex=" + lastIncludedIndex +
                ", lastIncludedTerm=" + lastIncludedTerm +
                ", minOffset=" + minOffset +
                ", snapshotTimestamp=" + ThreadSafeFormat.format(new Date(snapshotTimestamp)) +
                '}';
    }
}