info.archinnov.achilles.logger.ThriftLoggerHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of achilles-thrift Show documentation

Thrift implementation for Achilles using Hector library

The newest version!

/**
 *
 * Copyright (C) 2012-2013 DuyHai DOAN
 *
 * 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 info.archinnov.achilles.logger;

import info.archinnov.achilles.helper.LoggerHelper;
import info.archinnov.achilles.serializer.ThriftSerializerUtils;

import java.util.ArrayList;
import java.util.List;

import me.prettyprint.hector.api.Serializer;
import me.prettyprint.hector.api.beans.AbstractComposite.Component;
import me.prettyprint.hector.api.beans.Composite;

import org.apache.cassandra.utils.ByteBufferUtil;
import org.apache.commons.lang.StringUtils;

import com.google.common.base.Function;

public class ThriftLoggerHelper extends LoggerHelper {
	public static Function, String> srzToStringFn = new Function, String>() {
		public String apply(Serializer srz) {
			return srz.getComparatorType().getTypeName();
		}
	};

	public static String format(Composite comp) {
		String formatted = "null";
		if (comp != null) {
			formatted = format(comp.getComponents());
		}
		return formatted;
	}

	public static String format(List> components) {
		String formatted = "[]";
		if (components != null && components.size() > 0) {
			List componentsText = new ArrayList();
			int componentNb = components.size();
			for (int i = 0; i < componentNb; i++) {
				Component component = components.get(i);
				String componentValue;
				if (component.getSerializer() == ThriftSerializerUtils.BYTE_SRZ) {
					componentValue = ByteBufferUtil.getArray(component.getBytes())[0] + "";
				} else {
					componentValue = component.getValue(component.getSerializer()).toString();
				}

				if (i == componentNb - 1) {
					componentsText.add(componentValue + "(" + component.getEquality().name() + ")");
				} else {
					componentsText.add(componentValue);
				}
			}
			formatted = '[' + StringUtils.join(componentsText, ':') + ']';
			;
		}
		return formatted;
	}
}