diff --git a/.gitignore b/.gitignore index 9154f4c..549e00a 100644 --- a/.gitignore +++ b/.gitignore @@ -1,26 +1,33 @@ -# ---> Java -# Compiled class file -*.class +HELP.md +target/ +!.mvn/wrapper/maven-wrapper.jar +!**/src/main/**/target/ +!**/src/test/**/target/ -# Log file -*.log +### STS ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache -# BlueJ files -*.ctxt +### IntelliJ IDEA ### +.idea +*.iws +*.iml +*.ipr -# Mobile Tools for Java (J2ME) -.mtj.tmp/ - -# Package Files # -*.jar -*.war -*.nar -*.ear -*.zip -*.tar.gz -*.rar - -# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml -hs_err_pid* -replay_pid* +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/dist/ +/nbdist/ +/.nb-gradle/ +build/ +!**/src/main/**/build/ +!**/src/test/**/build/ +### VS Code ### +.vscode/ diff --git a/pom.xml b/pom.xml new file mode 100644 index 0000000..f090830 --- /dev/null +++ b/pom.xml @@ -0,0 +1,78 @@ + + + 4.0.0 + + com.njcn + influxdb-springboot-starter + 1.0.0 + influxdb-springboot-starter + 封装influxdb的orm工具 + + 1.8 + UTF-8 + 2.3.12.RELEASE + 2.0.0.RELEASE + 2.18 + + + + + hongawen + 83944980@qq.com + + + + + + com.njcn + common-core + 1.0.0 + + + org.springframework.boot + spring-boot-configuration-processor + true + ${configuration-processor.version} + + + org.springframework.boot + spring-boot-autoconfigure + ${autoconfigure.version} + + + org.influxdb + influxdb-java + ${influxdb-java.version} + + + + + + + org.apache.maven.plugins + maven-compiler-plugin + 3.8.1 + + 1.8 + 1.8 + + + + org.sonatype.plugins + nexus-staging-maven-plugin + 1.6.7 + true + + ossrh + https://s01.oss.sonatype.org/ + false + + + + + + + + + diff --git a/src/main/java/com/njcn/influx/ano/Delete.java b/src/main/java/com/njcn/influx/ano/Delete.java new file mode 100644 index 0000000..fcf38ba --- /dev/null +++ b/src/main/java/com/njcn/influx/ano/Delete.java @@ -0,0 +1,27 @@ +package com.njcn.influx.ano; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 删除记录注解 + * @author hongawen + */ +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +public @interface Delete { + + /** + * 用于传参 + * @return String 参数内容 + */ + String value(); + + /*** + * 数据库名 + * todo... 处理成获取配置文件内的，减少开发人员编码输出 + */ + String database(); +} diff --git a/src/main/java/com/njcn/influx/ano/Insert.java b/src/main/java/com/njcn/influx/ano/Insert.java new file mode 100644 index 0000000..758cb47 --- /dev/null +++ b/src/main/java/com/njcn/influx/ano/Insert.java @@ -0,0 +1,15 @@ +package com.njcn.influx.ano; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 插入注解 + * @author hongawen + */ +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +public @interface Insert { +} diff --git a/src/main/java/com/njcn/influx/ano/Param.java b/src/main/java/com/njcn/influx/ano/Param.java new file mode 100644 index 0000000..f0bfdcd --- /dev/null +++ b/src/main/java/com/njcn/influx/ano/Param.java @@ -0,0 +1,14 @@ +package com.njcn.influx.ano; + +import java.lang.annotation.*; + +/** + * 便于将方法中的参数赋值进注解内容中,通过#{paramName} + * @author hongawen + */ +@Documented +@Retention(RetentionPolicy.RUNTIME) +@Target({ElementType.PARAMETER}) +public @interface Param { + String value(); +} diff --git a/src/main/java/com/njcn/influx/ano/Select.java b/src/main/java/com/njcn/influx/ano/Select.java new file mode 100644 index 0000000..3a7a763 --- /dev/null +++ b/src/main/java/com/njcn/influx/ano/Select.java @@ -0,0 +1,25 @@ +package com.njcn.influx.ano; + +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +/** + * 查询记录 + * @author hongawen + */ +@Target(ElementType.METHOD) +@Retention(RetentionPolicy.RUNTIME) +public @interface Select { + + /*** + * 查询sql + */ + String value(); + + /*** + * 返回映射的实体对象 + */ + Class resultType(); +} diff --git a/src/main/java/com/njcn/influx/base/InfluxDbBaseMapper.java b/src/main/java/com/njcn/influx/base/InfluxDbBaseMapper.java new file mode 100644 index 0000000..0f105c8 --- /dev/null +++ b/src/main/java/com/njcn/influx/base/InfluxDbBaseMapper.java @@ -0,0 +1,26 @@ +package com.njcn.influx.base; + + +import com.njcn.influx.ano.Insert; + +import java.util.List; + +/** + * @author hongawen + */ +public interface InfluxDbBaseMapper { + + /*** + * 插入数据 + * @param entity 数据 + */ + @Insert + void insertOne(T entity); + + /*** + * 批量插入数据 + * @param entityList 数据 + */ + @Insert + void insertBatch(List entityList); +} diff --git a/src/main/java/com/njcn/influx/base/InfluxProxyMapperFactory.java b/src/main/java/com/njcn/influx/base/InfluxProxyMapperFactory.java new file mode 100644 index 0000000..8004f2b --- /dev/null +++ b/src/main/java/com/njcn/influx/base/InfluxProxyMapperFactory.java @@ -0,0 +1,33 @@ +package com.njcn.influx.base; + +import com.njcn.influx.core.InfluxExecutor; +import com.njcn.influx.core.ParameterHandler; +import com.njcn.influx.core.ResultSetHandler; +import org.springframework.beans.factory.FactoryBean; + +import javax.annotation.Resource; +import java.lang.reflect.Proxy; + +public class InfluxProxyMapperFactory implements FactoryBean { + + @Resource + InfluxExecutor executor; + + private Class interfaceClass; + + public InfluxProxyMapperFactory(Class interfaceClass) { + this.interfaceClass = interfaceClass; + } + + @Override + public Object getObject() throws Exception { + Object proxyInstance = Proxy.newProxyInstance( + interfaceClass.getClassLoader(), new Class[]{interfaceClass}, new ProxyMapper(new ParameterHandler(), executor, new ResultSetHandler())); + return proxyInstance; + } + + @Override + public Class getObjectType() { + return interfaceClass; + } +} diff --git a/src/main/java/com/njcn/influx/base/ProxyMapper.java b/src/main/java/com/njcn/influx/base/ProxyMapper.java new file mode 100644 index 0000000..3518879 --- /dev/null +++ b/src/main/java/com/njcn/influx/base/ProxyMapper.java @@ -0,0 +1,68 @@ +package com.njcn.influx.base; + + +import com.njcn.influx.ano.Delete; +import com.njcn.influx.ano.Select; +import com.njcn.influx.ano.Insert; +import com.njcn.influx.core.InfluxExecutor; +import com.njcn.influx.core.ParameterHandler; +import com.njcn.influx.core.ResultSetHandler; + +import java.lang.annotation.Annotation; +import java.lang.reflect.InvocationHandler; +import java.lang.reflect.Method; +import java.lang.reflect.Parameter; +import java.util.List; + +public class ProxyMapper implements InvocationHandler { + + private ParameterHandler parameterHandler; + private InfluxExecutor executor; + private ResultSetHandler resultSetHandler; + + public ProxyMapper(ParameterHandler parameterHandler, InfluxExecutor executor, ResultSetHandler resultSetHandler) { + this.parameterHandler = parameterHandler; + this.executor = executor; + this.resultSetHandler = resultSetHandler; + } + + @Override + public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { + Annotation[] annotations = method.getAnnotations(); + if (annotations.length == 1) { + Annotation annotation = annotations[0]; + Class annotationType = annotation.annotationType(); + //是查询的 + if (annotationType == Select.class) { + Select selectAnnotation = method.getAnnotation(Select.class); + //拼接sql + String sql = selectAnnotation.value(); + Parameter[] parameters = method.getParameters(); + sql = parameterHandler.handleParameter(parameters, args, sql); + //查询结果 + Class resultType = selectAnnotation.resultType(); + List list = executor.select(sql, resultType); + //根据返回类型返回结果 + Class returnType = method.getReturnType(); + return resultSetHandler.handleResultSet(list, returnType); + } + //是插入的 + if (annotationType == Insert.class) { + executor.insert(args); + } + //是删除的 + if (annotationType == Delete.class) { + Delete deleteAnnotation = method.getAnnotation(Delete.class); + //拼接sql + String sql = deleteAnnotation.value(); + Parameter[] parameters = method.getParameters(); + + String database = deleteAnnotation.database(); + sql = parameterHandler.handleParameter(parameters, args, sql); + //执行sql + executor.delete(sql, database); + } + } + return null; + } +} diff --git a/src/main/java/com/njcn/influx/base/ProxyMapperRegister.java b/src/main/java/com/njcn/influx/base/ProxyMapperRegister.java new file mode 100644 index 0000000..2374eaa --- /dev/null +++ b/src/main/java/com/njcn/influx/base/ProxyMapperRegister.java @@ -0,0 +1,70 @@ +package com.njcn.influx.base; + +import com.njcn.influx.utils.InfluxStrUtil; +import com.njcn.influx.utils.ManualRegisterBeanUtil; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.core.io.Resource; +import org.springframework.core.io.ResourceLoader; +import org.springframework.core.io.support.ResourcePatternResolver; +import org.springframework.core.io.support.ResourcePatternUtils; +import org.springframework.core.type.classreading.CachingMetadataReaderFactory; +import org.springframework.core.type.classreading.MetadataReader; +import org.springframework.core.type.classreading.MetadataReaderFactory; +import org.springframework.util.ClassUtils; +import org.springframework.util.SystemPropertyUtils; + +import javax.annotation.PostConstruct; +import java.io.IOException; +import java.util.HashSet; + +public class ProxyMapperRegister { + + private String mapperLocation; + ConfigurableApplicationContext applicationContext; + ResourceLoader resourceLoader; + + public ProxyMapperRegister(String mapperLocation, ConfigurableApplicationContext applicationContext, ResourceLoader resourceLoader) { + this.mapperLocation = mapperLocation; + this.applicationContext = applicationContext; + this.resourceLoader = resourceLoader; + } + + @PostConstruct + public void afterInit() { + //获取mapper包下所有的Mapper + HashSet> classes = new HashSet<>(); + String packageSearchPath = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX + .concat(ClassUtils.convertClassNameToResourcePath(SystemPropertyUtils.resolvePlaceholders(mapperLocation)) + .concat("/**/*.class")); + ResourcePatternResolver resolver = ResourcePatternUtils.getResourcePatternResolver(resourceLoader); + MetadataReaderFactory metadataReaderFactory = new CachingMetadataReaderFactory(resourceLoader); + MetadataReader metadataReader; + try { + Resource[] resources = resolver.getResources(packageSearchPath); + for (Resource resource : resources) { + if (resource.isReadable()) { + metadataReader = metadataReaderFactory.getMetadataReader(resource); + String className = metadataReader.getClassMetadata().getClassName(); + Class aClass = Class.forName(className); + + //当这个Mapper实现InfluxBaseMapper时加集合 + Class[] interfaces = aClass.getInterfaces(); + for (Class anInterface : interfaces) { + if (anInterface == InfluxDbBaseMapper.class) { + classes.add(aClass); + } + } + } + } + } catch (IOException e) { + e.printStackTrace(); + } catch (ClassNotFoundException e) { + e.printStackTrace(); + } + + for (Class aClass : classes) { + //注册接口类至工厂Bean，再用动态代理生成对应Mapper + ManualRegisterBeanUtil.registerBean(applicationContext, InfluxStrUtil.captureName(aClass.getSimpleName()), InfluxProxyMapperFactory.class, aClass); + } + } +} diff --git a/src/main/java/com/njcn/influx/config/InfluxDbConfig.java b/src/main/java/com/njcn/influx/config/InfluxDbConfig.java new file mode 100644 index 0000000..5d9ba92 --- /dev/null +++ b/src/main/java/com/njcn/influx/config/InfluxDbConfig.java @@ -0,0 +1,44 @@ +package com.njcn.influx.config; + +import com.njcn.influx.base.ProxyMapperRegister; +import com.njcn.influx.core.InfluxExecutor; +import org.influxdb.InfluxDB; +import org.influxdb.impl.InfluxDBMapper; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.context.ConfigurableApplicationContext; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; +import org.springframework.core.io.ResourceLoader; + + +/** + * @author hongawen + * @version 1.0.0 + */ +@Configuration +public class InfluxDbConfig { + + + /*** + * influxMapper存放路径 + */ + @Value("${spring.influx.mapper-location}") + private String mapperLocation; + + @Bean(name = "influxDbMapper") + public InfluxDBMapper influxDBMapper(InfluxDB influxDb) { + influxDb.setLogLevel(InfluxDB.LogLevel.BASIC); + return new InfluxDBMapper(influxDb); + } + + @Bean(name = "influxDbExecutor") + public InfluxExecutor executor(InfluxDB influxDb, InfluxDBMapper influxDbMapper) { + return new InfluxExecutor(influxDb, influxDbMapper); + } + + @Bean(name = "proxyMapperRegister") + public ProxyMapperRegister proxyMapperRegister(ConfigurableApplicationContext applicationContext, ResourceLoader resourceLoader) { + return new ProxyMapperRegister(mapperLocation, applicationContext, resourceLoader); + } + +} diff --git a/src/main/java/com/njcn/influx/constant/InfluxDbSqlConstant.java b/src/main/java/com/njcn/influx/constant/InfluxDbSqlConstant.java new file mode 100644 index 0000000..ac75cdc --- /dev/null +++ b/src/main/java/com/njcn/influx/constant/InfluxDbSqlConstant.java @@ -0,0 +1,107 @@ +package com.njcn.influx.constant; + +import cn.hutool.core.text.StrPool; + +/** + * 拼接influx查询语句相关常量 + * + * @author hongawen + * @version 1.0.0 + * @date 2022年10月14日 14:02 + */ +public interface InfluxDbSqlConstant { + + /** + * "SELECT " + */ + String SELECT = "SELECT" + StrPool.C_SPACE; + + /** + * "* " + */ + String ALL = "*" + StrPool.C_SPACE; + + /** + * "FROM " + */ + String FROM = "FROM" + StrPool.C_SPACE; + + /** + * " WHERE " + */ + String WHERE = StrPool.C_SPACE + "WHERE" + StrPool.C_SPACE; + + /** + * " AND " + */ + String AND = StrPool.C_SPACE + "AND" + StrPool.C_SPACE; + + /** + * " OR " + */ + String OR = StrPool.C_SPACE + "OR" + StrPool.C_SPACE; + + /** + * " GROUP BY " + */ + String GB = StrPool.C_SPACE + "GROUP BY" + StrPool.C_SPACE; + + /** + * " ORDER BY " + */ + String OB = StrPool.C_SPACE + "ORDER BY" + StrPool.C_SPACE; + + /** + * “DESC ” + */ + String DESC = "DESC" + StrPool.C_SPACE; + + /** + * " as value " + */ + String AS_VALUE = StrPool.C_SPACE + "as value" + StrPool.C_SPACE; + + String EQ = "="; + String QM = "'"; + String LBK = "("; + String RBK = ")"; + String GT = ">"; + String GE = ">="; + String LT = "<"; + String LE = "<="; + + /** + * influxDB函数拼接 + */ + String MAX = "MAX"; + String MIN = "MIN"; + String AVG = "MEAN"; + String NUM_95 = ",95"; + String CP95 = "PERCENTILE"; + + /** + * “ tz('Asia/Shanghai')” + */ + String TZ = StrPool.C_SPACE + "tz('Asia/Shanghai')"; + String TIME_AREA = "Asia/Shanghai"; + + + /** + * 日起始时间 + */ + String START_TIME = " 00:00:00"; + + /** + * 日结束时间 + */ + String END_TIME = " 23:59:59"; + + /** + * 时间 + */ + String TIME = "time" + StrPool.C_SPACE; + + + + +} diff --git a/src/main/java/com/njcn/influx/core/InfluxExecutor.java b/src/main/java/com/njcn/influx/core/InfluxExecutor.java new file mode 100644 index 0000000..6c89b87 --- /dev/null +++ b/src/main/java/com/njcn/influx/core/InfluxExecutor.java @@ -0,0 +1,90 @@ +package com.njcn.influx.core; + +import com.njcn.influx.constant.InfluxDbSqlConstant; +import lombok.AllArgsConstructor; +import lombok.NoArgsConstructor; +import org.influxdb.InfluxDB; +import org.influxdb.annotation.Measurement; +import org.influxdb.dto.BatchPoints; +import org.influxdb.dto.Point; +import org.influxdb.dto.Query; +import org.influxdb.impl.InfluxDBMapper; + +import java.util.ArrayList; +import java.util.List; + +/** + * @author hongawen + */ +@NoArgsConstructor +@AllArgsConstructor +public class InfluxExecutor { + + InfluxDB influxDb; + InfluxDBMapper influxDbMapper; + /*** + * 查询数据方法 + * @author hongawen + * @param sql 查询语句 + * @param domainClass 映射实体 + */ + public List select(String sql, Class domainClass) { + List results = influxDbMapper.query(new Query(dealTimeZone(sql)), domainClass); + return results; + } + + /*** + * 插入数据 + * @author hongawen + * @param args 数据内容 + */ + public void insert(Object args[]) { + if (args.length != 1) { + throw new RuntimeException(); + } + Object obj = args[0]; + //插入的是集合类型的 + if (obj instanceof List) { + List list = (ArrayList) obj; + if (list.size() > 0) { + Object firstObj = list.get(0); + Class domainClass = firstObj.getClass(); + List pointList = new ArrayList<>(); + for (Object o : list) { + Point point = Point + .measurementByPOJO(domainClass) + .addFieldsFromPOJO(o) + .build(); + pointList.add(point); + } + //获取数据库名和rp + Measurement measurement = firstObj.getClass().getAnnotation(Measurement.class); + String database = measurement.database(); + String retentionPolicy = measurement.retentionPolicy(); + BatchPoints batchPoints = BatchPoints + .builder() + .points(pointList) + .retentionPolicy(retentionPolicy).build(); + influxDb.setDatabase(database); + influxDb.write(batchPoints); + } + } else { + //插入单个 + influxDbMapper.save(obj); + } + } + + /*** + * 删除操作 + * @author hongawen + * @param sql 数据库语句 + * @param database 数据库名称 + */ + public void delete(String sql, String database) { + influxDb.query(new Query(dealTimeZone(sql), database)); + } + + private String dealTimeZone(String originalSql){ + return originalSql + InfluxDbSqlConstant.TZ; + } +} diff --git a/src/main/java/com/njcn/influx/core/ParameterHandler.java b/src/main/java/com/njcn/influx/core/ParameterHandler.java new file mode 100644 index 0000000..6f1e61d --- /dev/null +++ b/src/main/java/com/njcn/influx/core/ParameterHandler.java @@ -0,0 +1,41 @@ +package com.njcn.influx.core; + + +import com.njcn.influx.ano.Param; + +import java.lang.reflect.Parameter; + +/** + * 参数处理器 + */ +public class ParameterHandler { + + /** + * 拼接sql + * + * @param parameters 参数名 + * @param args 参数实际值 + * @param sql 未拼接参数的sql语句 + * @return 拼接好的sql + */ + public String handleParameter(Parameter[] parameters, Object[] args, String sql) { + for (int i = 0; i < parameters.length; i++) { + Class parameterType = parameters[i].getType(); + String parameterName = parameters[i].getName(); + + Param param = parameters[i].getAnnotation(Param.class); + if (param != null) { + parameterName = param.value(); + } + + if (parameterType == String.class) { + sql = sql.replaceAll("\\#\\{" + parameterName + "\\}", "'" + args[i] + "'"); + sql = sql.replaceAll("\\$\\{" + parameterName + "\\}", args[i].toString()); + } else { + sql = sql.replaceAll("\\#\\{" + parameterName + "\\}", args[i].toString()); + sql = sql.replaceAll("\\$\\{" + parameterName + "\\}", args[i].toString()); + } + } + return sql; + } +} diff --git a/src/main/java/com/njcn/influx/core/ResultSetHandler.java b/src/main/java/com/njcn/influx/core/ResultSetHandler.java new file mode 100644 index 0000000..81c03c2 --- /dev/null +++ b/src/main/java/com/njcn/influx/core/ResultSetHandler.java @@ -0,0 +1,20 @@ +package com.njcn.influx.core; + +import java.util.List; + +/** + * 结果集处理器 + */ +public class ResultSetHandler { + public Object handleResultSet(List list, Class returnType) { + if (returnType == List.class) { + return list; + } + if (returnType != List.class) { + if (list.size() > 0) { + return list.get(0); + } + } + return null; + } +} diff --git a/src/main/java/com/njcn/influx/utils/InfluxStrUtil.java b/src/main/java/com/njcn/influx/utils/InfluxStrUtil.java new file mode 100644 index 0000000..003c39e --- /dev/null +++ b/src/main/java/com/njcn/influx/utils/InfluxStrUtil.java @@ -0,0 +1,12 @@ +package com.njcn.influx.utils; + +/** + * @author hongawen + */ +public class InfluxStrUtil { + public static String captureName(String name) { + //UpperCase大写 + name = name.substring(0, 1).toLowerCase() + name.substring(1); + return name; + } +} diff --git a/src/main/java/com/njcn/influx/utils/ManualRegisterBeanUtil.java b/src/main/java/com/njcn/influx/utils/ManualRegisterBeanUtil.java new file mode 100644 index 0000000..13eb7f0 --- /dev/null +++ b/src/main/java/com/njcn/influx/utils/ManualRegisterBeanUtil.java @@ -0,0 +1,30 @@ +package com.njcn.influx.utils; + +import org.springframework.beans.factory.config.BeanDefinition; +import org.springframework.beans.factory.support.BeanDefinitionBuilder; +import org.springframework.beans.factory.support.BeanDefinitionRegistry; +import org.springframework.context.ConfigurableApplicationContext; + +/** + * 手动将类加入到spring容器中 + * @author hongawen + */ +public class ManualRegisterBeanUtil { + + public static void registerBean(ConfigurableApplicationContext applicationContext, String name, Class clazz, + Object... args) { + if (applicationContext.containsBean(name)) { + Object bean = applicationContext.getBean(name); + if (!bean.getClass().isAssignableFrom(clazz)) { + throw new RuntimeException("BeanName 重复 " + name); + } + } + BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(clazz); + for (Object arg : args) { + beanDefinitionBuilder.addConstructorArgValue(arg); + } + BeanDefinition beanDefinition = beanDefinitionBuilder.getRawBeanDefinition(); + BeanDefinitionRegistry beanFactory = (BeanDefinitionRegistry) applicationContext.getBeanFactory(); + beanFactory.registerBeanDefinition(name, beanDefinition); + } +} diff --git a/src/main/resources/META-INF/spring.factories b/src/main/resources/META-INF/spring.factories new file mode 100644 index 0000000..1e99d14 --- /dev/null +++ b/src/main/resources/META-INF/spring.factories @@ -0,0 +1 @@ +org.springframework.boot.autoconfigure.EnableAutoConfiguration=com.njcn.influx.config.InfluxDbConfig \ No newline at end of file