cloud/resourcedirectory/src/main/java/org/iotivity/cloud/rdserver/resources/directory/res/DiscoveryResource.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.rdserver.resources.directory.res;
  23
  24 import java.util.ArrayList;
  25 import java.util.Arrays;
  26 import java.util.HashMap;
  27 import java.util.List;
  28
  29 import org.iotivity.cloud.base.device.Device;
  30 import org.iotivity.cloud.base.exception.ServerException;
  31 import org.iotivity.cloud.base.exception.ServerException.BadRequestException;
  32 import org.iotivity.cloud.base.exception.ServerException.PreconditionFailedException;
  33 import org.iotivity.cloud.base.protocols.IRequest;
  34 import org.iotivity.cloud.base.protocols.IResponse;
  35 import org.iotivity.cloud.base.protocols.MessageBuilder;
  36 import org.iotivity.cloud.base.protocols.enums.ContentFormat;
  37 import org.iotivity.cloud.base.protocols.enums.ResponseStatus;
  38 import org.iotivity.cloud.base.resource.Resource;
  39 import org.iotivity.cloud.rdserver.Constants;
  40 import org.iotivity.cloud.rdserver.db.DBManager;
  41 import org.iotivity.cloud.rdserver.util.TypeCastingManager;
  42 import org.iotivity.cloud.util.Cbor;
  43 import org.iotivity.cloud.util.Log;
  44
  45 public class DiscoveryResource extends Resource {
  46     private Cbor<HashMap<Object, Object>>      mCbor                     = new Cbor<>();
  47     private TypeCastingManager<DiscoveryTags>  mDiscoveryTagsTypeManager = new TypeCastingManager<>();
  48     private TypeCastingManager<DiscoveryLinks> mDiscoveryLinkTypeManager = new TypeCastingManager<>();
  49
  50     public DiscoveryResource() {
  51         super(Arrays.asList(Constants.PREFIX_OIC, Constants.WELL_KNOWN_URI));
  52     }
  53
  54     @Override
  55     public void onDefaultRequestReceived(Device srcDevice, IRequest request)
  56             throws ServerException {
  57
  58         IResponse response = null;
  59
  60         switch (request.getMethod()) {
  61             case GET:
  62                 response = handleGetRequest(request);
  63                 break;
  64
  65             default:
  66                 throw new BadRequestException(
  67                         request.getMethod() + " request type is not supported");
  68         }
  69
  70         srcDevice.sendResponse(response);
  71     }
  72
  73     private IResponse handleGetRequest(IRequest request)
  74             throws ServerException {
  75
  76         HashMap<String, List<String>> queryMap = request.getUriQueryMap();
  77
  78         ArrayList<DiscoveryPayload> resourceList = new ArrayList<DiscoveryPayload>();
  79
  80         if (queryMap == null) {
  81             throw new PreconditionFailedException("query is null");
  82         }
  83
  84         List<String> deviceList = queryMap.get(Constants.DEVICE_ID);
  85
  86         if (deviceList == null) {
  87             throw new PreconditionFailedException(
  88                     "di property is not included");
  89         }
  90
  91         List<String> listRT = queryMap.get(Constants.RESOURCE_TYPE);
  92         List<String> listITF = queryMap.get(Constants.INTERFACE);
  93         String key = null, value = null;
  94         ArrayList<HashMap<Object, Object>> foundResList = null;
  95
  96         // TODO: Multiple RT or ITF support required
  97         if (listRT != null) {
  98             key = Constants.RESOURCE_TYPE;
  99             value = listRT.get(0);
 100         } else if (listITF != null) {
 101             key = Constants.INTERFACE;
 102             value = listITF.get(0);
 103         }
 104
 105         for (String deviceId : deviceList) {
 106             if(key != null && value != null){
 107                 foundResList = DBManager.getInstance().findResourceAboutDiAndFilter(deviceId,
 108                         key, value);
 109             } else {
 110                 foundResList = DBManager.getInstance().findResourceAboutDi(deviceId);
 111             }
 112
 113             if (foundResList != null) {
 114                 resourceList.add(makeDiscoveryPayloadSegment(foundResList));
 115             }
 116         }
 117
 118         return MessageBuilder.createResponse(request, ResponseStatus.CONTENT,
 119                 ContentFormat.APPLICATION_CBOR,
 120                 createDiscoveryResponse(resourceList));
 121     }
 122
 123     private DiscoveryPayload makeDiscoveryPayloadSegment(
 124             ArrayList<HashMap<Object, Object>> foundResList) {
 125
 126         ArrayList<DiscoveryLinks> discoveryLinksList = new ArrayList<DiscoveryLinks>();
 127
 128         for (HashMap<Object, Object> res : foundResList) {
 129             DiscoveryLinks discoveryLinksPayload = new DiscoveryLinks();
 130             discoveryLinksPayload = mDiscoveryLinkTypeManager
 131                     .convertMaptoObject(res, discoveryLinksPayload);
 132             discoveryLinksList.add(discoveryLinksPayload);
 133         }
 134
 135         DiscoveryPayload discoveryPayload = new DiscoveryPayload();
 136
 137         DiscoveryTags tagsPayload = new DiscoveryTags();
 138
 139         tagsPayload = mDiscoveryTagsTypeManager
 140                 .convertMaptoObject(foundResList.get(0), tagsPayload);
 141
 142         discoveryPayload.setTags(tagsPayload);
 143         discoveryPayload.setLinks(discoveryLinksList);
 144
 145         return discoveryPayload;
 146     }
 147
 148     private byte[] createDiscoveryResponse(
 149             ArrayList<DiscoveryPayload> discoveryPayloadList) {
 150         ArrayList<HashMap<Object, Object>> responseMapList = new ArrayList<HashMap<Object, Object>>();
 151
 152         for (DiscoveryPayload discoveryPayload : discoveryPayloadList) {
 153             HashMap<Object, Object> responseSegment = null;
 154             DiscoveryTags tags = discoveryPayload.getTags();
 155             responseSegment = mDiscoveryTagsTypeManager
 156                     .convertObjectToMap(tags);
 157
 158             ArrayList<DiscoveryLinks> discoveryLinksList = discoveryPayload
 159                     .getLinks();
 160
 161             ArrayList<HashMap<Object, Object>> links = new ArrayList<HashMap<Object, Object>>();
 162
 163             for (DiscoveryLinks discoveryLinks : discoveryLinksList) {
 164                 links.add(mDiscoveryLinkTypeManager
 165                         .convertObjectToMap(discoveryLinks));
 166             }
 167             responseSegment.put(Constants.LINKS, links);
 168
 169             responseMapList.add(responseSegment);
 170         }
 171
 172         Log.i("discover payload :" + responseMapList.toString());
 173
 174         byte[] encodedPaylod = mCbor.encodingPayloadToCbor(responseMapList);
 175
 176         return encodedPaylod;
 177     }
 178 }