com.instaclustr.cassandra.sidecar.operations.rebuild.RebuildOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-sidecar Show documentation
Sidecar for Apache Cassandra
There is a newer version: 2.0.0-alpha8
package com.instaclustr.cassandra.sidecar.operations.rebuild;

import static java.lang.String.format;
import static java.util.stream.Collectors.joining;

import java.time.Instant;
import java.util.Set;
import java.util.UUID;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.inject.Inject;
import com.google.inject.assistedinject.Assisted;
import com.instaclustr.operations.FunctionWithEx;
import com.instaclustr.operations.Operation;
import jmx.org.apache.cassandra.service.CassandraJMXService;
import jmx.org.apache.cassandra.service.cassandra3.StorageServiceMBean;

public class RebuildOperation extends Operation {

    private final CassandraJMXService cassandraJMXService;

    @Inject
    public RebuildOperation(final CassandraJMXService cassandraJMXService,
                            @Assisted final RebuildOperationRequest request) {
        super(request);
        this.cassandraJMXService = cassandraJMXService;
    }

    // this constructor is not meant to be instantiated manually
    // and it fulfills the purpose of deserialisation from JSON string to an Operation object, currently just for testing purposes
    @JsonCreator
    private RebuildOperation(@JsonProperty("type") final String type,
                             @JsonProperty("id") final UUID id,
                             @JsonProperty("creationTime") final Instant creationTime,
                             @JsonProperty("state") final State state,
                             @JsonProperty("failureCause") final Throwable failureCause,
                             @JsonProperty("progress") final float progress,
                             @JsonProperty("startTime") final Instant startTime,
                             @JsonProperty("sourceDC") final String sourceDC,
                             @JsonProperty("keyspace") final String keyspace,
                             @JsonProperty("specificTokens") final Set specificTokens,
                             @JsonProperty("specificSources") final Set specificSources) {
        super(type, id, creationTime, state, failureCause, progress, startTime, new RebuildOperationRequest(type,
                                                                                                            sourceDC,
                                                                                                            keyspace,
                                                                                                            specificTokens,
                                                                                                            specificSources));
        cassandraJMXService = null;
    }

    @Override
    protected void run0() throws Exception {
        assert cassandraJMXService != null;

        final String specificTokens = prepareSpecificTokens(request.specificTokens);
        final String specificSources = prepareSpecificSources(request.specificSources);

        cassandraJMXService.doWithStorageServiceMBean(new FunctionWithEx() {
            @Override
            public Void apply(final StorageServiceMBean object) {

                object.rebuild(request.sourceDC,
                               request.keyspace,
                               specificTokens,
                               specificSources);

                return null;
            }
        });
    }

    private String prepareSpecificTokens(Set specificTokens) {
        if (specificTokens == null || specificTokens.isEmpty()) {
            return null;
        }

        return specificTokens.stream().map(token -> format("(%s,%s]", token.start, token.end)).collect(joining(","));
    }

    private String prepareSpecificSources(Set specificSources) {
        if (specificSources == null || specificSources.isEmpty()) {
            return null;
        }

        return String.join(",", specificSources);
    }
}