Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@

import com.google.common.annotations.VisibleForTesting;
import org.apache.hadoop.hdds.conf.OzoneConfiguration;
import org.apache.hadoop.ozone.OmUtils;
import org.apache.hadoop.ozone.client.OzoneClient;
import org.apache.hadoop.ozone.client.OzoneClientFactory;
import org.apache.hadoop.ozone.om.protocol.S3Auth;
Expand Down Expand Up @@ -62,9 +63,6 @@ public class OzoneClientProducer {
@Inject
private OzoneConfiguration ozoneConfiguration;

@Inject
private String omServiceID;

@Context
private ContainerRequestContext context;

Expand Down Expand Up @@ -115,6 +113,7 @@ OzoneClient createOzoneClient() throws IOException {
// Set the expected OM version if not set via config.
ozoneConfiguration.setIfUnset(OZONE_OM_CLIENT_PROTOCOL_VERSION_KEY,
OZONE_OM_CLIENT_PROTOCOL_VERSION);
String omServiceID = OmUtils.getOzoneManagerServiceId(ozoneConfiguration);
if (omServiceID == null) {
return OzoneClientFactory.getRpcClient(ozoneConfiguration);
} else {
Expand Down

This file was deleted.

Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,39 @@ public void testGetClientFailure() {
}
}

@Test
public void testGetClientFailureWithMultipleServiceIds() {
try {
OzoneConfiguration configuration = new OzoneConfiguration();
configuration.set(OMConfigKeys.OZONE_OM_SERVICE_IDS_KEY, "ozone1,ozone2");
producer.setOzoneConfiguration(configuration);
producer.createClient();
fail("testGetClientFailureWithMultipleServiceIds");
} catch (Exception ex) {
Assert.assertTrue(ex instanceof IOException);
Assert.assertTrue(ex.getMessage().contains(
"More than 1 OzoneManager ServiceID"));
}
}

@Test
public void testGetClientFailureWithMultipleServiceIdsAndInternalServiceId() {
try {
OzoneConfiguration configuration = new OzoneConfiguration();
configuration.set(OMConfigKeys.OZONE_OM_INTERNAL_SERVICE_ID, "ozone1");
configuration.set(OMConfigKeys.OZONE_OM_SERVICE_IDS_KEY, "ozone1,ozone2");
producer.setOzoneConfiguration(configuration);
producer.createClient();
fail("testGetClientFailureWithMultipleServiceIdsAndInternalServiceId");
} catch (Exception ex) {
Assert.assertTrue(ex instanceof IOException);
// Still test will fail, as config is not complete. But it should pass
// the service id check.
Assert.assertFalse(ex.getMessage().contains(
"More than 1 OzoneManager ServiceID"));
}
}

private void setupContext() throws Exception {
headerMap.putSingle(AUTHORIZATION_HEADER, authHeader);
headerMap.putSingle(CONTENT_MD5, contentMd5);
Expand Down