From fbd1ad520352b6aaa861af2468f8914880a12032 Mon Sep 17 00:00:00 2001
From: Henning Baldersheim <balder@yahoo-inc.com>
Date: Wed, 6 Oct 2021 06:40:23 +0200
Subject: Attach context

---
 .../src/main/java/com/yahoo/jdisc/ProxyRequestHandler.java    | 11 ++++++++---
 jdisc_core/src/main/java/com/yahoo/jdisc/Request.java         |  2 +-
 .../src/main/java/com/yahoo/jdisc/core/ContainerSnapshot.java |  5 +++++
 .../main/java/com/yahoo/jdisc/core/TimeoutManagerImpl.java    |  5 +++++
 .../java/com/yahoo/jdisc/handler/ThreadedRequestHandler.java  |  2 +-
 5 files changed, 20 insertions(+), 5 deletions(-)

(limited to 'jdisc_core/src/main')

diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/ProxyRequestHandler.java b/jdisc_core/src/main/java/com/yahoo/jdisc/ProxyRequestHandler.java
index 6e385535e40..4e9e45f2f6e 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/ProxyRequestHandler.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/ProxyRequestHandler.java
@@ -62,6 +62,11 @@ class ProxyRequestHandler implements DelegatedRequestHandler {
         return delegate.refer();
     }
 
+    @Override
+    public ResourceReference refer(Object context) {
+        return delegate.refer(context);
+    }
+
     @Override
     public void release() {
         delegate.release();
@@ -89,7 +94,7 @@ class ProxyRequestHandler implements DelegatedRequestHandler {
             Objects.requireNonNull(delegate, "delegate");
             this.request = request;
             this.delegate = delegate;
-            this.requestReference = request.refer();
+            this.requestReference = request.refer(this);
         }
 
         @Override
@@ -121,7 +126,7 @@ class ProxyRequestHandler implements DelegatedRequestHandler {
             Objects.requireNonNull(delegate, "delegate");
             this.request = request;
             this.delegate = delegate;
-            this.requestReference = request.refer();
+            this.requestReference = request.refer(this);
         }
 
         @Override
@@ -169,7 +174,7 @@ class ProxyRequestHandler implements DelegatedRequestHandler {
 
         public ProxyCompletionHandler(SharedResource request, CompletionHandler delegate) {
             this.delegate = delegate;
-            this.requestReference = request.refer();
+            this.requestReference = request.refer(this);
         }
 
         @Override
diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/Request.java b/jdisc_core/src/main/java/com/yahoo/jdisc/Request.java
index eeb787e59dd..a6e6e734e6f 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/Request.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/Request.java
@@ -121,7 +121,7 @@ public class Request extends AbstractResource {
         creationTime = parent.container().currentTimeMillis();
         serverRequest = false;
         setUri(uri);
-        parentReference = this.parent.refer();
+        parentReference = this.parent.refer(this);
     }
 
     /** Returns the {@link Container} for which this Request was created */
diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/core/ContainerSnapshot.java b/jdisc_core/src/main/java/com/yahoo/jdisc/core/ContainerSnapshot.java
index becfa94e71f..33bec0ec6dc 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/core/ContainerSnapshot.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/core/ContainerSnapshot.java
@@ -100,6 +100,11 @@ class ContainerSnapshot extends AbstractResource implements Container {
             return delegate.refer();
         }
 
+        @Override
+        public ResourceReference refer(Object context) {
+            return delegate.refer(context);
+        }
+
         @Override
         public void release() {
             delegate.release();
diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/core/TimeoutManagerImpl.java b/jdisc_core/src/main/java/com/yahoo/jdisc/core/TimeoutManagerImpl.java
index a6d6e32df06..33ed23c9eee 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/core/TimeoutManagerImpl.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/core/TimeoutManagerImpl.java
@@ -144,6 +144,11 @@ public class TimeoutManagerImpl {
             return delegate.refer();
         }
 
+        @Override
+        public ResourceReference refer(Object context) {
+            return delegate.refer(context);
+        }
+
         @Override
         public void release() {
             delegate.release();
diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/handler/ThreadedRequestHandler.java b/jdisc_core/src/main/java/com/yahoo/jdisc/handler/ThreadedRequestHandler.java
index 6e2895f118b..97e68c72658 100644
--- a/jdisc_core/src/main/java/com/yahoo/jdisc/handler/ThreadedRequestHandler.java
+++ b/jdisc_core/src/main/java/com/yahoo/jdisc/handler/ThreadedRequestHandler.java
@@ -150,7 +150,7 @@ public abstract class ThreadedRequestHandler extends AbstractRequestHandler {
             this.request = request;
             this.content = content;
             this.responseHandler = responseHandler;
-            this.requestReference = request.refer();
+            this.requestReference = request.refer(this);
         }
 
         @Override
-- 
cgit v1.2.3