基于Influxdb对InfluxDBResultMapper的一点扩展

理想很饱满,现实很骨感。java

因为业务须要“灵活可配置”的功能需求,在使用java开发Influxdb查询功能的时候,遇到了一个问题,Measurement注解的名称有可能须要动态变化。api

咱们先看下 @Measurement 注解的代码:app

package org.influxdb.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.concurrent.TimeUnit;

/**
 * @author fmachado
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface Measurement {

  String name();

  String database() default "[unassigned]";

  String retentionPolicy() default "autogen";

  TimeUnit timeUnit() default TimeUnit.MILLISECONDS;

问题能够转换为:
-> 在@Measurement 注解生效以前,将变更的name值写入。code

通过大概两天时间的资料查找,发现了能够经过:java反射去解决
主要须要操做如下两个类的api继承

java.lang.reflect.InvocationHandler;
java.lang.reflect.Proxy;

最终解决方案,经过继承 InfluxDBResultMapper 的 toPOJO 方法得以解决。
如下贴出代码:flux

public class InfluxDBResultMapperHelper extends InfluxDBResultMapper {

    public <T> List<T> toPOJO(final QueryResult queryResult, final Class<T> clazz,String name)
            throws InfluxDBMapperException {
        InvocationHandler handler = Proxy.getInvocationHandler(clazz.getAnnotation(Measurement.class));
        Field hField = null;
        try {
            hField = handler.getClass().getDeclaredField(ME_VALUE);
            hField.setAccessible(true);
            Map memberValues = (Map) hField.get(handler);
            memberValues.put(ME_NAME, name);
        } catch (NoSuchFieldException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return toPOJO(queryResult, clazz, TimeUnit.MILLISECONDS);
    }
}
相关文章
相关标签/搜索