initial commit
diff --git a/src/main/java/org.onosproject.xran/wrapper/UeMap.java b/src/main/java/org.onosproject.xran/wrapper/UeMap.java
new file mode 100644
index 0000000..a42ecbd
--- /dev/null
+++ b/src/main/java/org.onosproject.xran/wrapper/UeMap.java
@@ -0,0 +1,73 @@
+/*
+ * Copyright 2015-present Open Networking Laboratory
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.onosproject.xran.wrapper;
+
+import com.google.common.collect.BiMap;
+import com.google.common.collect.HashBiMap;
+import io.netty.channel.ChannelHandlerContext;
+import org.onosproject.net.HostId;
+import org.onosproject.xran.XranStore;
+import org.onosproject.xran.codecs.api.CRNTI;
+import org.onosproject.xran.codecs.api.ENBUES1APID;
+import org.onosproject.xran.codecs.api.MMEUES1APID;
+import org.onosproject.xran.entities.RnibUe;
+
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+
+public class UeMap {
+ private BiMap<CRNTI, MMEUES1APID> crntUe = HashBiMap.create();
+
+ private XranStore xranStore;
+
+ public UeMap(XranStore xranStore) {
+ this.xranStore = xranStore;
+ }
+
+ public void put(RnibUe ue) {
+
+ MMEUES1APID mmeS1apId = ue.getMmeS1apId();
+ if (mmeS1apId != null) {
+ xranStore.storeUe(ue);
+ }
+
+ CRNTI ranId = ue.getRanId();
+ if (ranId != null) {
+ crntUe.put(ranId, ue.getMmeS1apId());
+ }
+ }
+
+ public RnibUe get(CRNTI id) {
+ MMEUES1APID mme = crntUe.get(id);
+ if (mme != null) {
+ return xranStore.getUe(mme);
+ }
+ return null;
+ }
+
+ public RnibUe get(MMEUES1APID mme) {
+ if (mme != null) {
+ return xranStore.getUe(mme);
+ }
+ return null;
+ }
+
+ public boolean remove(MMEUES1APID id) {
+ crntUe.inverse().remove(id);
+ return xranStore.removeUe(id);
+ }
+}