feat(graphql): add MutableTypeBatchResolver

datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/MutableTypeBatchResolver.java

@@ -0,0 +1,51 @@
+package com.linkedin.datahub.graphql.resolvers.mutate;
+
+import com.codahale.metrics.Timer;
+import com.linkedin.datahub.graphql.exception.AuthorizationException;
+import com.linkedin.datahub.graphql.types.MutableType;
+import com.linkedin.metadata.utils.metrics.MetricUtils;
+import graphql.schema.DataFetcher;
+import graphql.schema.DataFetchingEnvironment;
+import java.util.concurrent.CompletableFuture;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.*;
+
+
+/**
+ * Generic GraphQL resolver responsible for performing updates against particular types.
+ *
+ * @param <I> the generated GraphQL POJO corresponding to the input type.
+ * @param <T> the generated GraphQL POJO corresponding to the return type.
+ */
+public class MutableTypeBatchResolver<I, T> implements DataFetcher<CompletableFuture<T>> {
+
+  private static final Logger _logger = LoggerFactory.getLogger(MutableTypeResolverBatch.class.getName());
+
+  private final MutableType<I> _mutableType;
+
+  public MutableTypeResolverBatch(final MutableType<I> mutableType) {
+    _mutableType = mutableType;
+  }
+
+  @Override
+  public CompletableFuture<T> get(DataFetchingEnvironment environment) throws Exception {
+    final I[] inputs = bindArgument(environment.getArgument("inputs"), _mutableType.arrayInputClass());
+
+    return CompletableFuture.supplyAsync(() -> {
+      Timer.Context timer = MetricUtils.timer(this.getClass(), "batchMutate").time();
+
+      try {
+        return _mutableType.batchUpdate(inputs, environment.getContext());
+      } catch (AuthorizationException e) {
+        throw e;
+      } catch (Exception e) {
+        _logger.error("Failed to perform batchUpdate", e);
+        throw new IllegalArgumentException(e);
+      } finally {
+        timer.stop();
+      }
+    });
+  }
+}

datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/MutableType.java

@@ -24,4 +24,14 @@ public interface MutableType<I, T> {
      * @param context the {@link QueryContext} corresponding to the request.
      */
     T update(@Nonnull final String urn, @Nonnull final I input, @Nonnull final QueryContext context) throws Exception;
+
+   /**
+    * Update many entities
+    *
+    * @param input input type
+    * @param context the {@link QueryContext} corresponding to the request.
+    */
+   default <T> T batchUpdate(@Nonnull final I[] input, @Nonnull final QueryContext context) throws Exception {
+    return null;
+   }
 }