Skip to content

Commit 8aeb40d

Browse files
XiaofeiCaoXiaofeiCao
committed
implementation
implementation
1 parent efccdba commit 8aeb40d

File tree

8 files changed

+107
-8
lines changed

8 files changed

+107
-8
lines changed

‎sdk/resourcemanager/azure-resourcemanager-compute/src/main/java/com/azure/resourcemanager/compute/implementation/VirtualMachineImpl.java‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1735,6 +1735,12 @@ public NetworkInterface getPrimaryNetworkInterface() {
17351735
return this.getPrimaryNetworkInterfaceAsync().block();
17361736
}
17371737

1738+
@Override
1739+
public NetworkInterface getPrimaryNetworkInterface(Context context) {
1740+
return this.getPrimaryNetworkInterfaceAsync().contextWrite(c -> FluxUtil.toReactorContext(context))
1741+
.block();
1742+
}
1743+
17381744
@Override
17391745
public Mono<NetworkInterface> getPrimaryNetworkInterfaceAsync() {
17401746
return this.networkManager.networkInterfaces().getByIdAsync(primaryNetworkInterfaceId());

‎sdk/resourcemanager/azure-resourcemanager-network/src/main/java/com/azure/resourcemanager/network/implementation/ApplicationSecurityGroupsImpl.java‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
// Licensed under the MIT License.
33
package com.azure.resourcemanager.network.implementation;
44

5+
import com.azure.core.util.Context;
6+
import com.azure.core.util.FluxUtil;
57
import com.azure.resourcemanager.network.NetworkManager;
68
import com.azure.resourcemanager.network.fluent.ApplicationSecurityGroupsClient;
79
import com.azure.resourcemanager.network.fluent.models.ApplicationSecurityGroupInner;
@@ -36,4 +38,11 @@ protected ApplicationSecurityGroupImpl wrapModel(ApplicationSecurityGroupInner i
3638
}
3739
return new ApplicationSecurityGroupImpl(inner.name(), inner, this.manager());
3840
}
41+
42+
@Override
43+
public ApplicationSecurityGroup getByResourceGroup(String resourceGroupName, String name, Context context) {
44+
return getByResourceGroupAsync(resourceGroupName, name)
45+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly()))
46+
.block();
47+
}
3948
}

‎sdk/resourcemanager/azure-resourcemanager-network/src/main/java/com/azure/resourcemanager/network/implementation/NetworkInterfacesImpl.java‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,4 +129,11 @@ public Accepted<Void> beginDeleteByResourceGroup(String resourceGroupName, Strin
129129
.block(),
130130
Function.identity(), Void.class, null, context);
131131
}
132+
133+
@Override
134+
public PagedIterable<NetworkInterface> listByResourceGroup(String resourceGroupName, Context context) {
135+
return this.inner().listByResourceGroup(resourceGroupName, context)
136+
.mapPage(networkInterfaceInner ->
137+
new NetworkInterfaceImpl(networkInterfaceInner.name(), networkInterfaceInner, NetworkInterfacesImpl.this.manager()));
138+
}
132139
}

‎sdk/resourcemanager/azure-resourcemanager-network/src/main/java/com/azure/resourcemanager/network/implementation/NetworkSecurityGroupsImpl.java‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,9 @@
22
// Licensed under the MIT License.
33
package com.azure.resourcemanager.network.implementation;
44

5+
import com.azure.core.util.Context;
56
import com.azure.core.util.CoreUtils;
7+
import com.azure.core.util.FluxUtil;
68
import com.azure.resourcemanager.network.NetworkManager;
79
import com.azure.resourcemanager.network.fluent.NetworkSecurityGroupsClient;
810
import com.azure.resourcemanager.network.fluent.models.NetworkSecurityGroupInner;
@@ -68,4 +70,11 @@ protected NetworkSecurityGroupImpl wrapModel(NetworkSecurityGroupInner inner) {
6870
}
6971
return new NetworkSecurityGroupImpl(inner.name(), inner, this.manager());
7072
}
73+
74+
@Override
75+
public NetworkSecurityGroup getByResourceGroup(String resourceGroupName, String name, Context context) {
76+
return getByResourceGroupAsync(resourceGroupName, name)
77+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly()))
78+
.block();
79+
}
7180
}

‎sdk/resourcemanager/azure-resourcemanager-network/src/main/java/com/azure/resourcemanager/network/implementation/NetworksImpl.java‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
// Licensed under the MIT License.
33
package com.azure.resourcemanager.network.implementation;
44

