2 * Copyright 2017 ZTE Corporation.
\r
4 * Licensed under the Apache License, Version 2.0 (the "License");
\r
5 * you may not use this file except in compliance with the License.
\r
6 * You may obtain a copy of the License at
\r
8 * http://www.apache.org/licenses/LICENSE-2.0
\r
10 * Unless required by applicable law or agreed to in writing, software
\r
11 * distributed under the License is distributed on an "AS IS" BASIS,
\r
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
13 * See the License for the specific language governing permissions and
\r
14 * limitations under the License.
\r
16 package org.openo.holmes.common.utils;
\r
18 import io.dropwizard.db.DataSourceFactory;
\r
19 import io.dropwizard.jdbi.DBIFactory;
\r
20 import io.dropwizard.setup.Environment;
\r
21 import javax.annotation.PostConstruct;
\r
22 import javax.inject.Inject;
\r
23 import javax.inject.Singleton;
\r
24 import lombok.extern.slf4j.Slf4j;
\r
25 import org.jvnet.hk2.annotations.Service;
\r
26 import org.skife.jdbi.v2.DBI;
\r
27 import org.skife.jdbi.v2.Handle;
\r
32 public class DbDaoUtil {
\r
36 private Environment environmentProvider;
\r
38 private DataSourceFactory dataSourceFactoryProvider;
\r
40 private DBIFactory factory = new DBIFactory();
\r
43 public void init() {
\r
45 synchronized (DbDaoUtil.class) {
\r
47 jdbi = factory.build(environmentProvider, dataSourceFactoryProvider, "mysql");
\r
53 public <K> K getDao(Class<K> clazz) {
\r
55 return jdbi.open(clazz);
\r
56 } catch (Exception e) {
\r
57 log.warn("get object instance of Dao error.", e);
\r
62 public Handle getHandle() {
\r
65 } catch (Exception e) {
\r
66 log.warn("get object instance of Dao error.", e);
\r
71 public void close(Object obj) {
\r
75 } catch (Exception e) {
\r
76 log.warn("close jdbi connection error.", e);
\r
81 public <T> T getJdbiDaoByOnDemand(Class<T> daoClazz) {
\r
83 return jdbi.onDemand(daoClazz);
\r
87 public <T> T getJdbiDaoByOpen(Class<T> daoClazz) {
\r
89 return jdbi.open(daoClazz);
\r