feat: first commit code

Author: lokic

dataloader-plus-core/pom.xml

@@ -0,0 +1,45 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <groupId>me.lokic</groupId>
+        <artifactId>dataloader-plus</artifactId>
+        <version>1.0-SNAPSHOT</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>dataloader-plus-core</artifactId>
+
+    <properties>
+        <maven.compiler.source>8</maven.compiler.source>
+        <maven.compiler.target>8</maven.compiler.target>
+        <dataloader.version>2.2.3</dataloader.version>
+        <logback.version>1.1.11</logback.version>
+        <lombok.version>1.18.18</lombok.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>com.graphql-java</groupId>
+            <artifactId>java-dataloader</artifactId>
+            <version>${dataloader.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>${lombok.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+            <version>${logback.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-core</artifactId>
+            <version>${logback.version}</version>
+        </dependency>
+    </dependencies>
+
+</project>

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/DataLoaderExecutor.java

@@ -0,0 +1,8 @@
+package me.lokic.dataloaderplus.core;
+
+
+@FunctionalInterface
+public interface DataLoaderExecutor<R> {
+
+    R execute(ExDataLoaderRegistry registry);
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/DataLoaderFactory.java

@@ -0,0 +1,28 @@
+package me.lokic.dataloaderplus.core;
+
+import org.dataloader.DataLoader;
+import org.dataloader.DataLoaderOptions;
+
+import java.util.Map;
+import java.util.Optional;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * DataLoader的工厂，基于 {@link MultiKeyMappedBatchLoader} 创建 {@link DataLoader}
+ */
+public class DataLoaderFactory {
+
+    private final Map<String, MultiKeyMappedBatchLoader<?, ?>> DATA_LOADER_CREATORS = new ConcurrentHashMap<>();
+
+    public void addMultiKeyMappedBatchLoader(MultiKeyMappedBatchLoader<?, ?> dataLoaderProvider) {
+        DATA_LOADER_CREATORS.put(dataLoaderProvider.getClass().getName(), dataLoaderProvider);
+    }
+
+    public DataLoader<?, ?> create(String name, DataLoaderOptions options) {
+        MultiKeyMappedBatchLoader<?, ?> loader = Optional.ofNullable(DATA_LOADER_CREATORS.get(name))
+                .orElseThrow(() -> new IllegalArgumentException("not found data loader supplier"));
+
+        return DataLoader.newMappedDataLoader(loader, options);
+    }
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/DataLoaderTemplate.java

@@ -0,0 +1,50 @@
+package me.lokic.dataloaderplus.core;
+
+import me.lokic.dataloaderplus.core.kits.CompletableFutures;
+import org.dataloader.DataLoaderOptions;
+
+import java.util.concurrent.CompletableFuture;
+
+public class DataLoaderTemplate {
+
+    private final DataLoaderOptions options;
+    private final DataLoaderFactory factory;
+
+
+    public DataLoaderTemplate(TemplateOptions options) {
+        this.options = options.getOptions();
+        this.factory = options.getFactory();
+    }
+
+
+    public <R> R using(DataLoaderExecutor<CompletableFuture<R>> executor) {
+        ExDataLoaderRegistry registry = new ExDataLoaderRegistry(options, factory);
+        return execute(registry, reg -> CompletableFutures.join(using(reg, executor)));
+    }
+
+
+    public <R> CompletableFuture<R> using(ExDataLoaderRegistry registry, DataLoaderExecutor<CompletableFuture<R>> executor) {
+        return execute(registry, executor);
+    }
+
+    private static <R> R execute(ExDataLoaderRegistry registry, DataLoaderExecutor<R> executor) {
+        boolean isOutermost = false;
+        try {
+            if (RegistryHolder.getRegistry() == null) {
+                isOutermost = true;
+                RegistryHolder.setRegistry(registry);
+            } else {
+                if (RegistryHolder.getRegistry() != registry) {
+                    throw new IllegalArgumentException("register must be same");
+                }
+            }
+            return executor.execute(registry);
+        } finally {
+            // 如果是最外层设置registry的方法，则最后需要清除registry
+            if (isOutermost) {
+                RegistryHolder.clear();
+            }
+        }
+    }
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/ExDataLoaderRegistry.java

@@ -0,0 +1,67 @@
+package me.lokic.dataloaderplus.core;
+
+import lombok.extern.slf4j.Slf4j;
+import me.lokic.dataloaderplus.core.proxy.InterfaceProxy;
+import org.dataloader.DataLoader;
+import org.dataloader.DataLoaderOptions;
+import org.dataloader.DataLoaderRegistry;
+import org.dataloader.MappedBatchLoaderWithContext;
+
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * DataLoaderRegistry的扩展
+ */
+@Slf4j
+public class ExDataLoaderRegistry extends DataLoaderRegistry {
+
+    private final DataLoaderOptions options;
+    private final DataLoaderFactory factory;
+
+    /**
+     * 使用 @{@link me.lokic.dataloaderplus.core.annotation.DataLoaderService} 注释接口的实现类的缓存，避免重复创建
+     */
+    private static final Map<Class<?>, Object> SERVICE_CACHE = new ConcurrentHashMap<>();
+
+    protected ExDataLoaderRegistry(DataLoaderOptions options, DataLoaderFactory factory) {
+        this.options = options;
+        this.factory = factory;
+    }
+
+    /**
+     * 获取或者创建对应的 {@link DataLoader}
+     *
+     * @param clazz
+     * @param <K>
+     * @param <V>
+     * @return
+     */
+    public <K, V> DataLoader<K, V> getDataLoader(Class<? extends MappedBatchLoaderWithContext<?, ?>> clazz) {
+        return computeIfAbsent(clazz.getName(), key -> factory.create(key, options));
+    }
+
+    /**
+     * 获取 {@link me.lokic.dataloaderplus.core.annotation.DataLoaderService} 对应的 {@code serviceClazz} 的实现
+     *
+     * @param serviceClazz
+     * @param <S>
+     * @return
+     */
+    @SuppressWarnings("unchecked")
+    public static <S> S getService(Class<S> serviceClazz) {
+        return (S) SERVICE_CACHE.computeIfAbsent(
+                serviceClazz,
+                InterfaceProxy::newInstance
+        );
+    }
+
+
+    @Override
+    public void dispatchAll() {
+        List<DataLoader<?, ?>> dataLoaders = getDataLoaders();
+        dataLoaders.forEach(DataLoader::dispatch);
+    }
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/MultiKey2MappedBatchLoader.java

@@ -0,0 +1,9 @@
+package me.lokic.dataloaderplus.core;
+
+
+import me.lokic.dataloaderplus.core.tuples.Tuple2;
+
+public interface MultiKey2MappedBatchLoader<K1, K2, V> extends MultiKeyMappedBatchLoader<Tuple2<K1, K2>, V> {
+
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/MultiKey3MappedBatchLoader.java

@@ -0,0 +1,8 @@
+package me.lokic.dataloaderplus.core;
+
+
+import me.lokic.dataloaderplus.core.tuples.Tuple3;
+
+public interface MultiKey3MappedBatchLoader<K1, K2, K3, V> extends MultiKeyMappedBatchLoader<Tuple3<K1, K2, K3>, V> {
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/MultiKeyMappedBatchLoader.java

@@ -0,0 +1,17 @@
+package me.lokic.dataloaderplus.core;
+
+import org.dataloader.MappedBatchLoaderWithContext;
+
+/**
+ * 用于对应一些多参数的方法，不需要去重新定义一个包含所有参数的参数对象
+ * <p>
+ * 如, <pre>{@code CompletableFuture<String> method(String p1, String p2)}</pre>
+ * <p>
+ * <a href="https://www.graphql-java.com/documentation/v16/batching/">graphql-java batching</a>
+ *
+ * @param <K>
+ * @param <V>
+ */
+public interface MultiKeyMappedBatchLoader<K, V> extends MappedBatchLoaderWithContext<K, V> {
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/RegistryHolder.java

@@ -0,0 +1,47 @@
+package me.lokic.dataloaderplus.core;
+
+
+/**
+ * 以线程绑定的形式获取 {@link ExDataLoaderRegistry}
+ */
+public class RegistryHolder {
+
+    public static final ThreadLocal<ExDataLoaderRegistry> HOLDER = new ThreadLocal<>();
+
+    /**
+     * 获取当前线程的CustomizeDataLoaderRegistry
+     *
+     * @return
+     */
+    public static ExDataLoaderRegistry getRegistry() {
+        return HOLDER.get();
+    }
+
+    /**
+     * 当前线程设置CustomizeDataLoaderRegistry
+     *
+     * @param registry
+     */
+    public static void setRegistry(ExDataLoaderRegistry registry) {
+        HOLDER.set(registry);
+    }
+
+    /**
+     * 针对当前线程的CustomizeDataLoaderRegistry，
+     * 如果存在则调用 {@link ExDataLoaderRegistry#dispatchAll()}, 不存在则忽略该请求
+     */
+    public static void tryDispatchAll() {
+        ExDataLoaderRegistry registry = getRegistry();
+        if (registry != null) {
+            registry.dispatchAll();
+        }
+    }
+
+    /**
+     * 清除当前线程的CustomizeDataLoaderRegistry
+     */
+    public static void clear() {
+        HOLDER.remove();
+    }
+
+}

dataloader-plus-core/src/main/java/me/lokic/dataloaderplus/core/TemplateOptions.java

@@ -0,0 +1,33 @@
+package me.lokic.dataloaderplus.core;
+
+import lombok.Builder;
+import lombok.Getter;
+import org.dataloader.DataLoaderOptions;
+
+/**
+ * {@link DataLoaderTemplate} 需要的配置
+ */
+@Getter
+@Builder
+public class TemplateOptions {
+
+    private static final DataLoaderOptions DEFAULT_OPTIONS = DataLoaderOptions.newOptions();
+    public static final DataLoaderFactory DEFAULT_FACTORY = new DataLoaderFactory();
+
+    private final DataLoaderOptions options;
+    private final DataLoaderFactory factory;
+
+
+    public TemplateOptions(DataLoaderOptions options, DataLoaderFactory factory) {
+        this.options = options == null ? DEFAULT_OPTIONS : options;
+        this.factory = factory == null ? DEFAULT_FACTORY : factory;
+    }
+
+    public DataLoaderOptions getOptions() {
+        return options;
+    }
+
+    public DataLoaderFactory getFactory() {
+        return factory;
+    }
+}