7f0ccc5fe2729cd5b0ebdc4ec9cc310201adf7c8
[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.Iterator;
29 import java.util.Map;
30 import java.util.Set;
31 import java.util.Timer;
32 import java.util.TimerTask;
33
34 import org.iotivity.cloud.base.device.Device;
35 import org.iotivity.cloud.base.exception.ServerException;
36 import org.iotivity.cloud.base.exception.ServerException.BadRequestException;
37 import org.iotivity.cloud.base.protocols.IRequest;
38 import org.iotivity.cloud.base.protocols.IResponse;
39 import org.iotivity.cloud.base.protocols.MessageBuilder;
40 import org.iotivity.cloud.base.protocols.enums.ContentFormat;
41 import org.iotivity.cloud.base.protocols.enums.ResponseStatus;
42 import org.iotivity.cloud.base.resource.Resource;
43 import org.iotivity.cloud.ciserver.Constants;
44 import org.iotivity.cloud.util.Cbor;
45
46 /**
47  *
48  * This class provides a set of APIs to use KeepAlive Resource for ensuring the
49  * connection.
50  *
51  */
52 public class KeepAliveResource extends Resource {
53     private int[]                          mIntervals      = null;
54     private Timer                          mTimer          = new Timer();
55     private Cbor<HashMap<String, Integer>> mCbor           = new Cbor<>();
56     private HashMap<Device, Long>          mConnectionPool = new HashMap<>();
57
58     public KeepAliveResource(int[] intervals) {
59         super(Arrays.asList(Constants.PREFIX_OIC, Constants.KEEP_ALIVE_URI));
60         mIntervals = intervals;
61     }
62
63     @Override
64     public void onDefaultRequestReceived(Device srcDevice, IRequest request)
65             throws ServerException {
66
67         IResponse response = null;
68
69         switch (request.getMethod()) {
70             case GET:
71                 response = handleGetPingConfig(request);
72                 mConnectionPool.put(srcDevice, System.currentTimeMillis()
73                         + (mIntervals[0] * (long) 60000));
74                 break;
75
76             case PUT:
77                 response = handlePutPingConfig(srcDevice, request);
78                 break;
79
80             default:
81                 throw new BadRequestException(
82                         request.getMethod() + " request type is not support");
83         }
84
85         srcDevice.sendResponse(response);
86     }
87
88     public void startSessionChecker() {
89         mTimer.schedule(new KeepAliveTask(), 30000, 60000);
90     }
91
92     public void stopSessionChecker() {
93         mTimer.cancel();
94     }
95
96     /**
97      * API for making interval and first response to Resource
98      * 
99      * @param request
100      *            ChannelHandlerContext of request message
101      */
102     private IResponse handleGetPingConfig(IRequest request) {
103
104         HashMap<String, int[]> payloadData = new HashMap<>();
105         payloadData.put("inarray", mIntervals);
106
107         return MessageBuilder.createResponse(request, ResponseStatus.CONTENT,
108                 ContentFormat.APPLICATION_CBOR,
109                 mCbor.encodingPayloadToCbor(payloadData));
110     }
111
112     private IResponse handlePutPingConfig(Device srcDevice, IRequest request) {
113
114         HashMap<String, Integer> payloadData = mCbor
115                 .parsePayloadFromCbor(request.getPayload(), HashMap.class);
116         if (payloadData != null) {
117             if (payloadData.containsKey("in")) {
118                 mConnectionPool.put(srcDevice, System.currentTimeMillis()
119                         + (payloadData.get("in") * (long) 60000));
120             }
121         }
122         return MessageBuilder.createResponse(request, ResponseStatus.VALID);
123     }
124
125     /**
126      * API for managing session
127      */
128     private class KeepAliveTask extends TimerTask {
129
130         @Override
131         public void run() {
132             Map<Device, Long> map = Collections
133                     .synchronizedMap(mConnectionPool);
134             Set<Device> keySet = map.keySet();
135             ArrayList<Device> deleteList = new ArrayList<>();
136             Iterator<Device> iterator = null;
137             synchronized (map) {
138                 iterator = keySet.iterator();
139                 Long currentTime = System.currentTimeMillis();
140                 // check interval
141                 while (iterator.hasNext()) {
142                     Device key = iterator.next();
143                     if (map.containsKey(key)) {
144                         if (map.get(key) != null) {
145                             Long lifeTime = (Long) map.get(key);
146                             if (lifeTime != null) {
147                                 if (lifeTime < currentTime) {
148                                     deleteList.add(key);
149                                 }
150                             }
151                         }
152                     }
153                 }
154
155             }
156             iterator = deleteList.iterator();
157             // remove session
158             while (iterator.hasNext()) {
159                 Device key = iterator.next();
160                 mConnectionPool.remove(key);
161                 key.getCtx().fireChannelInactive();
162                 key.getCtx().close();
163             }
164         }
165     }
166 }