1 files changed, 531 insertions, 0 deletions
diff --git a/sli/common/src/main/java/org/openecomp/sdnc/sli/SvcLogicDblibStore.java b/sli/common/src/main/java/org/openecomp/sdnc/sli/SvcLogicDblibStore.java
new file mode 100644
index 0000000..5fca466
--- /dev/null
+++ b/sli/common/src/main/java/org/openecomp/sdnc/sli/SvcLogicDblibStore.java
@@ -0,0 +1,531 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * openECOMP : SDN-C
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights
+ *                         reserved.
+ * ================================================================================
+ * 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.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.sdnc.sli;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+import java.sql.Blob;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Properties;
+
+import javax.sql.rowset.CachedRowSet;
+
+import org.openecomp.sdnc.sli.resource.dblib.DBResourceManager;
+import org.openecomp.sdnc.sli.resource.dblib.DbLibService;
+import org.osgi.framework.Bundle;
+import org.osgi.framework.BundleContext;
+import org.osgi.framework.FrameworkUtil;
+import org.osgi.framework.ServiceReference;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class SvcLogicDblibStore implements SvcLogicStore {
+
+	private static final String SDNC_CONFIG_DIR = "SDNC_CONFIG_DIR";
+	
+	private static final Logger LOG = LoggerFactory
+			.getLogger(SvcLogicDblibStore.class);
+
+	private static final String DBLIB_SERVICE =
+	// "org.openecomp.sdnc.sli.resource.dblib.DBLibService";
+	"org.openecomp.sdnc.sli.resource.dblib.DBResourceManager";
+	
+	Properties props = null;
+
+	public void init(Properties props) throws ConfigurationException {
+
+		DbLibService dbSvc = getDbLibService();
+		if(dbSvc == null) {
+			LOG.error("SvcLogic cannot acquire DBLIB_SERVICE");
+			return;
+		}
+		try {
+			dbSvc.getData("select 1 from DUAL", new ArrayList<String>(), null);
+			LOG.debug("SQL test was successful");
+		} catch (SQLException e) {
+			LOG.error("Failed SQL test", e);
+		}
+	}
+
+	public boolean hasGraph(String module, String rpc, String version,
+			String mode) throws SvcLogicException {
+
+		DbLibService dbSvc = getDbLibService();
+
+		boolean retval = false;
+		CachedRowSet results = null;
+		String hasVersionGraphSql = "SELECT count(*) FROM SVC_LOGIC"
+				+ " WHERE module = ? AND rpc = ? AND mode = ? AND version = ?";
+
+		String hasActiveGraphSql = "SELECT count(*) FROM SVC_LOGIC"
+				+ " WHERE module = ? AND rpc = ? AND mode = ? AND active = 'Y'";
+
+		PreparedStatement hasGraphStmt = null;
+
+		ArrayList<String> args = new ArrayList<String>();
+		args.add(module);
+		args.add(rpc);
+		args.add(mode);
+
+		try {
+
+			if (version == null) {
+				results = dbSvc.getData(hasActiveGraphSql, args, null);
+			} else {
+				args.add(version);
+				results = dbSvc.getData(hasVersionGraphSql, args, null);
+			}
+
+			if (results.next()) {
+				int cnt = results.getInt(1);
+
+				if (cnt > 0) {
+					retval = true;
+				}
+
+			}
+		} catch (Exception e) {
+			throw new ConfigurationException("SQL query failed", e);
+		} finally {
+			if (results != null) {
+				try {
+
+					results.close();
+				} catch (SQLException x) {
+				}
+			}
+
+		}
+
+		return (retval);
+
+	}
+
+	public SvcLogicGraph fetch(String module, String rpc, String version,
+			String mode) throws SvcLogicException {
+
+		DbLibService dbSvc = getDbLibService();
+
+		Connection dbConn = null;
+		SvcLogicGraph retval = null;
+		ResultSet results = null;
+
+		String fetchVersionGraphSql = "SELECT graph FROM SVC_LOGIC"
+				+ " WHERE module = ? AND rpc = ? AND mode = ? AND version = ?";
+
+		String fetchActiveGraphSql = "SELECT graph FROM SVC_LOGIC"
+				+ " WHERE module = ? AND rpc = ? AND mode = ? AND active = 'Y'";
+
+
+		try {
+			dbConn = ((DBResourceManager) dbSvc).getConnection();
+			PreparedStatement fetchGraphStmt;
+			
+			ArrayList<String> args = new ArrayList<String>();
+			args.add(module);
+			args.add(rpc);
+			args.add(mode);
+
+			if (version == null) {
+				fetchGraphStmt = dbConn.prepareStatement(fetchActiveGraphSql);
+			} else {
+				fetchGraphStmt = dbConn.prepareStatement(fetchVersionGraphSql);
+			}
+			
+			fetchGraphStmt.setString(1, module);
+			fetchGraphStmt.setString(2,  rpc);
+			fetchGraphStmt.setString(3, mode);
+			if (version != null) {
+				fetchGraphStmt.setString(4,version);
+			}
+
+			results = fetchGraphStmt.executeQuery();
+			
+			if (results.next()) {
+				Blob graphBlob = results.getBlob("graph");
+
+				ObjectInputStream gStream = new ObjectInputStream(
+						graphBlob.getBinaryStream());
+
+				Object graphObj = gStream.readObject();
+				gStream.close();
+
+				if (graphObj instanceof SvcLogicGraph) {
+					retval = (SvcLogicGraph) graphObj;
+				} else {
+					throw new ConfigurationException("invalid type for graph ("
+							+ graphObj.getClass().getName());
+
+				}
+
+			} else {
+				return (null);
+			}
+		} catch (Exception e) {
+			throw new ConfigurationException("SQL query failed", e);
+		} finally {
+			if (results != null) {
+				try {
+					results.close();
+				} catch (SQLException x) {
+				}
+			}
+			try {
+				if (dbConn != null && !dbConn.isClosed()) {
+					dbConn.close();
+				}
+			} catch (Throwable exc) {
+				// the exception not monitored
+			} finally {
+				dbConn = null;
+			}
+
+		}
+
+		return (retval);
+
+	}
+
+	public void store(SvcLogicGraph graph) throws SvcLogicException {
+
+		DbLibService dbSvc = getDbLibService();
+
+		String storeGraphSql = "INSERT INTO SVC_LOGIC (module, rpc, version, mode, active, graph)"
+				+ " VALUES(?, ?, ?, ?, ?, ?)";
+
+		if (graph == null) {
+			throw new SvcLogicException("graph cannot be null");
+		}
+
+		byte[] graphBytes = null;
+
+		ByteArrayOutputStream byteStr = null;
+		ObjectOutputStream goutStr = null;
+
+		try {
+			byteStr = new ByteArrayOutputStream();
+			goutStr = new ObjectOutputStream(byteStr);
+			goutStr.writeObject(graph);
+
+			graphBytes = byteStr.toByteArray();
+
+		} catch (Exception e) {
+			throw new SvcLogicException("could not serialize graph", e);
+		} finally {
+
+			if (goutStr != null) {
+				try {
+					goutStr.close();
+				} catch (IOException e) {
+
+				}
+			}
+
+			if (byteStr != null) {
+				try {
+					byteStr.close();
+				} catch (IOException e) {
+
+				}
+			}
+		}
+
+		// If object already stored in database, delete it
+		if (hasGraph(graph.getModule(), graph.getRpc(), graph.getVersion(),
+				graph.getMode())) {
+			delete(graph.getModule(), graph.getRpc(), graph.getVersion(),
+					graph.getMode());
+		}
+
+		Connection dbConn = null;
+
+		try {
+			dbConn = ((DBResourceManager) dbSvc).getConnection();
+			boolean oldAutoCommit = dbConn.getAutoCommit();
+			dbConn.setAutoCommit(false);
+			PreparedStatement storeGraphStmt = dbConn
+					.prepareStatement(storeGraphSql);
+			storeGraphStmt.setString(1, graph.getModule());
+			storeGraphStmt.setString(2, graph.getRpc());
+			storeGraphStmt.setString(3, graph.getVersion());
+			storeGraphStmt.setString(4, graph.getMode());
+			storeGraphStmt.setString(5, "N");
+			storeGraphStmt.setBlob(6, new ByteArrayInputStream(graphBytes));
+
+			storeGraphStmt.executeUpdate();
+			dbConn.commit();
+
+			dbConn.setAutoCommit(oldAutoCommit);
+		} catch (Exception e) {
+			throw new SvcLogicException("Could not write object to database", e);
+		} finally {
+			try {
+				if (dbConn != null && !dbConn.isClosed()) {
+					dbConn.close();
+				}
+			} catch (Throwable exc) {
+				// the exception not monitored
+			} finally {
+				dbConn = null;
+			}
+
+		}
+	}
+
+	public void delete(String module, String rpc, String version, String mode)
+			throws SvcLogicException {
+
+		DbLibService dbSvc = getDbLibService();
+
+		String deleteGraphSql = "DELETE FROM SVC_LOGIC WHERE module = ? AND rpc = ? AND version = ? AND mode = ?";
+
+		ArrayList<String> args = new ArrayList<String>();
+
+		args.add(module);
+		args.add(rpc);
+		args.add(version);
+		args.add(mode);
+
+		try {
+			dbSvc.writeData(deleteGraphSql, args, null);
+		} catch (Exception e) {
+			throw new SvcLogicException(
+					"Could not delete object from database", e);
+		}
+	}
+
+	public void activate(SvcLogicGraph graph) throws SvcLogicException {
+		DbLibService dbSvc = getDbLibService();
+
+		String deactivateSql = "UPDATE SVC_LOGIC SET active = 'N' WHERE module = ? AND rpc = ? AND mode = ?";
+
+		String activateSql = "UPDATE SVC_LOGIC SET active = 'Y' WHERE module = ? AND rpc = ? AND mode = ? AND version = ?";
+
+		ArrayList<String> args = new ArrayList<String>();
+
+		args.add(graph.getModule());
+		args.add(graph.getRpc());
+		args.add(graph.getMode());
+
+		try {
+
+			dbSvc.writeData(deactivateSql, args, null);
+
+			args.add(graph.getVersion());
+			dbSvc.writeData(activateSql, args, null);
+
+		} catch (Exception e) {
+			throw new SvcLogicException("Could not activate graph", e);
+		}
+	}
+
+	@Override
+	public void registerNodeType(String nodeType) throws SvcLogicException {
+
+		String registerNodeSql = "INSERT INTO NODE_TYPES (nodetype) VALUES(?)";
+
+		if (isValidNodeType(nodeType)) {
+			return;
+		}
+
+		DbLibService dbSvc = getDbLibService();
+		ArrayList<String> args = new ArrayList<String>();
+
+		args.add(nodeType);
+
+		try {
+			dbSvc.writeData(registerNodeSql, args, null);
+		} catch (Exception e) {
+			throw new SvcLogicException("Could not add node type to database",
+					e);
+		}
+
+	}
+
+	@Override
+	public void unregisterNodeType(String nodeType) throws SvcLogicException {
+
+		if (!isValidNodeType(nodeType)) {
+			return;
+		}
+
+		String unregisterNodeSql = "DELETE FROM NODE_TYPES WHERE nodetype = ?";
+
+		DbLibService dbSvc = getDbLibService();
+		ArrayList<String> args = new ArrayList<String>();
+
+		args.add(nodeType);
+
+		try {
+			dbSvc.writeData(unregisterNodeSql, args, null);
+		} catch (Exception e) {
+			throw new SvcLogicException(
+					"Could not delete node type from database", e);
+		}
+
+	}
+
+	@Override
+	public boolean isValidNodeType(String nodeType) throws SvcLogicException {
+
+		String validateNodeSql = "SELECT count(*) FROM NODE_TYPES WHERE nodetype = ?";
+
+		DbLibService dbSvc = getDbLibService();
+
+		ArrayList<String> args = new ArrayList<String>();
+
+		args.add(nodeType);
+
+		boolean isValid = false;
+
+		CachedRowSet results = null;
+		try {
+			results = dbSvc.getData(validateNodeSql, args, null);
+			if (results != null) {
+				if (results.next()) {
+					int cnt = results.getInt(1);
+
+					if (cnt > 0) {
+						isValid = true;
+					}
+				}
+			}
+		} catch (Exception e) {
+			throw new SvcLogicException(
+					"Cannot select node type from database", e);
+		} finally {
+			if (results != null) {
+				try {
+					results.close();
+				} catch (SQLException x) {
+				}
+			}
+
+		}
+
+		return (isValid);
+	}
+
+	private DbLibService getDbLibService() {
+
+		// Get DbLibService interface object.
+		DbLibService dblibSvc = null;
+		ServiceReference sref = null;
+		BundleContext bctx = null;
+	
+		Bundle bundle = FrameworkUtil.getBundle(SvcLogicDblibStore.class);
+		
+		if (bundle != null) {
+			bctx = bundle.getBundleContext();
+			
+			if (bctx != null) {
+				sref = bctx.getServiceReference(DBLIB_SERVICE);
+			}
+
+			if (sref == null) {
+				LOG.warn("Could not find service reference for DBLIB service ("
+						+ DBLIB_SERVICE + ")");
+			} else {
+				dblibSvc = (DbLibService) bctx.getService(sref);
+				if (dblibSvc == null) {
+					
+					LOG.warn("Could not find service reference for DBLIB service ("
+							+ DBLIB_SERVICE + ")");
+				}
+			}
+		}
+
+		// initialize a stand-alone instance of dblib resource
+		else {
+			// Try to create a DbLibService object from dblib properties
+			if(JavaSingleton.getInstance() == null){
+				Properties dblibProps = new Properties();
+				
+				String propDir = System.getenv(SDNC_CONFIG_DIR);
+				if (propDir == null) {
+					
+					propDir = "/opt/sdnc/data/properties";
+				}
+				String propPath = propDir + "/dblib.properties";
+				
+				File propFile = new File(propPath);
+				
+				if (!propFile.exists()) {
+					
+					LOG.warn(
+							"Missing configuration properties file : "
+									+ propFile);
+					return(null);
+				}
+				
+				try {
+					
+					dblibProps.load(new FileInputStream(propFile));
+				} catch (Exception e) {
+					LOG.warn(
+							"Could not load properties file " + propPath, e);
+					return(null);
+					
+				}
+				
+				try {
+					dblibSvc = DBResourceManager.create(dblibProps);
+					JavaSingleton.setInstance(dblibSvc);
+				} catch (Exception e) {
+					LOG.warn("Caught exception trying to create DBResourceManager", e);
+				}
+			} else {
+				dblibSvc = JavaSingleton.getInstance();
+			}
+		}
+		return (dblibSvc);
+	}
+
+	
+	static class JavaSingleton {
+	     /* Private constructor     */
+	     private JavaSingleton() {
+	        /* the body of the constructor here */
+	     }
+
+	     /* instance of the singleton declaration */
+	     private static volatile  DbLibService INSTANCE ;
+
+	     /* Access point to the unique instance of the singleton */
+	     public static DbLibService getInstance() {
+	        return INSTANCE;
+	     }
+	     
+	     public static void setInstance(DbLibService dbresource) {
+		        INSTANCE = dbresource;
+		     }
+	}
+}