mirrors/nomulus
1
0
Fork 0
mirror of https://github.com/google/nomulus synced 2026-01-08 23:23:32 +00:00
Code Issues Releases Wiki Activity

Add global cursor functionality to UpdateCursorsCommand

Browse Source 
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=143965268
This commit is contained in:
ctingue
2017-01-09 08:28:29 -08:00
committed by Ben McIlwain
parent 3a5a7e030d
commit bca2169c68
2 changed files with 84 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
java/google/registry/tools/UpdateCursorsCommand.java
View File

@@ -15,6 +15,7 @@
package google.registry.tools;
import static google.registry.model.ofy.ObjectifyService.ofy;
import static google.registry.util.CollectionUtils.isNullOrEmpty;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;
@@ -29,9 +30,7 @@ import org.joda.time.DateTime;
@Parameters(separators = " =", commandDescription = "Modifies cursor timestamps used by LRC tasks")
final class UpdateCursorsCommand extends MutatingCommand {
@Parameter(
description = "TLDs on which to operate.",
required = true)
@Parameter(description = "TLDs on which to operate. Omit for global cursors.")
private List<String> tlds;
@Parameter(
@@ -49,12 +48,17 @@ final class UpdateCursorsCommand extends MutatingCommand {
@Override
protected void init() throws Exception {
for (String tld : tlds) {
Registry registry = Registry.get(tld);
Cursor cursor = ofy().load().key(Cursor.createKey(cursorType, registry)).now();
stageEntityChange(
cursor,
Cursor.create(cursorType, newTimestamp, registry));
if (isNullOrEmpty(tlds)) {
Cursor cursor = ofy().load().key(Cursor.createGlobalKey(cursorType)).now();
stageEntityChange(cursor, Cursor.createGlobal(cursorType, newTimestamp));
} else {
for (String tld : tlds) {
Registry registry = Registry.get(tld);
Cursor cursor = ofy().load().key(Cursor.createKey(cursorType, registry)).now();
stageEntityChange(
cursor,
Cursor.create(cursorType, newTimestamp, registry));
}
}
}
}