Flatten and inject the poll flows

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=133302791
This commit is contained in:
cgoldfeder 2016-09-15 13:46:12 -07:00 committed by Ben McIlwain
parent 939112318b
commit 4a723576d5
6 changed files with 107 additions and 131 deletions

View file

@ -24,6 +24,7 @@ import google.registry.model.eppoutput.EppResponse;
import google.registry.model.eppoutput.EppResponse.ResponseData;
import google.registry.model.eppoutput.EppResponse.ResponseExtension;
import google.registry.model.eppoutput.Result;
import google.registry.model.poll.MessageQueueInfo;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
@ -82,13 +83,22 @@ public abstract class Flow {
Result.Code code,
@Nullable ResponseData responseData,
@Nullable ImmutableList<? extends ResponseExtension> extensions) {
return createOutput(
code, responseData == null ? null : ImmutableList.of(responseData), extensions, null);
}
protected EppOutput createOutput(
Result.Code code,
@Nullable ImmutableList<ResponseData> responseData,
@Nullable ImmutableList<? extends ResponseExtension> responseExtensions,
@Nullable MessageQueueInfo messageQueueInfo) {
return EppOutput.create(new EppResponse.Builder()
.setTrid(trid)
.setResult(Result.create(code))
.setMessageQueueInfo(messageQueueInfo)
.setResData(responseData)
.setExtensions(responseExtensions)
.setExecutionTime(now)
.setCreatedRepoId(getCreatedRepoId())
.setResData(responseData == null ? null : ImmutableList.of(responseData))
.setExtensions(extensions)
.build());
}