diff options
Diffstat (limited to 'vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java')
-rwxr-xr-x | vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java | 80 |
1 files changed, 11 insertions, 69 deletions
diff --git a/vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java b/vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java index dd1e5858aee..c644b551a79 100755 --- a/vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java +++ b/vespaclient-core/src/main/java/com/yahoo/feedapi/DummySessionFactory.java @@ -2,8 +2,6 @@ package com.yahoo.feedapi; import com.yahoo.document.Document; -import com.yahoo.documentapi.VisitorParameters; -import com.yahoo.documentapi.VisitorSession; import com.yahoo.documentapi.messagebus.protocol.PutDocumentMessage; import com.yahoo.jdisc.Metric; import com.yahoo.messagebus.EmptyReply; @@ -24,97 +22,41 @@ public class DummySessionFactory implements SessionFactory { } public final List<Message> messages; - private boolean autoReply = false; - private ReplyFactory autoReplyFactory = null; - private Error autoError; - private int sessionsCreated = 0; - OutputStream output = null; + private boolean autoReply; + private OutputStream output = null; - protected DummySessionFactory() { - messages = new ArrayList<>(); - } - - public static DummySessionFactory createDefault() { - return new DummySessionFactory(); - } - - protected DummySessionFactory(boolean autoReply) { + private DummySessionFactory(boolean autoReply) { this.autoReply = autoReply; messages = new ArrayList<>(); } - protected DummySessionFactory(ReplyFactory autoReplyFactory) { - this.autoReply = true; - this.autoReplyFactory = autoReplyFactory; - messages = new ArrayList<>(); - } - - public static DummySessionFactory createWithAutoReplyFactory(ReplyFactory autoReplyFactory) { - return new DummySessionFactory(autoReplyFactory); - } - - protected DummySessionFactory(Error e) { - autoReply = true; - this.autoError = e; - messages = new ArrayList<>(); - } - - public static DummySessionFactory createWithErrorAutoReply(Error e) { - return new DummySessionFactory(e); - } - public static DummySessionFactory createWithAutoReply() { return new DummySessionFactory(true); } - public DummySessionFactory(Error e, OutputStream out) { + public DummySessionFactory(OutputStream out) { messages = null; autoReply = true; output = out; } - public int sessionsCreated() { - return sessionsCreated; - } - void add(Message m) { + private void add(Message m) { if (messages != null) { messages.add(m); } - } @Override public SendSession createSendSession(ReplyHandler r, Metric metric) { - ++sessionsCreated; - if (output != null) { return new DumpDocuments(output, r, this); } if (autoReply) { - return new AutoReplySession(r, autoReplyFactory, autoError, this); + return new AutoReplySession(r, null, null, this); } return new DummySession(r, this); } - @Override - public VisitorSession createVisitorSession(VisitorParameters p) { - return null; - } - - public void sendReply(Message m, Error error) { - MyContext ctxt = (MyContext) m.getContext(); - - Reply r = new EmptyReply(); - r.setMessage(m); - r.setContext(ctxt.oldContext); - - if (error != null) { - r.addError(error); - } - - ctxt.handler.handleReply(r); - } - private class MyContext { MyContext(ReplyHandler handler, Object ctxt) { this.handler = handler; @@ -132,7 +74,7 @@ public class DummySessionFactory implements SessionFactory { Error e; DummySessionFactory owner; - public AutoReplySession(ReplyHandler handler, ReplyFactory replyFactory, + AutoReplySession(ReplyHandler handler, ReplyFactory replyFactory, Error e, DummySessionFactory owner) { this.handler = handler; this.replyFactory = replyFactory; @@ -145,7 +87,7 @@ public class DummySessionFactory implements SessionFactory { } @Override - protected Result onSend(Message m, boolean blockIfQueueFull) throws InterruptedException { + protected Result onSend(Message m, boolean blockIfQueueFull) { owner.add(m); handleMessage(m); Reply r; @@ -174,7 +116,7 @@ public class DummySessionFactory implements SessionFactory { private class DumpDocuments extends AutoReplySession { final OutputStream out; - public DumpDocuments(OutputStream out, ReplyHandler r, DummySessionFactory factory) { + DumpDocuments(OutputStream out, ReplyHandler r, DummySessionFactory factory) { super(r, null, null, factory); this.out = out; } @@ -191,13 +133,13 @@ public class DummySessionFactory implements SessionFactory { ReplyHandler handler; DummySessionFactory owner; - public DummySession(ReplyHandler handler, DummySessionFactory owner) { + DummySession(ReplyHandler handler, DummySessionFactory owner) { this.handler = handler; this.owner = owner; } @Override - protected Result onSend(Message m, boolean blockIfQueueFull) throws InterruptedException { + protected Result onSend(Message m, boolean blockIfQueueFull) { m.setContext(new MyContext(handler, m.getContext())); owner.add(m); return Result.ACCEPTED; |