Add graceful shutdown for Session Manager
[cps.git] / cps-ri / src / main / java / org / onap / cps / spi / config / CpsSessionFactory.java
1 /*
2  *  ============LICENSE_START=======================================================
3  *  Copyright (C) 2022 Nordix Foundation
4  *  ================================================================================
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *        http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an "AS IS" BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  *  SPDX-License-Identifier: Apache-2.0
18  *  ============LICENSE_END=========================================================
19  */
20
21 package org.onap.cps.spi.config;
22
23 import org.hibernate.HibernateException;
24 import org.hibernate.Session;
25 import org.hibernate.SessionFactory;
26 import org.onap.cps.spi.entities.AnchorEntity;
27 import org.onap.cps.spi.entities.DataspaceEntity;
28 import org.onap.cps.spi.entities.SchemaSetEntity;
29 import org.onap.cps.spi.entities.YangResourceEntity;
30 import org.springframework.beans.factory.config.ConfigurableBeanFactory;
31 import org.springframework.context.annotation.Scope;
32 import org.springframework.stereotype.Component;
33
34 @Component
35 @Scope(ConfigurableBeanFactory.SCOPE_SINGLETON)
36 public class CpsSessionFactory {
37
38     private SessionFactory sessionFactory = null;
39
40     /**
41      * Open a session from session factory.
42      *
43      * @return session
44      * @throws HibernateException hibernate exception
45      */
46     public Session openSession() throws HibernateException {
47         return getSessionFactory().openSession();
48     }
49
50     /**
51      * Close session factory.
52      *
53      * @throws HibernateException hibernate exception
54      */
55     public void closeSessionFactory() throws HibernateException {
56         getSessionFactory().close();
57     }
58
59     private SessionFactory getSessionFactory() {
60         if (sessionFactory == null) {
61             sessionFactory = new org.hibernate.cfg.Configuration().configure("hibernate.cfg.xml")
62                     .addAnnotatedClass(AnchorEntity.class)
63                     .addAnnotatedClass(DataspaceEntity.class)
64                     .addAnnotatedClass(SchemaSetEntity.class)
65                     .addAnnotatedClass(YangResourceEntity.class)
66                     .buildSessionFactory();
67         }
68         return sessionFactory;
69     }
70 }