5+
import com.azure.core.util.Context;
6+
import com.azure.core.util.FluxUtil;
57
import com.azure.resourcemanager.network.NetworkManager;
68
import com.azure.resourcemanager.network.fluent.VirtualNetworksClient;
79
import com.azure.resourcemanager.network.fluent.models.SubnetInner;
@@ -71,4 +73,11 @@ protected NetworkImpl wrapModel(VirtualNetworkInner inner) {
7173
}
7274
return new NetworkImpl(inner.name(), inner, this.manager());
7375
}
76+
77+
@Override
78+
public Network getByResourceGroup(String resourceGroupName, String name, Context context) {
79+
return getByResourceGroupAsync(resourceGroupName, name)
80+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly()))
81+
.block();
82+
}
7483
}

‎sdk/resourcemanager/azure-resourcemanager-network/src/main/java/com/azure/resourcemanager/network/implementation/NicIpConfigurationBaseImpl.java‎

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
package com.azure.resourcemanager.network.implementation;
55

66
import com.azure.core.management.SubResource;
7+
import com.azure.core.util.Context;
78
import com.azure.core.util.CoreUtils;
9+
import com.azure.core.util.FluxUtil;
810
import com.azure.resourcemanager.network.NetworkManager;
911
import com.azure.resourcemanager.network.fluent.models.ApplicationSecurityGroupInner;
1012
import com.azure.resourcemanager.network.models.ApplicationGatewayBackend;
@@ -48,7 +50,7 @@ abstract class NicIpConfigurationBaseImpl<ParentImplT extends ParentT, ParentT e
4850
private final NetworkManager networkManager;
4951

5052
protected NicIpConfigurationBaseImpl(NetworkInterfaceIpConfigurationInner inner, ParentImplT parent,
51-
NetworkManager networkManager) {
53+
NetworkManager networkManager) {
5254
super(inner, parent);
5355
this.networkManager = networkManager;
5456
}
@@ -171,13 +173,20 @@ public List<LoadBalancerInboundNatRule> listAssociatedLoadBalancerInboundNatRule
171173

172174
@Override
173175
public List<ApplicationSecurityGroup> listAssociatedApplicationSecurityGroups() {
176+
return listAssociatedApplicationSecurityGroups(Context.NONE);
177+
}
178+
179+
@Override
180+
public List<ApplicationSecurityGroup> listAssociatedApplicationSecurityGroups(Context context) {
174181
if (CoreUtils.isNullOrEmpty(this.innerModel().applicationSecurityGroups())) {
175182
return Collections.emptyList();
176183
}
177184

178185
List<ApplicationSecurityGroup> applicationSecurityGroups = Flux
179186
.fromStream(this.innerModel().applicationSecurityGroups().stream().map(ApplicationSecurityGroupInner::id))
180-
.flatMapSequential(id -> this.networkManager.applicationSecurityGroups().getByIdAsync(id))
187+
.flatMapSequential(id ->
188+
this.networkManager.applicationSecurityGroups().getByIdAsync(id)
189+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly())))
181190
.collectList()
182191
.block();
183192
return applicationSecurityGroups == null

‎sdk/resourcemanager/azure-resourcemanager-resources/src/main/java/com/azure/resourcemanager/resources/implementation/DeploymentsImpl.java‎

Lines changed: 42 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,24 +5,33 @@
55

66
import com.azure.core.http.rest.PagedFlux;
77
import com.azure.core.http.rest.PagedIterable;
8+
import com.azure.core.util.Context;
89
import com.azure.core.util.CoreUtils;
10+
import com.azure.core.util.FluxUtil;
11+
import com.azure.core.util.logging.ClientLogger;
912
import com.azure.resourcemanager.resources.ResourceManager;
10-
import com.azure.resourcemanager.resources.models.Deployment;
11-
import com.azure.resourcemanager.resources.models.Deployments;
13+
import com.azure.resourcemanager.resources.fluent.DeploymentsClient;
14+
import com.azure.resourcemanager.resources.fluent.models.DeploymentExtendedInner;
1215
import com.azure.resourcemanager.resources.fluentcore.arm.ResourceUtils;
1316
import com.azure.resourcemanager.resources.fluentcore.arm.collection.implementation.SupportsGettingByResourceGroupImpl;
1417
import com.azure.resourcemanager.resources.fluentcore.arm.models.HasManager;
18+
import com.azure.resourcemanager.resources.fluentcore.model.Accepted;
19+
import com.azure.resourcemanager.resources.fluentcore.model.implementation.AcceptedImpl;
1520
import com.azure.resourcemanager.resources.fluentcore.utils.PagedConverter;
16-
import com.azure.resourcemanager.resources.fluent.models.DeploymentExtendedInner;
17-
import com.azure.resourcemanager.resources.fluent.DeploymentsClient;
21+
import com.azure.resourcemanager.resources.models.Deployment;
22+
import com.azure.resourcemanager.resources.models.Deployments;
1823
import reactor.core.publisher.Mono;
1924

