Skip to content
Snippets Groups Projects
Commit 971bbcc7 authored by OZGCloud's avatar OZGCloud
Browse files

OZG-4392 add timeout to command request

parent d3d1f69a
No related branches found
No related tags found
No related merge requests found
......@@ -14,6 +14,7 @@ import net.devh.boot.grpc.client.inject.GrpcClient;
public class GrpcOzgCloudCommandService implements OzgCloudCommandService {
private static final int WAIT_TIME_MS = 500;
public static final int DEFAULT_COMMAND_REQUEST_THRESHOLD_MILLIS = 10000;
@GrpcClient("command-manager")
private final CommandServiceBlockingStub commandServiceStub;
......@@ -22,6 +23,8 @@ public class GrpcOzgCloudCommandService implements OzgCloudCommandService {
private final OzgCloudCallContextProvider contextProvider;
private final int commandRequestThresholdMillis;
@Override
public OzgCloudCommand createAndWaitUntilDone(OzgCloudCommand commandToCreate) {
return waitUntilDone(create(commandToCreate));
......@@ -34,8 +37,8 @@ public class GrpcOzgCloudCommandService implements OzgCloudCommandService {
OzgCloudCommand waitUntilDone(OzgCloudCommand commandToWaitFor) {
var command = commandToWaitFor;
// TODO timeout
while (!command.getStatus().isFinalState()) {
var timeout = System.currentTimeMillis() + commandRequestThresholdMillis;
while (!command.getStatus().isFinalState() && System.currentTimeMillis() < timeout) {
synchronized (this) {
try {
wait(WAIT_TIME_MS);
......@@ -48,7 +51,7 @@ public class GrpcOzgCloudCommandService implements OzgCloudCommandService {
return command;
}
private OzgCloudCommand reloadCommand(OzgCloudCommandId commandId) {
OzgCloudCommand reloadCommand(OzgCloudCommandId commandId) {
GrpcGetCommandRequest request = GrpcGetCommandRequest.newBuilder().setId(commandId.toString()).build();
return mapper.fromGrpc(getCommandServiceStub().getCommand(request));
}
......
......@@ -7,23 +7,29 @@ import static org.mockito.Mockito.*;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Nested;
import org.junit.jupiter.api.Test;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Spy;
import org.mockito.Mockito;
import de.itvsh.ozg.pluto.grpc.command.CommandServiceGrpc.CommandServiceBlockingStub;
import de.ozgcloud.apilib.common.callcontext.OzgCloudCallContextProvider;
import de.ozgcloud.apilib.common.command.OzgCloudCommand;
import de.ozgcloud.apilib.common.command.OzgCloudCommandStatus;
class GrpcOzgCloudCommandServiceTest {
@Spy
@InjectMocks
private GrpcOzgCloudCommandService service;
@Mock
private CommandServiceBlockingStub serviceStub;
@Mock
private CommandMapper mapper;
@Mock
private OzgCloudCallContextProvider contextProvider;
@BeforeEach
void init() {
service = spy(new GrpcOzgCloudCommandService(serviceStub, mapper, contextProvider, 600));
}
@Nested
class TestCreateAndWaitUntilDone {
......@@ -94,4 +100,14 @@ class GrpcOzgCloudCommandServiceTest {
}
}
@Test
void shouldTerminateByTimeout() {
var command = OzgCloudCommandTestFactory.createBuilder().status(OzgCloudCommandStatus.PENDING).build();
doReturn(command).when(service).reloadCommand(any());
service.waitUntilDone(command);
Mockito.verify(service, Mockito.times(2)).reloadCommand(command.getId());
}
}
......@@ -105,7 +105,8 @@ public class OzgCloudClientAutoConfiguration {
@ConditionalOnProperty("ozgcloud.command-manager.address")
OzgCloudCommandService grpcCommandService(@GrpcClient("command-manager") CommandServiceBlockingStub commandServiceStub,
CommandMapper commandMapper, OzgCloudCallContextProvider contextProvider) {
return new GrpcOzgCloudCommandService(commandServiceStub, commandMapper, contextProvider);
return new GrpcOzgCloudCommandService(commandServiceStub, commandMapper, contextProvider,
GrpcOzgCloudCommandService.DEFAULT_COMMAND_REQUEST_THRESHOLD_MILLIS);
}
@Bean
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment