fix : thread safe token supplier

LogNet · Oct 15, 2020 · 145e771 · 145e771
1 parent 641e386
commit 145e771
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/...ent-spring-boot-starter/src/main/java/org/lognet/springboot/grpc/security/AuthHeader.java b/...ent-spring-boot-starter/src/main/java/org/lognet/springboot/grpc/security/AuthHeader.java
@@ -28,11 +28,9 @@ public AuthHeader.AuthHeaderBuilder basic(String userName, byte[] password) {
                     .put(password);
             buffer.rewind();
             ByteBuffer token = Base64.getEncoder().encode(buffer);
+            token.rewind();
             return authScheme(Constants.BASIC_AUTH_SCHEME)
-                    .tokenSupplier(() -> {
-                        token.rewind();
-                        return token;
-                    });
+                    .tokenSupplier( token::duplicate);
         }
 
 
@@ -41,7 +39,7 @@ public Metadata attach(Metadata metadataHeader){
         ByteBuffer token = tokenSupplier.get();
         final byte[] header = ByteBuffer.allocate(authScheme.length() + token.remaining() + 1)
                 .put(authScheme.getBytes())
-                .put((byte) ' ')
+                .put((byte)' ')
                 .put(token)
                 .array();
         metadataHeader.put(Constants.AUTH_HEADER_KEY,header);