Merge pull request #86 from asabbi/feature/custom-attributes

Added ability to set attributes on GraphQLRequest

Author: oliemansm

gradle.properties

@@ -1,4 +1,4 @@
-version=1.0.1-SNAPSHOT
+version=1.1.0-SNAPSHOT
 
 PROJECT_GROUP = com.graphql-java-kickstart
 PROJECT_NAME = graphql-spring-webclient

graphql-webclient/src/main/java/graphql/kickstart/spring/webclient/boot/GraphQLRequest.java

@@ -2,6 +2,8 @@
 
 import org.springframework.http.HttpHeaders;
 
+import java.util.Map;
+
 public interface GraphQLRequest {
 
   static GraphQLRequestBuilder builder() {
@@ -12,4 +14,6 @@ static GraphQLRequestBuilder builder() {
 
   HttpHeaders getHeaders();
 
+  Map<String, Object> getAttributes();
+
 }

graphql-webclient/src/main/java/graphql/kickstart/spring/webclient/boot/GraphQLRequestBuilder.java

@@ -4,6 +4,9 @@
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
 import lombok.SneakyThrows;
 import org.springframework.core.io.ClassPathResource;
 import org.springframework.core.io.Resource;
@@ -12,12 +15,18 @@
 
 public class GraphQLRequestBuilder {
 
+  private final Map<String, Object> attributes = new LinkedHashMap<>();
   private final HttpHeaders headers = new HttpHeaders();
   private final GraphQLRequestBody.GraphQLRequestBodyBuilder bodyBuilder = GraphQLRequestBody.builder();
 
   GraphQLRequestBuilder() {
   }
 
+  public GraphQLRequestBuilder attribute(String name, Object value) {
+    attributes.put(name, value);
+    return this;
+  }
+
   public GraphQLRequestBuilder header(String name, String... values) {
     headers.addAll(name, Arrays.asList(values));
     return this;
@@ -54,7 +63,7 @@ public GraphQLRequestBuilder operationName(String operationName) {
   }
 
   public GraphQLRequest build() {
-    return new GraphQLRequestImpl(headers, bodyBuilder.build());
+    return new GraphQLRequestImpl(attributes, headers, bodyBuilder.build());
   }
 
 }

graphql-webclient/src/main/java/graphql/kickstart/spring/webclient/boot/GraphQLRequestImpl.java

@@ -3,9 +3,12 @@
 import lombok.Value;
 import org.springframework.http.HttpHeaders;
 
+import java.util.Map;
+
 @Value
 class GraphQLRequestImpl implements GraphQLRequest {
 
+  Map<String, Object> attributes;
   HttpHeaders headers;
   GraphQLRequestBody requestBody;
 

graphql-webclient/src/main/java/graphql/kickstart/spring/webclient/boot/GraphQLWebClientImpl.java

@@ -33,6 +33,8 @@ public <T> Mono<T> post(String resource, Map<String, Object> variables, Class<T>
   @Override
   public Mono<GraphQLResponse> post(GraphQLRequest request) {
     WebClient.RequestBodySpec spec = webClient.post().contentType(MediaType.APPLICATION_JSON);
+    request.getAttributes()
+        .forEach(spec::attribute);
     request.getHeaders()
         .forEach((header, values) -> spec.header(header, values.toArray(new String[0])));
     return spec.bodyValue(request.getRequestBody())