com.caseystella.analytics.converters.primitive.PrimitiveConverter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of logisland-outlier-detection-plugin Show documentation
There is a newer version: 0.9.7
Show newest version
/**
 * Copyright (C) 2016 Hurence ([email protected])
 *
 * 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.caseystella.analytics.converters.primitive;

import com.caseystella.analytics.converters.MappingConverter;
import com.caseystella.analytics.converters.TimestampConverter;
import com.caseystella.analytics.converters.MeasurementConverter;
import com.google.common.base.Function;
import org.apache.hadoop.hbase.util.Bytes;

import java.util.HashMap;
import java.util.Map;

public class PrimitiveConverter {
    public static final String TYPE_CONF = "type";
    public static final String NAME_CONF = "name";
    public enum Type implements Function {
        DOUBLE(new Function() {
            @Override
            public Object apply(byte[] bytes) {
                return Bytes.toDouble(bytes);
            }
        })
        ,LONG(new Function() {
            @Override
            public Object apply(byte[] bytes) {
                return Bytes.toLong(bytes);
            }
        })
        ,INTEGER(new Function() {
            @Override
            public Object apply(byte[] bytes) {
                return Bytes.toInt(bytes);
            }
        })
        ,STRING(new Function() {
            @Override
            public Object apply(byte[] bytes) {
                return Bytes.toString(bytes);
            }
        })
        ;
        private Function _func;
        Type(Function func) {
            _func = func;
        }
        @Override
        public Object apply(byte[] bytes) {
            return _func.apply(bytes);
        }
    }

    public static class PrimitiveMeasurementConverter implements MeasurementConverter {

        @Override
        public Double convert(Object in, Map config) {
            if(in instanceof Double) {
                return (Double)in;
            }
            else if(in instanceof Number) {
                return ((Number)in).doubleValue();
            }
            else if(in instanceof String) {
                return Double.parseDouble(in.toString());
            }
            else
            {
                throw new RuntimeException("Unable to convert " + in + " to a double");
            }
        }
    }

    public static class PrimitiveTimestampConverter implements TimestampConverter{
        @Override
        public Long convert(Object in, Map config) {
            if(in instanceof Long) {
                return (Long)in;
            }
            else if(in instanceof Number) {
                return ((Number)in).longValue();
            }
            else if(in instanceof String) {
                return Long.parseLong(in.toString());
            }
            else
            {
                throw new RuntimeException("Unable to convert " + in + " to a long");
            }
        }
    }

    public static class PrimitiveMappingConverter implements MappingConverter {

        @Override
        public Map convert(final byte[] in, final Map config) {
            final Type t = Type.valueOf((String) config.get(TYPE_CONF));
            return new HashMap() {{
                put((String) config.get(NAME_CONF), t.apply(in));
            }};
        }
    }
}