Skip to content

Commit 36bcc82

Browse files
Fix *SetContextOnSubscribe implementations
1 parent d530433 commit 36bcc82

File tree

3 files changed

+3
-51
lines changed

3 files changed

+3
-51
lines changed

servicetalk-concurrent-api/src/main/java/io/servicetalk/concurrent/api/CompletableSetContextOnSubscribe.java

+1-17
Original file line numberDiff line numberDiff line change
@@ -30,23 +30,7 @@ final class CompletableSetContextOnSubscribe extends AbstractNoHandleSubscribeCo
3030

3131
@Override
3232
CapturedContext contextForSubscribe(AsyncContextProvider provider) {
33-
CapturedContext parentContext = super.contextForSubscribe(provider);
34-
return new CapturedContext() {
35-
@Override
36-
public ContextMap captured() {
37-
return context;
38-
}
39-
40-
@Override
41-
public Scope attachContext() {
42-
Scope outer = parentContext.attachContext();
43-
Scope inner = AsyncContext.provider().attachContextMap(context);
44-
return () -> {
45-
inner.close();
46-
outer.close();
47-
};
48-
}
49-
};
33+
return provider.captureContext(context);
5034
}
5135

5236
@Override

servicetalk-concurrent-api/src/main/java/io/servicetalk/concurrent/api/PublisherSetContextOnSubscribe.java

+1-17
Original file line numberDiff line numberDiff line change
@@ -30,23 +30,7 @@ final class PublisherSetContextOnSubscribe<T> extends AbstractNoHandleSubscribeP
3030

3131
@Override
3232
CapturedContext contextForSubscribe(AsyncContextProvider provider) {
33-
CapturedContext parentContext = super.contextForSubscribe(provider);
34-
return new CapturedContext() {
35-
@Override
36-
public ContextMap captured() {
37-
return context;
38-
}
39-
40-
@Override
41-
public Scope attachContext() {
42-
Scope outer = parentContext.attachContext();
43-
Scope inner = AsyncContext.provider().attachContextMap(context);
44-
return () -> {
45-
inner.close();
46-
outer.close();
47-
};
48-
}
49-
};
33+
return provider.captureContext(context);
5034
}
5135

5236
@Override

servicetalk-concurrent-api/src/main/java/io/servicetalk/concurrent/api/SingleSetContextOnSubscribe.java

+1-17
Original file line numberDiff line numberDiff line change
@@ -30,23 +30,7 @@ final class SingleSetContextOnSubscribe<T> extends AbstractNoHandleSubscribeSing
3030

3131
@Override
3232
CapturedContext contextForSubscribe(AsyncContextProvider provider) {
33-
CapturedContext parentContext = super.contextForSubscribe(provider);
34-
return new CapturedContext() {
35-
@Override
36-
public ContextMap captured() {
37-
return context;
38-
}
39-
40-
@Override
41-
public Scope attachContext() {
42-
Scope outer = parentContext.attachContext();
43-
Scope inner = AsyncContext.provider().attachContextMap(context);
44-
return () -> {
45-
inner.close();
46-
outer.close();
47-
};
48-
}
49-
};
33+
return provider.captureContext(context);
5034
}
5135

5236
@Override

0 commit comments

Comments
 (0)