/*
* Copyright 2012 NGDATA nv
*
* 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.lilyproject.indexer.model.indexerconf;
import java.io.IOException;
import java.util.Set;
import com.google.common.collect.Sets;
import org.lilyproject.repository.api.IdGenerator;
import org.lilyproject.repository.api.LRepository;
import org.lilyproject.repository.api.LTable;
import org.lilyproject.repository.api.Record;
import org.lilyproject.repository.api.RecordNotFoundException;
import org.lilyproject.repository.api.RepositoryException;
import org.lilyproject.repository.api.VersionNotFoundException;
import org.lilyproject.util.repo.VersionTag;
/**
* Represents a -prop1[,-prop2 ...] follow
*/
public class VariantFollow implements Follow {
private Set<String> dimensions;
public VariantFollow(Set<String> dimensions) {
this.dimensions = dimensions;
}
public Set<String> getDimensions() {
return dimensions;
}
@Override
public void follow(IndexUpdateBuilder indexUpdateBuilder, FollowCallback callback)
throws RepositoryException, IOException, InterruptedException {
LRepository repository = indexUpdateBuilder.getRepository();
LTable table = repository.getTable(indexUpdateBuilder.getTable());
IdGenerator idGenerator = repository.getIdGenerator();
RecordContext ctx = indexUpdateBuilder.getRecordContext();
Set<String> currentDimensions = Sets.newHashSet(ctx.dep.id.getVariantProperties().keySet());
currentDimensions.addAll(ctx.dep.moreDimensionedVariants);
if (!currentDimensions.containsAll(dimensions)) {
// the current dimension doesn't contain all the dimensions we need to subtract -> stop here
return;
}
Dep newDep = ctx.dep.minus(idGenerator, dimensions);
Record lessDimensionedRecord = null;
try {
lessDimensionedRecord = VersionTag.getIdRecord(newDep.id, indexUpdateBuilder.getVTag(), table, repository);
} catch (RecordNotFoundException e) {
// It's ok that the variant does not exist
} catch (VersionNotFoundException e) {
// It's ok that the variant does not exist
}
indexUpdateBuilder.push(lessDimensionedRecord, newDep);
callback.call();
indexUpdateBuilder.pop();
}
}