Fix Spike Event Processing with Common Format
[aai/data-router.git] / src / test / java / org / onap / aai / datarouter / policy / SpikeAutosuggestProcessorStubbed.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 Amdocs
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  * ============LICENSE_END=========================================================
20  */
21 package org.onap.aai.datarouter.policy;
22
23 import java.io.FileNotFoundException;
24
25 import org.onap.aai.datarouter.entity.DocumentStoreDataEntity;
26
27 public class SpikeAutosuggestProcessorStubbed extends SpikeAutosuggestIndexProcessor {
28
29   private InMemorySearchDatastore searchDb;
30
31   public SpikeAutosuggestProcessorStubbed(SpikeEventPolicyConfig config)
32       throws FileNotFoundException {
33     super(config);
34   }
35
36   public InMemorySearchDatastore getSearchDb() {
37     return searchDb;
38   }
39
40   public SpikeAutosuggestProcessorStubbed withSearchDb(InMemorySearchDatastore searchDb) {
41     this.searchDb = searchDb;
42     return this;
43   }
44
45   protected void handleSearchServiceOperation(DocumentStoreDataEntity eventEntity, String action,
46       String index) {
47     // Stub out the actual call to Search Data service and instead store/update documents in memory
48     try {
49       switch (action.toLowerCase()) {
50         case "create":
51           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
52                                                                       // variable == c1
53           break;
54         case "update":
55           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
56                                                                       // variable == c1
57           break;
58         case "delete":
59           searchDb.remove(eventEntity.getId()); // they are executed if variable == c1
60           break;
61         default:
62           break;
63       }
64     } catch (Exception ex) {
65     }
66   }
67 }