这是indexloc提供的服务,不要输入任何密码
Skip to content
This repository was archived by the owner on Mar 4, 2021. It is now read-only.
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
28 changes: 28 additions & 0 deletions src/main/java/com/netflix/simianarmy/aws/SimpleDBRecorder.java
Original file line number Diff line number Diff line change
Expand Up @@ -28,10 +28,15 @@
import java.util.Set;

import org.apache.commons.lang.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.amazonaws.AmazonClientException;
import com.amazonaws.services.simpledb.AmazonSimpleDB;
import com.amazonaws.services.simpledb.model.Attribute;
import com.amazonaws.services.simpledb.model.CreateDomainRequest;
import com.amazonaws.services.simpledb.model.Item;
import com.amazonaws.services.simpledb.model.ListDomainsResult;
import com.amazonaws.services.simpledb.model.PutAttributesRequest;
import com.amazonaws.services.simpledb.model.ReplaceableAttribute;
import com.amazonaws.services.simpledb.model.SelectRequest;
Expand All @@ -45,6 +50,8 @@
*/
@SuppressWarnings("serial")
public class SimpleDBRecorder implements MonkeyRecorder {
/** The Constant LOGGER. */
private static final Logger LOGGER = LoggerFactory.getLogger(SimpleDBRecorder.class);

private final AmazonSimpleDB simpleDBClient;

Expand Down Expand Up @@ -244,4 +251,25 @@ public List<Event> findEvents(Enum monkeyType, Enum eventType, Map<String, Strin
copy.put(Keys.eventType.name(), enumToValue(eventType));
return findEvents(copy, after);
}

/**
* Creates the SimpleDB domain, if it does not already exist.
*/
public void init() {
try {
ListDomainsResult listDomains = sdbClient().listDomains();
for (String d : listDomains.getDomainNames()) {
if (d.equals(domain)) {
LOGGER.debug("SimpleDB domain found: {}", domain);
return;
}
}
LOGGER.info("Creating SimpleDB domain: {}", domain);
CreateDomainRequest createDomainRequest = new CreateDomainRequest(
domain);
sdbClient().createDomain(createDomainRequest);
} catch (AmazonClientException e) {
LOGGER.warn("Error while trying to auto-create SimpleDB domain", e);
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -150,7 +150,9 @@ private void createScheduler() {
private void createRecorder() {
String domain = config.getStrOrElse("simianarmy.recorder.sdb.domain", "SIMIAN_ARMY");
if (client != null) {
setRecorder(new SimpleDBRecorder(client, domain));
SimpleDBRecorder simpleDbRecorder = new SimpleDBRecorder(client, domain);
simpleDbRecorder.init();
setRecorder(simpleDbRecorder);
}
}

Expand Down