resource-assignment/provider/src/main/java/org/openecomp/sdnc/lock/dao/ResourceLockDaoImpl.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * openECOMP : SDN-C
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights
   6  *                      reserved.
   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
  22 package org.openecomp.sdnc.lock.dao;
  23
  24 import java.sql.ResultSet;
  25 import java.sql.SQLException;
  26 import java.util.Date;
  27 import java.util.List;
  28
  29 import org.openecomp.sdnc.lock.data.ResourceLock;
  30 import org.openecomp.sdnc.util.db.CachedDataSourceWrap;
  31 import org.slf4j.Logger;
  32 import org.slf4j.LoggerFactory;
  33 import org.springframework.jdbc.core.JdbcTemplate;
  34 import org.springframework.jdbc.core.RowMapper;
  35
  36 public class ResourceLockDaoImpl implements ResourceLockDao {
  37
  38     private static final Logger log = LoggerFactory.getLogger(ResourceLockDaoImpl.class);
  39
  40     private JdbcTemplate jdbcTemplate;
  41     private boolean testing = false;
  42
  43     @Override
  44     public void lockTable() {
  45         if (!testing) {
  46             jdbcTemplate.update("LOCK TABLES RESOURCE_LOCK WRITE");
  47         log.info("Table RESOURCE_LOCK locked.");
  48     }
  49     }
  50
  51     @Override
  52     public void unlockTable() {
  53         if (!testing) {
  54             jdbcTemplate.update("UNLOCK TABLES");
  55         log.info("Table RESOURCE_LOCK unlocked.");
  56
  57             CachedDataSourceWrap ds = (CachedDataSourceWrap) jdbcTemplate.getDataSource();
  58             ds.releaseConnection();
  59         }
  60     }
  61
  62     @Override
  63     public void add(ResourceLock l) {
  64         jdbcTemplate.update(
  65                 "INSERT INTO RESOURCE_LOCK (resource_name, lock_holder, lock_count, lock_time, expiration_time)\n" +
  66                         "VALUES (?, ?, ?, ?, ?)",
  67                 new Object[] { l.resourceName, l.lockHolder, l.lockCount, l.lockTime, l.expirationTime });
  68     }
  69
  70     @Override
  71     public void update(long id, Date lockTime, Date expirationTime, int lockCount) {
  72         jdbcTemplate.update(
  73                 "UPDATE RESOURCE_LOCK SET lock_time = ?, expiration_time = ?, lock_count = ? WHERE resource_lock_id = ?",
  74                 new Object[] { lockTime, expirationTime, lockCount, id });
  75     }
  76
  77     @Override
  78     public ResourceLock getByResourceName(String resourceName) {
  79         List<ResourceLock> ll = jdbcTemplate.query("SELECT * FROM RESOURCE_LOCK WHERE resource_name = ?",
  80                 new Object[] { resourceName }, new RowMapper<ResourceLock>() {
  81
  82                     @Override
  83                     public ResourceLock mapRow(ResultSet rs, int rowNum) throws SQLException {
  84                         ResourceLock rl = new ResourceLock();
  85                         rl.id = rs.getLong("resource_lock_id");
  86                         rl.resourceName = rs.getString("resource_name");
  87                         rl.lockHolder = rs.getString("lock_holder");
  88                         rl.lockCount = rs.getInt("lock_count");
  89                         rl.lockTime = rs.getTimestamp("lock_time");
  90                         rl.expirationTime = rs.getTimestamp("expiration_time");
  91                         return rl;
  92                     }
  93                 });
  94         return ll != null && !ll.isEmpty() ? ll.get(0) : null;
  95     }
  96
  97     @Override
  98     public void delete(long id) {
  99         jdbcTemplate.update("DELETE FROM RESOURCE_LOCK WHERE resource_lock_id = ?", new Object[] { id });
 100     }
 101
 102     @Override
 103     public void decrementLockCount(long id) {
 104         jdbcTemplate.update("UPDATE RESOURCE_LOCK SET lock_count = lock_count - 1 WHERE resource_lock_id = ?",
 105                 new Object[] { id });
 106     }
 107
 108     public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
 109         this.jdbcTemplate = jdbcTemplate;
 110     }
 111
 112     public void setTesting(boolean testing) {
 113         this.testing = testing;
 114     }
 115 }