com.google.cloud.bigtable.hbase.adapters.HBaseMutationAdapter Maven / Gradle / Ivy
Show all versions of bigtable-hbase Show documentation
/*
* Copyright 2015 Google LLC
*
* 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 com.google.cloud.bigtable.hbase.adapters;
import com.google.api.core.InternalApi;
import java.util.HashMap;
import java.util.Map;
import org.apache.hadoop.hbase.client.Append;
import org.apache.hadoop.hbase.client.Delete;
import org.apache.hadoop.hbase.client.Increment;
import org.apache.hadoop.hbase.client.Mutation;
import org.apache.hadoop.hbase.client.Put;
/**
* Adapt a generic Mutation to a Google Cloud Java {@link
* com.google.cloud.bigtable.data.v2.models.MutationApi}.
*
* This class uses instanceof checking to determine an appropriate adaptation to apply.
*
*
For internal use only - public for technical reasons.
*/
@InternalApi("For internal usage only")
public class HBaseMutationAdapter extends MutationAdapter {
static class AdapterInstanceMap {
private Map, MutationAdapter> unsafeMap = new HashMap<>();
public > Class put(
Class key, U adapter) {
unsafeMap.put(key, adapter);
return key;
}
// The only way to add to the unsafeMap is via put which enforces our type constraints at
// compile-time. The unchecked cast should be safe.
@SuppressWarnings("unchecked")
public > U get(Class key) {
return (U) unsafeMap.get(key);
}
}
private final AdapterInstanceMap adapterMap = new AdapterInstanceMap();
/**
* Constructor for MutationAdapter.
*
* @param deleteAdapter a {@link com.google.cloud.bigtable.hbase.adapters.OperationAdapter}
* object.
* @param putAdapter a {@link com.google.cloud.bigtable.hbase.adapters.OperationAdapter} object.
* @param incrementAdapter a {@link com.google.cloud.bigtable.hbase.adapters.OperationAdapter}
* object.
* @param appendAdapter a {@link com.google.cloud.bigtable.hbase.adapters.OperationAdapter}
* object.
*/
public HBaseMutationAdapter(
MutationAdapter deleteAdapter,
MutationAdapter putAdapter,
MutationAdapter incrementAdapter,
MutationAdapter appendAdapter) {
adapterMap.put(Delete.class, deleteAdapter);
adapterMap.put(Put.class, putAdapter);
adapterMap.put(Increment.class, incrementAdapter);
adapterMap.put(Append.class, appendAdapter);
}
/** {@inheritDoc} */
@Override
public void adapt(
Mutation mutation, com.google.cloud.bigtable.data.v2.models.MutationApi mutationModel) {
MutationAdapter adapter = adapterMap.get(mutation.getClass());
if (adapter == null) {
throw new UnsupportedOperationException(
String.format(
"Cannot adapt mutation of type %s.", mutation.getClass().getCanonicalName()));
}
adapter.adapt(mutation, mutationModel);
}
}