25+
import java.util.function.Function;
26+
2027
/**
2128
* The implementation for {@link Deployments}.
2229
*/
2330
public final class DeploymentsImpl extends SupportsGettingByResourceGroupImpl<Deployment>
2431
implements Deployments, HasManager<ResourceManager> {
2532

33+
private final ClientLogger logger = new ClientLogger(this.getClass());
34+
2635
private final ResourceManager resourceManager;
2736

2837
public DeploymentsImpl(final ResourceManager resourceManager) {
@@ -103,6 +112,35 @@ public boolean checkExistence(String resourceGroupName, String deploymentName) {
103112
return this.manager().deploymentClient().getDeployments().checkExistence(resourceGroupName, deploymentName);
104113
}
105114

115+
@Override
116+
public Accepted<Void> beginDeleteById(String id) {
117+
return beginDeleteById(id, Context.NONE);
118+
}
119+
120+
@Override
121+
public Accepted<Void> beginDeleteById(String id, Context context) {
122+
return beginDeleteByResourceGroup(ResourceUtils.groupFromResourceId(id), ResourceUtils.nameFromResourceId(id), context);
123+
}
124+
125+
@Override
126+
public Accepted<Void> beginDeleteByResourceGroup(String resourceGroupName, String name) {
127+
return beginDeleteByResourceGroup(resourceGroupName, name, Context.NONE);
128+
}
129+
130+
@Override
131+
public Accepted<Void> beginDeleteByResourceGroup(String resourceGroupName, String name, Context context) {
132+
return AcceptedImpl.newAccepted(logger, this.manager().serviceClient().getHttpPipeline(),
133+
this.manager().serviceClient().getDefaultPollInterval(),
134+
() -> this.manager()
135+
.deploymentClient()
136+
.getDeployments()
137+
.deleteWithResponseAsync(resourceGroupName, name)
138+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly()))
139+
.block(),
140+
Function.identity(),
141+
Void.class, null, context);
142+
}
143+
106144
protected DeploymentImpl createFluentModel(String name) {
107145
return new DeploymentImpl(new DeploymentExtendedInner(), name, this.resourceManager);
108146
}

‎sdk/resourcemanager/azure-resourcemanager-resources/src/main/java/com/azure/resourcemanager/resources/implementation/ResourceGroupsImpl.java‎

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import com.azure.core.http.rest.PagedFlux;
77
import com.azure.core.http.rest.PagedIterable;
88
import com.azure.core.util.Context;
9+
import com.azure.core.util.FluxUtil;
910
import com.azure.core.util.logging.ClientLogger;
1011
import com.azure.resourcemanager.resources.fluentcore.model.Accepted;
1112
import com.azure.resourcemanager.resources.fluentcore.model.implementation.AcceptedImpl;
@@ -112,20 +113,31 @@ protected ResourceGroupImpl wrapModel(ResourceGroupInner inner) {
112113

113114
@Override
114115
public Accepted<Void> beginDeleteByName(String name) {
115-
return beginDeleteByName(name, null);
116+
return beginDeleteByName(name, Context.NONE);
117+
}
118+
119+
@Override
120+
public Accepted<Void> beginDeleteByName(String name, Context context) {
121+
return beginDeleteByName(name, null, context);
116122
}
117123

118124
@Override
119125
public Accepted<Void> beginDeleteByName(String name,
120126
Collection<ForceDeletionResourceType> forceDeletionResourceTypes) {
127+
return beginDeleteByName(name, forceDeletionResourceTypes, Context.NONE);
128+
}
129+
130+
@Override
131+
public Accepted<Void> beginDeleteByName(String name, Collection<ForceDeletionResourceType> forceDeletionResourceTypes, Context context) {
121132
return AcceptedImpl.newAccepted(logger, this.manager().serviceClient().getHttpPipeline(),
122133
this.manager().serviceClient().getDefaultPollInterval(),
123134
() -> this.manager()
124135
.serviceClient()
125136
.getResourceGroups()
126137
.deleteWithResponseAsync(name, forceDeletionTypes(forceDeletionResourceTypes))
138+
.contextWrite(c -> c.putAll(FluxUtil.toReactorContext(context).readOnly()))
127139
.block(),
128-
Function.identity(), Void.class, null, Context.NONE);
140+
Function.identity(), Void.class, null, context);
129141
}
130142

131143
// @Override

0 commit comments

Comments
 (0)