org.joyqueue.convert.NsrPartitionGroupConverter Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2019 The JoyQueue 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 org.joyqueue.convert;

import org.joyqueue.domain.PartitionGroup;
import org.joyqueue.model.domain.Namespace;
import org.joyqueue.model.domain.PartitionGroupReplica;
import org.joyqueue.model.domain.Topic;
import org.joyqueue.model.domain.TopicPartitionGroup;

import java.util.Arrays;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;

/**
 * Created by wangxiaofei1 on 2019/1/3.
 */
public class NsrPartitionGroupConverter extends Converter {
    @Override
    protected PartitionGroup forward(TopicPartitionGroup topicPartitionGroup) {
        PartitionGroup partitionGroup = new PartitionGroup();
        partitionGroup.setTopic(CodeConverter.convertTopic(topicPartitionGroup.getNamespace(),topicPartitionGroup.getTopic()));
        partitionGroup.setElectType(PartitionGroup.ElectType.valueOf(topicPartitionGroup.getElectType()));
        partitionGroup.setGroup(topicPartitionGroup.getGroupNo());
        if (topicPartitionGroup.getPartitionSet() != null && topicPartitionGroup.getPartitionSet().size() > 0) {
            Set partitionSet = topicPartitionGroup.getPartitionSet();
            partitionGroup.setPartitions(partitionSet.stream().map( partition-> Short.valueOf(String.valueOf(partition))).collect(Collectors.toSet()));
        }
        if (topicPartitionGroup.getReplicaGroups() != null) {
            int leader = -1;
            Set learners = new TreeSet<>();
            Set replicaGroups = new TreeSet<>();
            for(PartitionGroupReplica replica : topicPartitionGroup.getReplicaGroups()){
                replicaGroups.add(replica.getBrokerId());
                if(replica.getRole()==PartitionGroupReplica.ROLE_LEARNER)learners.add(replica.getBrokerId());
                else if(replica.getRole()==PartitionGroupReplica.ROLE_MASTER)leader = replica.getBrokerId();
            }
            partitionGroup.setLearners(learners);
            partitionGroup.setLeader(leader);
        }
        partitionGroup.setRecLeader(topicPartitionGroup.getRecLeader());
        return partitionGroup;
    }

    @Override
    protected TopicPartitionGroup backward(PartitionGroup partitionGroup) {
        TopicPartitionGroup topicPartitionGroup = new TopicPartitionGroup();
        topicPartitionGroup.setId(partitionGroup.getId());
        topicPartitionGroup.setElectType(partitionGroup.getElectType().type());
        topicPartitionGroup.setGroupNo(partitionGroup.getGroup());
        topicPartitionGroup.setIsr(partitionGroup.getIsrs());
        topicPartitionGroup.setLeader(partitionGroup.getLeader());
        if (partitionGroup.getPartitions() != null) {
            topicPartitionGroup.setPartitions(Arrays.toString(partitionGroup.getPartitions().toArray()));
        }
        topicPartitionGroup.setReplicas(partitionGroup.getReplicas());
        topicPartitionGroup.setRecLeader(partitionGroup.getRecLeader());
        topicPartitionGroup.setLearners(partitionGroup.getLearners());
        topicPartitionGroup.setTerm(partitionGroup.getTerm());
        topicPartitionGroup.setTopic(new Topic(partitionGroup.getTopic().getCode()));
        topicPartitionGroup.setNamespace(new Namespace(partitionGroup.getTopic().getNamespace()));
        return topicPartitionGroup;
    }
}