import Java.util.concurrent.CountDownLatch;
import quickfix.Initiator;
public class UserSession {
private final CountDownLatch latch = new CountDownLatch(1);
public String await() {
try {
System.out.println("waiting...");
if (latch.await(5, TimeUnit.SECONDS))
System.out.println("released!");
else
System.out.println("timed out");
return secret;
} catch (InterruptedException e) {
// TODO Auto-generated catch block
System.out.println(e.getMessage());
e.printStackTrace();
}
return null;
}
public void countdown(String s) {
System.out.println("In countdown: "+s+ ". Latch count: "+latch.getCount());
secret = s;
latch.countDown();
System.out.println("Latch count: "+latch.getCount());
}
}
public class LogonHandler extends AbstractHandler {
public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response)
throws IOException, ServletException
{
Map<String,String[]> query = request.getParameterMap();
if (query.containsKey("method")) {
if (query.get("method")[0].compareTo(method) == 0) {
baseRequest.setHandled(true);
response.getWriter().println(logon(query));
}
}
else
baseRequest.setHandled(false);
}
private String logon(Map<String,String[]> query) {
if (query.containsKey("username") && query.containsKey("password") && query.containsKey("sendercompid")) {
app.mapUser(query.get("sendercompid")[0], new UserSession(query.get("username")[0], query.get("password")[0]));
SessionID session = new SessionID(new BeginString("FIX.4.4"), new SenderCompID(query.get("sendercompid")[0]), new TargetCompID("PARFX"));
try {
ThreadedSocketInitiator tsi = new ThreadedSocketInitiator(app, app.getFileStoreFactory(), settings, app.getLogFactory(), app.getMessageFactory());
UserSession userSession = new UserSession(query.get("username")[0], query.get("password")[0]);
userSession.setInitiator(tsi);
tsi.start();
return userSession.await();
} catch (ConfigError e) {
// TODO Auto-generated catch block
e.printStackTrace();
return e.toString();
}
}
return "fail";
}
}
public class QuickfixjApplication implements Application {
private Map<String,UserSession> users = new HashMap<String,UserSession>();
public void mapUser(String s, UserSession u) {
users.put(s, u);
}
public void toAdmin(Message message, SessionID sessionId) {
try {
if (message.getHeader().getField(new StringField(MsgType.FIELD)).valueEquals(Logon.MSGTYPE)) {
UserSession user = users.get(sessionId.getSenderCompID());
message.setField(new Username(user.getUsername()));
message.setField(new Password(user.getPassword()));
}
} catch (FieldNotFound e) {
e.printStackTrace();
}
}
public void fromAdmin(Message message, SessionID sessionId)
throws FieldNotFound, IncorrectDataFormat, IncorrectTagValue, RejectLogon {
if (message.getHeader().getField(new StringField(MsgType.FIELD)).valueEquals(Logon.MSGTYPE)) {
System.out.println(message.toString());
UserSession user = users.get(sessionId.getSenderCompID());
user.countdown(message.toString());
}
}
}
ここでは、最小限のコードのみを含めようとしました。 3つの興味深いクラスがあります。UserSessionは、JettyハンドラーとQuickFix/jアプリケーションの間の内部的な接着剤です。
LogonHandlerはHTTPログオン要求を受信し、QuickFix/jアプリケーションセッションにユーザーをログオンさせようとします。
QuickFix/jがログオンメッセージをFIXサーバーに送信しています。このログオン要求/応答は非同期です。もちろん、HTTPログオン要求は同期です。したがって、HTTPリクエストから戻る前に、FIXサーバーからの応答を待つ必要があります。これは、CountDownLatchとこのUserSessionオブジェクトを使用して行います。
QuickFix/jセッションオブジェクトを作成するときに、UserSessionオブジェクトも作成し、それをマップに追加します(これはLogonHandlerログオンメソッドで発生します)。
QuickFix/jアプリケーションオブジェクトには、toAdmin()とfromAdmin()の2つのコールバックがあります。 fromAdmin()で、メッセージがログオン応答かどうかを確認し、それがUserSessionのメソッドを呼び出してラッチをカウントダウンします。コードをデバッグすると、fromAdmin()メソッドがヒットし、UserSessionオブジェクトがマップで見つかり、countdown()メソッドが呼び出され、latch.getCount()が1から0に変わりますが、latch.await( )UserSession await()のメソッドが戻ることはありません。常にタイムアウトします。
次のようにCountDownLatch
を使用できます。
_public class LogonHandler implements Handler {
private final CountDownLatch loginLatch = new CountDownLatch (1);
private boolean callbackResults;
public void serverResponseCallback(boolean result) {
callbackResults = result;
loginLatch.countDown ();
}
public boolean tryLogon(Credentials creds) throws InterruptedException {
SomeServer server = new SomeServer(address);
server.tryLogon (creds.getName (), creds.getPass ());
loginLatch.await ();
return callbackResults;
}
}
_
待機時間を5秒などに制限したい場合は、loginLatch.await ()
の代わりに以下を使用します。
_if (loginLatch.await (5L, TimeUnit.SECONDS))
return callbackResults;
else
return false; // Timeout exceeded
_