add getObjectType - required for linking within SQLDev

PhilippSalvisberg · PhilippSalvisberg · commit 61b9b835fab8 · 2019-06-20T02:07:59.000+02:00
diff --git a/sqldev/src/main/java/org/utplsql/sqldev/dal/UtplsqlDao.xtend b/sqldev/src/main/java/org/utplsql/sqldev/dal/UtplsqlDao.xtend
@@ -897,4 +897,30 @@ class UtplsqlDao {
 		})
 		return ret
 	}
+	
+	/**
+	 * gets the object type of a database object
+	 * 
+	 * The object types "PACKAGE BODY", "TYPE BODY" have higher priority.
+	 * "PACKAGE" OR "TYPE" will be returned only when no body exists.
+	 * 
+	 * @param owner owner of the object (schema)
+	 * @param objectName name of the object
+	 * @return the object type, e.g. PACKAGE BODY, TYPE BODY, PROCEDURE, FUNCTION
+	 */
+	def getObjectType(String owner, String objectName) {
+		val sql = '''
+				SELECT object_type
+				  FROM (
+				          SELECT object_type
+				            FROM «IF dbaViewAccessible»dba«ELSE»all«ENDIF»_objects
+				           WHERE owner = ?
+				             AND object_name = ?
+				           ORDER BY decode(object_type, 'PACKAGE', 10, 'TYPE', 10, 'SYNONYM', 20, 1)
+					   )
+				 WHERE rownum = 1
+			'''
+		val objectType = jdbcTemplate.queryForObject(sql, #[owner, objectName], String)
+		return objectType
+	}
 }
diff --git a/sqldev/src/test/java/org/utplsql/sqldev/test/dal/DalTest.xtend b/sqldev/src/test/java/org/utplsql/sqldev/test/dal/DalTest.xtend
@@ -514,5 +514,21 @@ class DalTest extends AbstractJdbcTest {
 		Assert.assertTrue(actual.contains("PROCEDURE p1 IS"))
 		jdbcTemplate.execute("DROP PACKAGE BODY junit_utplsql_test_pkg")
 	}
+	
+	@Test
+	def void getObjectType() {
+		val dao = new UtplsqlDao(dataSource.connection)
+		jdbcTemplate.execute('''
+			CREATE OR REPLACE PACKAGE junit_utplsql_test_pkg IS
+			   -- %suite
+
+			   -- %test
+			   PROCEDURE p1;
+			END junit_utplsql_test_pkg;
+		''')
+		val actual = dao.getObjectType("SCOTT", "JUNIT_UTPLSQL_TEST_PKG")
+		Assert.assertEquals("PACKAGE", actual)
+		jdbcTemplate.execute("DROP PACKAGE junit_utplsql_test_pkg")
+	}
 
 }
