Использование базового абстрактного класса для объединения двух похожих классов - PullRequest
Новая
       9

Использование базового абстрактного класса для объединения двух похожих классов

0 голосов
/ 04 сентября 2018

Я работаю над запросом на извлечение проекта, и мне сказали, что было бы неплохо объединить код двух похожих классов в один абстрактный класс, из которого будут расширяться два других класса. Моя проблема в том, что я не на правильном пути, чтобы сделать это. Вот два класса: 

public class ClassOne<OUT> extends RichInputFormat<OUT, InputSplit> implements NonParallelInput {
private static final Logger LOG = LoggerFactory.getLogger(ClassOne.class);

private final String query;
private final ClusterBuilder builder;

private transient Cluster cluster;
private transient Session session;
private transient Result<OUT> resultSet;
private Class<OUT> inputClass;

public ClassOne(String query, ClusterBuilder builder, Class<OUT> inputClass) {
    Preconditions.checkArgument(!Strings.isNullOrEmpty(query), "Query cannot be null or empty");
    Preconditions.checkArgument(builder != null, "Builder cannot be null");
    Preconditions.checkArgument(inputClass != null, "InputClass cannot be null");

    this.query = query;
    this.builder = builder;
    this.inputClass = inputClass;
}

@Override
public void configure(Configuration parameters) {
    this.cluster = builder.getCluster();
}

@Override
public BaseStatistics getStatistics(BaseStatistics cachedStatistics) throws IOException {
    return cachedStatistics;
}

/**
 * Opens a Session and executes the query.
 *
 * @param ignored
 * @throws IOException
 */
@Override
public void open(InputSplit ignored) throws IOException {
    this.session = cluster.connect();
    MappingManager manager = new MappingManager(session);

    Mapper<OUT> mapper = manager.mapper(inputClass);

    this.resultSet = mapper.map(session.execute(query));
}

@Override
public boolean reachedEnd() throws IOException {
    return resultSet.isExhausted();
}

@Override
public OUT nextRecord(OUT reuse) throws IOException {
    return resultSet.one();
}

@Override
public InputSplit[] createInputSplits(int minNumSplits) throws IOException {
    GenericInputSplit[] split = {new GenericInputSplit(0, 1)};
    return split;
}

@Override
public InputSplitAssigner getInputSplitAssigner(InputSplit[] inputSplits) {
    return new DefaultInputSplitAssigner(inputSplits);
}

/**
 * Closes all resources used.
 */
@Override
public void close() throws IOException {
    try {
        if (session != null) {
            session.close();
        }
    } catch (Exception e) {
        LOG.error("Error while closing session.", e);
    }

    try {
        if (cluster != null) {
            cluster.close();
        }
    } catch (Exception e) {
        LOG.error("Error while closing cluster.", e);
    }
}
}

И второе: 

public class ClassTwo<OUT extends Tuple> extends RichInputFormat<OUT, InputSplit> implements NonParallelInput {
private static final Logger LOG = LoggerFactory.getLogger(ClassTwo.class);

private final String query;
private final ClusterBuilder builder;

private transient Cluster cluster;
private transient Session session;
private transient ResultSet resultSet;

public ClassTwo(String query, ClusterBuilder builder) {
    Preconditions.checkArgument(!Strings.isNullOrEmpty(query), "Query cannot be null or empty");
    Preconditions.checkArgument(builder != null, "Builder cannot be null");

    this.query = query;
    this.builder = builder;
}

@Override
public void configure(Configuration parameters) {
    this.cluster = builder.getCluster();
}

@Override
public BaseStatistics getStatistics(BaseStatistics cachedStatistics) throws IOException {
    return cachedStatistics;
}

/**
 * Opens a Session and executes the query.
 *
 * @param ignored
 * @throws IOException
 */
@Override
public void open(InputSplit ignored) throws IOException {
    this.session = cluster.connect();
    this.resultSet = session.execute(query);
}

@Override
public boolean reachedEnd() throws IOException {
    return resultSet.isExhausted();
}

@Override
public OUT nextRecord(OUT reuse) throws IOException {
    final Row item = resultSet.one();
    for (int i = 0; i < reuse.getArity(); i++) {
        reuse.setField(item.getObject(i), i);
    }
    return reuse;
}

@Override
public InputSplit[] createInputSplits(int minNumSplits) throws IOException {
    GenericInputSplit[] split = {new GenericInputSplit(0, 1)};
    return split;
}

@Override
public InputSplitAssigner getInputSplitAssigner(InputSplit[] inputSplits) {
    return new DefaultInputSplitAssigner(inputSplits);
}

/**
 * Closes all resources used.
 */
@Override
public void close() throws IOException {
    try {
        if (session != null) {
            session.close();
        }
    } catch (Exception e) {
        LOG.error("Error while closing session.", e);
    }

    try {
        if (cluster != null) {
            cluster.close();
        }
    } catch (Exception e) {
        LOG.error("Error while closing cluster.", e);
    }
}
}

