mirror of
https://github.com/google/nomulus.git
synced 2025-05-01 12:37:52 +02:00
5368489987
The callsites were inconsistent between whether they were passing empty list or null, and many of the ones that were passing null were not correctly annotated with @Nullable. I'm now going with empty list throughout except for the final step where the actual field that will be transformed into XML is set, where it is coerced to null to avoid an empty element in the XML output. ------------- Created by MOE: https://github.com/google/moe MOE_MIGRATED_REVID=139340837
89 lines
3.7 KiB
Java
89 lines
3.7 KiB
Java
// Copyright 2016 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package google.registry.flows.poll;
|
|
|
|
import static google.registry.flows.FlowUtils.validateClientIsLoggedIn;
|
|
import static google.registry.flows.poll.PollFlowUtils.getPollMessagesQuery;
|
|
import static google.registry.model.eppoutput.Result.Code.SUCCESS_WITH_ACK_MESSAGE;
|
|
import static google.registry.model.eppoutput.Result.Code.SUCCESS_WITH_NO_MESSAGES;
|
|
|
|
import com.googlecode.objectify.Key;
|
|
import google.registry.flows.EppException;
|
|
import google.registry.flows.EppException.ParameterValueSyntaxErrorException;
|
|
import google.registry.flows.ExtensionManager;
|
|
import google.registry.flows.Flow;
|
|
import google.registry.flows.FlowModule.ClientId;
|
|
import google.registry.flows.FlowModule.PollMessageId;
|
|
import google.registry.model.eppoutput.EppResponse;
|
|
import google.registry.model.poll.MessageQueueInfo;
|
|
import google.registry.model.poll.PollMessage;
|
|
import google.registry.model.poll.PollMessageExternalKeyConverter;
|
|
import google.registry.util.Clock;
|
|
import javax.inject.Inject;
|
|
import org.joda.time.DateTime;
|
|
|
|
/**
|
|
* An EPP flow for requesting {@link PollMessage}s.
|
|
*
|
|
* <p>This flow uses an eventually consistent Datastore query to return the oldest poll message for
|
|
* the registrar, as well as the total number of pending messages. Note that poll messages whose
|
|
* event time is in the future (i.e. they are speculative and could still be changed or rescinded)
|
|
* are ignored. The externally visible id for the poll message that the registrar sees is generated
|
|
* by {@link PollMessageExternalKeyConverter}.
|
|
*
|
|
* @error {@link PollRequestFlow.UnexpectedMessageIdException}
|
|
*/
|
|
public class PollRequestFlow implements Flow {
|
|
|
|
@Inject ExtensionManager extensionManager;
|
|
@Inject @ClientId String clientId;
|
|
@Inject @PollMessageId String messageId;
|
|
@Inject Clock clock;
|
|
@Inject EppResponse.Builder responseBuilder;
|
|
@Inject PollRequestFlow() {}
|
|
|
|
@Override
|
|
public final EppResponse run() throws EppException {
|
|
extensionManager.validate(); // There are no legal extensions for this flow.
|
|
validateClientIsLoggedIn(clientId);
|
|
if (!messageId.isEmpty()) {
|
|
throw new UnexpectedMessageIdException();
|
|
}
|
|
// Return the oldest message from the queue.
|
|
DateTime now = clock.nowUtc();
|
|
PollMessage pollMessage = getPollMessagesQuery(clientId, now).first().now();
|
|
if (pollMessage == null) {
|
|
return responseBuilder.setResultFromCode(SUCCESS_WITH_NO_MESSAGES).build();
|
|
}
|
|
return responseBuilder
|
|
.setResultFromCode(SUCCESS_WITH_ACK_MESSAGE)
|
|
.setMessageQueueInfo(new MessageQueueInfo.Builder()
|
|
.setQueueDate(pollMessage.getEventTime())
|
|
.setMsg(pollMessage.getMsg())
|
|
.setQueueLength(getPollMessagesQuery(clientId, now).count())
|
|
.setMessageId(PollMessage.EXTERNAL_KEY_CONVERTER.convert(Key.create(pollMessage)))
|
|
.build())
|
|
.setMultipleResData(pollMessage.getResponseData())
|
|
.setExtensions(pollMessage.getResponseExtensions())
|
|
.build();
|
|
}
|
|
|
|
/** Unexpected message id. */
|
|
static class UnexpectedMessageIdException extends ParameterValueSyntaxErrorException {
|
|
public UnexpectedMessageIdException() {
|
|
super("Unexpected message id");
|
|
}
|
|
}
|
|
}
|