mirror of
https://github.com/google/nomulus.git
synced 2025-06-10 06:24:45 +02:00
Use Dagger to @Inject WhoisReader instances
This is a precursor to adding metrics to WHOIS queries (as I'd like to be able to @Inject the metrics builders). ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=149418018
This commit is contained in:
parent
53785c2fc7
commit
bd7db61606
8 changed files with 51 additions and 51 deletions
|
@ -16,6 +16,7 @@ java_library(
|
|||
"//java/google/registry/xml",
|
||||
"//third_party/java/objectify:objectify-v4_1",
|
||||
"@com_google_appengine_api_1_0_sdk",
|
||||
"@com_google_auto_factory",
|
||||
"@com_google_auto_value",
|
||||
"@com_google_code_findbugs_jsr305",
|
||||
"@com_google_dagger",
|
||||
|
|
|
@ -26,24 +26,25 @@ public final class Whois {
|
|||
|
||||
private final Clock clock;
|
||||
private final String disclaimer;
|
||||
private final WhoisCommandFactory commandFactory;
|
||||
private final WhoisReaderFactory whoisReaderFactory;
|
||||
|
||||
@Inject
|
||||
public Whois(
|
||||
Clock clock,
|
||||
@Config("whoisDisclaimer") String disclaimer,
|
||||
@Config("whoisCommandFactory") WhoisCommandFactory commandFactory) {
|
||||
WhoisReaderFactory whoisReader) {
|
||||
this.clock = clock;
|
||||
this.disclaimer = disclaimer;
|
||||
this.commandFactory = commandFactory;
|
||||
this.whoisReaderFactory = whoisReader;
|
||||
}
|
||||
|
||||
/** Performs a WHOIS lookup on a plaintext query string. */
|
||||
public String lookup(String query, boolean preferUnicode) {
|
||||
DateTime now = clock.nowUtc();
|
||||
try {
|
||||
return new WhoisReader(new StringReader(query), commandFactory, now)
|
||||
.readCommand()
|
||||
return whoisReaderFactory
|
||||
.create(now)
|
||||
.readCommand(new StringReader(query))
|
||||
.executeQuery(now)
|
||||
.getPlainTextOutput(preferUnicode, disclaimer);
|
||||
} catch (WhoisException e) {
|
||||
|
|
|
@ -34,7 +34,6 @@ import google.registry.request.Response;
|
|||
import google.registry.util.Clock;
|
||||
import google.registry.util.FormattingLogger;
|
||||
import java.io.IOException;
|
||||
import java.io.Reader;
|
||||
import java.io.StringReader;
|
||||
import java.io.UnsupportedEncodingException;
|
||||
import java.net.URLDecoder;
|
||||
|
@ -132,7 +131,7 @@ public final class WhoisHttpServer implements Runnable {
|
|||
@Inject Response response;
|
||||
@Inject @Config("whoisDisclaimer") String disclaimer;
|
||||
@Inject @Config("whoisHttpExpires") Duration expires;
|
||||
@Inject @Config("whoisCommandFactory") WhoisCommandFactory commandFactory;
|
||||
@Inject WhoisReaderFactory whoisReaderFactory;
|
||||
@Inject @RequestPath String requestPath;
|
||||
@Inject WhoisHttpServer() {}
|
||||
|
||||
|
@ -143,10 +142,10 @@ public final class WhoisHttpServer implements Runnable {
|
|||
try {
|
||||
// Extremely permissive parsing that turns stuff like "/hello/world/" into "hello world".
|
||||
String command = decode(JOINER.join(SLASHER.split(path.substring(PATH.length())))) + "\r\n";
|
||||
Reader reader = new StringReader(command);
|
||||
DateTime now = clock.nowUtc();
|
||||
sendResponse(
|
||||
SC_OK, new WhoisReader(reader, commandFactory, now).readCommand().executeQuery(now));
|
||||
SC_OK,
|
||||
whoisReaderFactory.create(now).readCommand(new StringReader(command)).executeQuery(now));
|
||||
} catch (WhoisException e) {
|
||||
sendResponse(e.getStatus(), e);
|
||||
} catch (IOException e) {
|
||||
|
|
|
@ -20,15 +20,19 @@ import static google.registry.model.registry.Registries.findTldForName;
|
|||
import static google.registry.util.DomainNameUtils.canonicalizeDomainName;
|
||||
import static javax.servlet.http.HttpServletResponse.SC_BAD_REQUEST;
|
||||
|
||||
import com.google.auto.factory.AutoFactory;
|
||||
import com.google.auto.factory.Provided;
|
||||
import com.google.common.base.Joiner;
|
||||
import com.google.common.base.Optional;
|
||||
import com.google.common.io.CharStreams;
|
||||
import com.google.common.net.InetAddresses;
|
||||
import com.google.common.net.InternetDomainName;
|
||||
import google.registry.config.RegistryConfig.Config;
|
||||
import java.io.IOException;
|
||||
import java.io.Reader;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import javax.inject.Inject;
|
||||
import org.joda.time.DateTime;
|
||||
|
||||
/**
|
||||
|
@ -58,6 +62,7 @@ import org.joda.time.DateTime;
|
|||
* @see <a href="http://tools.ietf.org/html/rfc3912">RFC 3912</a>
|
||||
* @see <a href="http://www.iana.org/assignments/registrar-ids">IANA Registrar IDs</a>
|
||||
*/
|
||||
@AutoFactory
|
||||
class WhoisReader {
|
||||
|
||||
/**
|
||||
|
@ -68,15 +73,15 @@ class WhoisReader {
|
|||
static final String NAMESERVER_LOOKUP_COMMAND = "nameserver";
|
||||
static final String REGISTRAR_LOOKUP_COMMAND = "registrar";
|
||||
|
||||
private final Reader reader;
|
||||
private final DateTime now;
|
||||
private final WhoisCommandFactory commandFactory;
|
||||
|
||||
/** Creates a new WhoisReader that extracts its command from the specified Reader. */
|
||||
WhoisReader(Reader reader, WhoisCommandFactory commandFactory, DateTime now) {
|
||||
this.reader = checkNotNull(reader, "reader");
|
||||
this.commandFactory = checkNotNull(commandFactory, "commandFactory");
|
||||
this.now = checkNotNull(now, "now");
|
||||
@Inject
|
||||
WhoisReader(
|
||||
@Provided @Config("whoisCommandFactory") WhoisCommandFactory commandFactory, DateTime now) {
|
||||
this.commandFactory = commandFactory;
|
||||
this.now = now;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -86,8 +91,8 @@ class WhoisReader {
|
|||
* @throws IOException If the command could not be read from the reader.
|
||||
* @throws WhoisException If the command could not be parsed as a WhoisCommand.
|
||||
*/
|
||||
WhoisCommand readCommand() throws IOException, WhoisException {
|
||||
return parseCommand(CharStreams.toString(reader));
|
||||
WhoisCommand readCommand(Reader reader) throws IOException, WhoisException {
|
||||
return parseCommand(CharStreams.toString(checkNotNull(reader, "reader")));
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -97,7 +102,6 @@ class WhoisReader {
|
|||
private WhoisCommand parseCommand(String command) throws WhoisException {
|
||||
// Split the string into tokens based on whitespace.
|
||||
List<String> tokens = filterEmptyStrings(command.split("\\s"));
|
||||
|
||||
if (tokens.isEmpty()) {
|
||||
throw new WhoisException(now, SC_BAD_REQUEST, "No WHOIS command specified.");
|
||||
}
|
||||
|
|
|
@ -58,7 +58,7 @@ public class WhoisServer implements Runnable {
|
|||
@Inject Clock clock;
|
||||
@Inject Reader input;
|
||||
@Inject Response response;
|
||||
@Inject @Config("whoisCommandFactory") WhoisCommandFactory commandFactory;
|
||||
@Inject WhoisReaderFactory whoisReaderFactory;
|
||||
@Inject @Config("whoisDisclaimer") String disclaimer;
|
||||
@Inject WhoisServer() {}
|
||||
|
||||
|
@ -68,8 +68,9 @@ public class WhoisServer implements Runnable {
|
|||
DateTime now = clock.nowUtc();
|
||||
try {
|
||||
responseText =
|
||||
new WhoisReader(input, commandFactory, now)
|
||||
.readCommand()
|
||||
whoisReaderFactory
|
||||
.create(now)
|
||||
.readCommand(input)
|
||||
.executeQuery(now)
|
||||
.getPlainTextOutput(PREFER_UNICODE, disclaimer);
|
||||
} catch (WhoisException e) {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue