Fix memory leak in BA mount service
[controller.git] / opendaylight / md-sal / sal-binding-broker / src / main / java / org / opendaylight / controller / md / sal / binding / impl / BindingDOMMountPointServiceAdapter.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.controller.md.sal.binding.impl;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.base.Optional;
12 import com.google.common.cache.CacheBuilder;
13 import com.google.common.cache.CacheLoader;
14 import com.google.common.cache.LoadingCache;
15 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
16 import org.opendaylight.controller.md.sal.binding.api.MountPointService;
17 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
18 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
19 import org.opendaylight.yangtools.concepts.ListenerRegistration;
20 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
22 import org.slf4j.Logger;
23 import org.slf4j.LoggerFactory;
24
25 public class BindingDOMMountPointServiceAdapter implements MountPointService {
26     private static final Logger LOG = LoggerFactory.getLogger(BindingDOMMountPointServiceAdapter.class);
27
28     private final BindingToNormalizedNodeCodec codec;
29     private final DOMMountPointService mountService;
30     @VisibleForTesting
31     final LoadingCache<DOMMountPoint, BindingMountPointAdapter> bindingMountpoints = CacheBuilder.newBuilder()
32             .weakKeys().weakValues().build(new CacheLoader<DOMMountPoint, BindingMountPointAdapter>() {
33                 @Override
34                 public BindingMountPointAdapter load(final DOMMountPoint key) {
35                     return new BindingMountPointAdapter(codec, key);
36                 }
37             });
38
39     public BindingDOMMountPointServiceAdapter(final DOMMountPointService mountService,
40             final BindingToNormalizedNodeCodec codec) {
41         this.codec = codec;
42         this.mountService = mountService;
43     }
44
45     @Override
46     public Optional<MountPoint> getMountPoint(final InstanceIdentifier<?> mountPoint) {
47
48         YangInstanceIdentifier domPath = codec.toYangInstanceIdentifierBlocking(mountPoint);
49         Optional<DOMMountPoint> domMount = mountService.getMountPoint(domPath);
50         if (domMount.isPresent()) {
51             return Optional.<MountPoint>fromNullable(bindingMountpoints.getUnchecked(domMount.get()));
52         }
53         return Optional.absent();
54     }
55
56     @Override
57     public <T extends MountPointListener> ListenerRegistration<T> registerListener(final InstanceIdentifier<?> path,
58             final T listener) {
59         return new BindingDOMMountPointListenerAdapter<>(listener, codec, mountService);
60     }
61 }