Fix keep-alive to filter correctly.
[iotivity.git] / cloud / interface / src / main / java / org / iotivity / cloud / ciserver / resources / KeepAliveResource.java
1 /*
2  * //******************************************************************
3  * //
4  * // Copyright 2016 Samsung Electronics All Rights Reserved.
5  * //
6  * //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
7  * //
8  * // Licensed under the Apache License, Version 2.0 (the "License");
9  * // you may not use this file except in compliance with the License.
10  * // You may obtain a copy of the License at
11  * //
12  * //      http://www.apache.org/licenses/LICENSE-2.0
13  * //
14  * // Unless required by applicable law or agreed to in writing, software
15  * // distributed under the License is distributed on an "AS IS" BASIS,
16  * // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * // See the License for the specific language governing permissions and
18  * // limitations under the License.
19  * //
20  * //-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=
21  */
22 package org.iotivity.cloud.ciserver.resources;
23
24 import java.util.ArrayList;
25 import java.util.Arrays;
26 import java.util.Collections;
27 import java.util.HashMap;
28 import java.util.List;
29 import java.util.Map;
30 import java.util.Timer;
31 import java.util.TimerTask;
32
33 import org.iotivity.cloud.base.device.Device;
34 import org.iotivity.cloud.base.exception.ServerException;
35 import org.iotivity.cloud.base.exception.ServerException.BadRequestException;
36 import org.iotivity.cloud.base.protocols.IRequest;
37 import org.iotivity.cloud.base.protocols.IResponse;
38 import org.iotivity.cloud.base.protocols.MessageBuilder;
39 import org.iotivity.cloud.base.protocols.enums.ContentFormat;
40 import org.iotivity.cloud.base.protocols.enums.ResponseStatus;
41 import org.iotivity.cloud.base.resource.Resource;
42 import org.iotivity.cloud.ciserver.Constants;
43 import org.iotivity.cloud.util.Cbor;
44
45 /**
46  *
47  * This class provides a set of APIs to use KeepAlive Resource for ensuring the
48  * connection.
49  *
50  */
51 public class KeepAliveResource extends Resource {
52     private int[]                          mIntervals      = null;
53     private Timer                          mTimer          = new Timer();
54     private Cbor<HashMap<String, Integer>> mCbor           = new Cbor<>();
55     private HashMap<Device, Long>          mConnectionPool = new HashMap<>();
56
57     public KeepAliveResource(int[] intervals) {
58         super(Arrays.asList(Constants.PREFIX_OIC, Constants.KEEP_ALIVE_URI));
59         mIntervals = intervals;
60     }
61
62     @Override
63     public void onDefaultRequestReceived(Device srcDevice, IRequest request)
64             throws ServerException {
65
66         IResponse response = null;
67
68         switch (request.getMethod()) {
69             case GET:
70                 response = handleGetPingConfig(request);
71                 break;
72
73             case PUT:
74                 response = handlePutPingConfig(srcDevice, request);
75                 break;
76
77             default:
78                 throw new BadRequestException(
79                         request.getMethod() + " request type is not support");
80         }
81
82         srcDevice.sendResponse(response);
83     }
84
85     public void startSessionChecker() {
86         mTimer.schedule(new KeepAliveTask(), 30000, 60000);
87     }
88
89     public void stopSessionChecker() {
90         mTimer.cancel();
91     }
92
93     /**
94      * API for making interval and first response to Resource
95      * 
96      * @param request
97      *            ChannelHandlerContext of request message
98      */
99     private IResponse handleGetPingConfig(IRequest request) {
100
101         HashMap<String, int[]> payloadData = new HashMap<>();
102         payloadData.put("inarray", mIntervals);
103
104         return MessageBuilder.createResponse(request, ResponseStatus.CONTENT,
105                 ContentFormat.APPLICATION_CBOR,
106                 mCbor.encodingPayloadToCbor(payloadData));
107     }
108
109     private IResponse handlePutPingConfig(Device srcDevice, IRequest request) {
110
111         HashMap<String, Integer> payloadData = mCbor
112                 .parsePayloadFromCbor(request.getPayload(), HashMap.class);
113
114         if (payloadData.containsKey("in")) {
115             mConnectionPool.put(srcDevice, System.currentTimeMillis()
116                     + (payloadData.get("in") * (long) 60000));
117         }
118
119         return MessageBuilder.createResponse(request, ResponseStatus.VALID);
120     }
121
122     /**
123      * API for managing session
124      */
125     private class KeepAliveTask extends TimerTask {
126
127         @Override
128         public void run() {
129             Map<Device, Long> map = Collections
130                     .synchronizedMap(mConnectionPool);
131
132             List<Device> deleteList = new ArrayList<>();
133
134             synchronized (map) {
135                 Long currentTime = System.currentTimeMillis();
136                 for (Device device : map.keySet()) {
137                     Long lifeTime = (Long) map.get(device);
138                     if (lifeTime < currentTime) {
139                         deleteList.add(device);
140                     }
141                 }
142             }
143
144             for (Device device : deleteList) {
145                 mConnectionPool.remove(device);
146                 device.getCtx().fireChannelInactive();
147                 device.getCtx().close();
148             }
149         }
150     }
151 }