Различия заключаются в методах Constructor (), open () и nextRecord () и переменной resultSet. Также есть класс для ClassOne и ClassTwo. Я предполагаю, что методы open и nextRecord просто не реализуют их в абстрактном базовом классе и оставляют это делать в ClassOne и ClassTwo. Но у абстрактного базового класса должен быть конструктор, а у двух классов разные конструкторы, и у базового класса должен быть либо один, либо, что приведет к тому, что ClassOne или ClassTwo не будут работать в зависимости от используемого. Я чувствую, что пропускаю простое решение.

Любая помощь приветствуется! Спасибо!

1 Ответ

0 голосов
/ 04 сентября 2018

Вполне возможно иметь абстрактный класс в Java (и других языках), который содержит пустой конструктор. Таким образом, композиция двух классов, предоставленных вами с использованием абстрактного класса, может выглядеть примерно так: 

public abstract class ClassAbstract extends RichInputFormat<OUT, InputSplit> implements NonParallelInput {

    private final String query;
    private final ClusterBuilder builder;
    private transient Cluster cluster;
    private transient Session session;

    public ClassAbstract(String query, ClusterBuilder builder) {
        Preconditions.checkArgument(!Strings.isNullOrEmpty(query), "Query cannot be null or empty");
        Preconditions.checkArgument(builder != null, "Builder cannot be null");
        this.query = query;
        this.builder = builder;
    }

    @Override
    public void configure(Configuration parameters) {
        this.cluster = builder.getCluster();
    }

    @Override
    public BaseStatistics getStatistics(BaseStatistics cachedStatistics) throws IOException {
        return cachedStatistics;
    }

    @Override
    public boolean reachedEnd() throws IOException {
        return resultSet.isExhausted();
    }

    @Override
    public InputSplit[] createInputSplits(int minNumSplits) throws IOException {
        GenericInputSplit[] split = {new GenericInputSplit(0, 1)};
        return split;
    }

    @Override
    public InputSplitAssigner getInputSplitAssigner(InputSplit[] inputSplits) {
        return new DefaultInputSplitAssigner(inputSplits);
    }

    /**
     * Closes all resources used.
     */
    @Override
    public void close() throws IOException {
        try {
            if (session != null) {
                session.close();
            }
        } catch (Exception e) {
            LOG.error("Error while closing session.", e);
        }

        try {
            if (cluster != null) {
                cluster.close();
            }
        } catch (Exception e) {
            LOG.error("Error while closing cluster.", e);
        }
    }

}

Тогда ваш конструктор ClassOne может быть просто: 

    public ClassOne(String query, ClusterBuilder builder, Class<OUT> inputClass) {
        super(query, builder);
        Preconditions.checkArgument(inputClass != null, "InputClass cannot be null");
        this.inputClass = inputClass;
    }

И конструктор ClassTwo станет: 

    public ClassTwo(String query, ClusterBuilder builder) {
        super(query, builder);
    }

Затем вы можете создать экземпляр своего дочернего класса, расширив свой абстрактный класс и удалив весь код, уже существующий в абстрактном классе.

...