1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 *
19 */
20 package org.apache.mina.example.sumup;
21
22 import org.apache.mina.common.IoHandler;
23 import org.apache.mina.common.IoHandlerAdapter;
24 import org.apache.mina.common.IoSession;
25 import org.apache.mina.example.sumup.message.AddMessage;
26 import org.apache.mina.example.sumup.message.ResultMessage;
27 import org.apache.mina.util.SessionLog;
28
29 /**
30 * {@link IoHandler} for SumUp client.
31 *
32 * @author The Apache Directory Project (mina-dev@directory.apache.org)
33 * @version $Rev: 555855 $, $Date: 2007-07-13 12:19:00 +0900 (Fri, 13 Jul 2007) $
34 */
35 public class ClientSessionHandler extends IoHandlerAdapter {
36 private final int[] values;
37
38 private boolean finished;
39
40 public ClientSessionHandler(int[] values) {
41 this.values = values;
42 }
43
44 public boolean isFinished() {
45 return finished;
46 }
47
48 public void sessionOpened(IoSession session) {
49 // send summation requests
50 for (int i = 0; i < values.length; i++) {
51 AddMessage m = new AddMessage();
52 m.setSequence(i);
53 m.setValue(values[i]);
54 session.write(m);
55 }
56 }
57
58 public void messageReceived(IoSession session, Object message) {
59 // server only sends ResultMessage. otherwise, we will have to identify
60 // its type using instanceof operator.
61 ResultMessage rm = (ResultMessage) message;
62 if (rm.isOk()) {
63 // server returned OK code.
64 // if received the result message which has the last sequence
65 // number,
66 // it is time to disconnect.
67 if (rm.getSequence() == values.length - 1) {
68 // print the sum and disconnect.
69 SessionLog.info(session, "The sum: " + rm.getValue());
70 session.close();
71 finished = true;
72 }
73 } else {
74 // seever returned error code because of overflow, etc.
75 SessionLog.warn(session, "Server error, disconnecting...");
76 session.close();
77 finished = true;
78 }
79 }
80
81 public void exceptionCaught(IoSession session, Throwable cause) {
82 session.close();
83 }
84 }