Fix Spike Event Processing with Common Format
[aai/data-router.git] / src / test / java / org / onap / aai / datarouter / policy / SpikeEntitySearchProcessorStubbed.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 SpikeEntitySearchProcessorStubbed extends SpikeEntitySearchProcessor {
28
29   private InMemorySearchDatastore searchDb;
30
31   public SpikeEntitySearchProcessorStubbed(SpikeEventPolicyConfig config)
32       throws FileNotFoundException {
33     super(config);
34   }
35
36   public SpikeEntitySearchProcessorStubbed withSearchDb(InMemorySearchDatastore searchDb) {
37     this.searchDb = searchDb;
38     return this;
39   }
40
41
42   public InMemorySearchDatastore getSearchDb() {
43     return searchDb;
44   }
45
46   protected void handleSearchServiceOperation(DocumentStoreDataEntity eventEntity, String action,
47       String index) {
48     // Stub out the actual call to Search Data service and instead store/update documents in memory
49     try {
50       switch (action.toLowerCase()) {
51         case "create":
52           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
53                                                                       // variable == c1
54           break;
55         case "update":
56           searchDb.put(eventEntity.getId(), eventEntity.getAsJson()); // they are executed if
57                                                                       // variable == c1
58           break;
59         case "delete":
60           searchDb.remove(eventEntity.getId()); // they are executed if variable == c1
61           break;
62         default:
63           break;
64       }
65     } catch (Exception ex) {
66     }
67   }
68 }