Refactor CoapResponse with builder pattern

resp

Author: szysas

coap-cli/build.gradle.kts

@@ -17,6 +17,7 @@ dependencies {
     testImplementation("org.junit.jupiter:junit-jupiter-api:5.10.0")
     testRuntimeOnly("org.junit.jupiter:junit-jupiter-engine:5.10.0")
     testImplementation("org.awaitility:awaitility:4.2.0")
+    testImplementation(testFixtures(project(":coap-core")))
 }
 
 tasks {

coap-cli/src/main/java/com/mbed/coap/cli/DeviceEmulator.java

@@ -16,7 +16,6 @@
  */
 package com.mbed.coap.cli;
 
-import static java.util.concurrent.CompletableFuture.completedFuture;
 import com.mbed.coap.client.RegistrationManager;
 import com.mbed.coap.packet.CoapRequest;
 import com.mbed.coap.packet.CoapResponse;
@@ -77,20 +76,20 @@ public Integer call() throws Exception {
     }
 
     protected Service<CoapRequest, CoapResponse> createRouting() {
-        Service<CoapRequest, CoapResponse> timeResource = __ -> completedFuture(CoapResponse.ok(Instant.now().toString()));
+        Service<CoapRequest, CoapResponse> timeResource = __ -> CoapResponse.ok(Instant.now().toString()).toFuture();
 
         scheduledExecutor.scheduleAtFixedRate(() ->
                         obsManager.sendObservation("/time", timeResource),
                 30, 30, TimeUnit.SECONDS
         );
 
         return RouterService.builder()
-                .get("/3/0/1", __ -> completedFuture(CoapResponse.ok("Acme")))
-                .get("/3/0/2", __ -> completedFuture(CoapResponse.ok("Emulator")))
-                .get("/3/0/3", __ -> completedFuture(CoapResponse.ok("0.0.1")))
+                .get("/3/0/1", __ -> CoapResponse.ok("Acme").toFuture())
+                .get("/3/0/2", __ -> CoapResponse.ok("Emulator").toFuture())
+                .get("/3/0/3", __ -> CoapResponse.ok("0.0.1").toFuture())
                 .get("/delayed-10s", __ -> {
                     CompletableFuture<CoapResponse> promise = new CompletableFuture<>();
-                    scheduledExecutor.schedule(() -> promise.complete(CoapResponse.ok("OK")), 10, TimeUnit.SECONDS);
+                    scheduledExecutor.schedule(() -> promise.complete(CoapResponse.ok("OK").build()), 10, TimeUnit.SECONDS);
                     return promise;
                 })
                 .get("/time", obsManager.then(timeResource))

coap-cli/src/test/java/com/mbed/coap/cli/CoapCliTest.java

@@ -16,16 +16,17 @@
 package com.mbed.coap.cli;
 
 import static com.mbed.coap.packet.BlockSize.S_16;
+import static com.mbed.coap.packet.CoapRequest.get;
+import static com.mbed.coap.packet.CoapResponse.coapResponse;
+import static com.mbed.coap.packet.CoapOptionsBuilder.options;
 import static com.mbed.coap.packet.CoapResponse.ok;
 import static com.mbed.coap.transport.udp.DatagramSocketTransport.udp;
 import static java.lang.String.format;
-import static java.util.concurrent.CompletableFuture.completedFuture;
 import static org.awaitility.Awaitility.await;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import com.mbed.coap.packet.CoapRequest;
-import com.mbed.coap.packet.CoapResponse;
 import com.mbed.coap.packet.Code;
 import com.mbed.coap.packet.MediaTypes;
 import com.mbed.coap.packet.Opaque;
@@ -54,10 +55,10 @@ void beforeAll() throws IOException {
                 .route(RouterService.builder()
                         .post("/rd", req -> {
                             String epName = req.options().getUriQueryMap().get("ep");
-                            return completedFuture(CoapResponse.of(Code.C201_CREATED, Opaque.EMPTY, o -> o.locationPath("/rd/" + epName)));
+                            return coapResponse(Code.C201_CREATED).locationPath("/rd/" + epName).toFuture();
                         })
-                        .get("/test", __ -> completedFuture(ok("Dziala!")))
-                        .post("/test", req -> completedFuture(ok("Received " + req.getPayload().size())))
+                        .get("/test", __ -> ok("Dziala!").toFuture())
+                        .post("/test", req -> ok("Received " + req.getPayload().size()).toFuture())
 
                 )
                 .build()

coap-core/src/jmh/java/microbenchmark/ServerBenchmark.java

@@ -15,7 +15,6 @@
  */
 package microbenchmark;
 
-import static java.util.concurrent.CompletableFuture.completedFuture;
 import ch.qos.logback.classic.Level;
 import ch.qos.logback.classic.Logger;
 import com.mbed.coap.exception.CoapException;
@@ -66,10 +65,9 @@ public void setup() throws CoapException, IOException {
         server = CoapServer.builder()
                 .transport(mockTransport)
                 .route(RouterService.builder()
-                        .get("/path1/sub2/sub3", __ -> completedFuture(
-                                CoapResponse.ok("1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890")
-                                        .maxAge(requestCounter++)
-                        ))
+                        .get("/path1/sub2/sub3", __ ->
+                                CoapResponse.ok("1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890").maxAge((int) requestCounter++).toFuture()
+                        )
                 )
                 .build();
         server.start();

coap-core/src/main/java/com/mbed/coap/exception/CoapCodeException.java

@@ -1,5 +1,5 @@
-/**
- * Copyright (C) 2022 java-coap contributors (https://github.com/open-coap/java-coap)
+/*
+ * Copyright (C) 2022-2023 java-coap contributors (https://github.com/open-coap/java-coap)
  * Copyright (C) 2011-2018 ARM Limited. All rights reserved.
  * SPDX-License-Identifier: Apache-2.0
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,6 +16,7 @@
  */
 package com.mbed.coap.exception;
 
+import static com.mbed.coap.packet.CoapResponse.coapResponse;
 import com.mbed.coap.packet.CoapResponse;
 import com.mbed.coap.packet.Code;
 
@@ -46,7 +47,7 @@ public Code getCode() {
         return code;
     }
 
-    public CoapResponse toResponse() {
-        return CoapResponse.of(code, getMessage());
+    public CoapResponse.Builder toResponse() {
+        return coapResponse(code).payload(getMessage());
     }
 }

coap-core/src/main/java/com/mbed/coap/exception/CoapRequestEntityTooLarge.java

@@ -1,5 +1,5 @@
-/**
- * Copyright (C) 2022 java-coap contributors (https://github.com/open-coap/java-coap)
+/*
+ * Copyright (C) 2022-2023 java-coap contributors (https://github.com/open-coap/java-coap)
  * Copyright (C) 2011-2018 ARM Limited. All rights reserved.
  * SPDX-License-Identifier: Apache-2.0
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -38,14 +38,15 @@ public CoapRequestEntityTooLarge(BlockOption blockOptionHint, String message) {
     }
 
     @Override
-    public CoapResponse toResponse() {
-        CoapResponse resp = super.toResponse();
-        if (maxSize > 0) {
-            resp.options().setSize1(maxSize);
-        }
-        if (blockOptionHint != null) {
-            resp.options().setBlock1Req(blockOptionHint);
-        }
-        return resp;
+    public CoapResponse.Builder toResponse() {
+        return super.toResponse()
+                .options(o -> {
+                    if (maxSize > 0) {
+                        o.size1(maxSize);
+                    }
+                    if (blockOptionHint != null) {
+                        o.block1Req(blockOptionHint);
+                    }
+                });
     }
 }

coap-core/src/main/java/com/mbed/coap/packet/CoapOptionsBuilder.java

@@ -30,7 +30,7 @@ public static CoapOptionsBuilder from(HeaderOptions options) {
         return new CoapOptionsBuilder(options.duplicate());
     }
 
-    CoapOptionsBuilder(HeaderOptions options) {
+    private CoapOptionsBuilder(HeaderOptions options) {
         this.options = options;
     }