-
Notifications
You must be signed in to change notification settings - Fork 47
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
ec7850f
commit 26d7ae7
Showing
20 changed files
with
1,614 additions
and
29 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
230 changes: 230 additions & 0 deletions
230
...tor-commons/src/main/java/io/ballerina/modelgenerator/commons/ServiceDatabaseManager.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,230 @@ | ||
| /* | ||
| * Copyright (c) 2024, WSO2 LLC. (http://www.wso2.com) | ||
| * | ||
| * WSO2 LLC. licenses this file to you 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. | ||
| */ | ||
|
|
||
| package io.ballerina.modelgenerator.commons; | ||
|
|
||
| import java.io.IOException; | ||
| import java.net.URL; | ||
| import java.nio.file.Files; | ||
| import java.nio.file.Path; | ||
| import java.sql.Connection; | ||
| import java.sql.DriverManager; | ||
| import java.sql.PreparedStatement; | ||
| import java.sql.ResultSet; | ||
| import java.sql.SQLException; | ||
| import java.util.ArrayList; | ||
| import java.util.LinkedHashMap; | ||
| import java.util.List; | ||
| import java.util.Optional; | ||
| import java.util.logging.Logger; | ||
|
|
||
| /** | ||
| * Manages database operations for retrieving information about external connectors and functions. | ||
| * | ||
| * @since 2.0.0 | ||
| */ | ||
| public class ServiceDatabaseManager { | ||
|
|
||
| private static final String INDEX_FILE_NAME = "service-index.sqlite"; | ||
| private static final Logger LOGGER = Logger.getLogger(ServiceDatabaseManager.class.getName()); | ||
| private final String dbPath; | ||
|
|
||
| private static class Holder { | ||
| private static final ServiceDatabaseManager INSTANCE = new ServiceDatabaseManager(); | ||
| } | ||
|
|
||
| public static ServiceDatabaseManager getInstance() { | ||
| return Holder.INSTANCE; | ||
| } | ||
|
|
||
| private ServiceDatabaseManager() { | ||
| try { | ||
| Class.forName("org.sqlite.JDBC"); | ||
| } catch (ClassNotFoundException e) { | ||
| throw new RuntimeException("Failed to load SQLite JDBC driver", e); | ||
| } | ||
|
|
||
| Path tempDir; | ||
| try { | ||
| tempDir = Files.createTempDirectory("service-index"); | ||
| } catch (IOException e) { | ||
| throw new RuntimeException("Failed to create a temporary directory", e); | ||
| } | ||
|
|
||
| URL dbUrl = getClass().getClassLoader().getResource(INDEX_FILE_NAME); | ||
| if (dbUrl == null) { | ||
| throw new RuntimeException("Database resource not found: " + INDEX_FILE_NAME); | ||
| } | ||
| Path tempFile = tempDir.resolve(INDEX_FILE_NAME); | ||
| try { | ||
| Files.copy(dbUrl.openStream(), tempFile); | ||
| } catch (IOException e) { | ||
| throw new RuntimeException("Failed to copy the database file to the temporary directory", e); | ||
| } | ||
|
|
||
| dbPath = "jdbc:sqlite:" + tempFile.toString(); | ||
| } | ||
|
|
||
| public Optional<FunctionData> getListener(String org, String module) { | ||
| StringBuilder sql = new StringBuilder("SELECT "); | ||
| sql.append("l.listener_id, "); | ||
| sql.append("l.name AS listener_name, "); | ||
| sql.append("l.description AS listener_description, "); | ||
| sql.append("l.return_error, "); | ||
| sql.append("p.package_id, "); | ||
| sql.append("p.name AS package_name, "); | ||
| sql.append("p.org, "); | ||
| sql.append("p.version "); | ||
| sql.append("FROM Listener l "); | ||
| sql.append("JOIN Package p ON l.package_id = p.package_id "); | ||
| sql.append("WHERE p.org = ? "); | ||
| sql.append("AND p.name = ? "); | ||
|
|
||
| try (Connection conn = DriverManager.getConnection(dbPath); | ||
| PreparedStatement stmt = conn.prepareStatement(sql.toString())) { | ||
| stmt.setString(1, org); | ||
| stmt.setString(2, module); | ||
|
|
||
| ResultSet rs = stmt.executeQuery(); | ||
| if (rs.next()) { | ||
| FunctionData functionData = new FunctionData( | ||
| rs.getInt("listener_id"), | ||
| rs.getString("listener_name"), | ||
| rs.getString("listener_description"), | ||
| null, | ||
| rs.getString("package_name"), | ||
| rs.getString("org"), | ||
| rs.getString("version"), | ||
| null, | ||
| null, | ||
| rs.getBoolean("return_error"), | ||
| false); | ||
| functionData.setPackageId(rs.getString("package_id")); | ||
| return Optional.of(functionData); | ||
| } | ||
| return Optional.empty(); | ||
| } catch (SQLException e) { | ||
| Logger.getGlobal().severe("Error executing query: " + e.getMessage()); | ||
| return Optional.empty(); | ||
| } | ||
| } | ||
|
|
||
| public LinkedHashMap<String, ParameterData> getFunctionParametersAsMap(int listenerId) { | ||
| String sql = "SELECT " + | ||
| "p.parameter_id, " + | ||
| "p.name, " + | ||
| "p.type, " + | ||
| "p.kind, " + | ||
| "p.optional, " + | ||
| "p.default_value, " + | ||
| "p.description, " + | ||
| "p.import_statements, " + | ||
| "pmt.type AS member_type, " + | ||
| "pmt.kind AS member_kind, " + | ||
| "pmt.package AS member_package " + | ||
| "FROM Parameter p " + | ||
| "LEFT JOIN ParameterMemberType pmt ON p.parameter_id = pmt.parameter_id " + | ||
| "WHERE p.listener_id = ?;"; | ||
|
|
||
| try (Connection conn = DriverManager.getConnection(dbPath); | ||
| PreparedStatement stmt = conn.prepareStatement(sql)) { | ||
| stmt.setInt(1, listenerId); | ||
| ResultSet rs = stmt.executeQuery(); | ||
|
|
||
| // Use a builder to accumulate parameter data and member types | ||
| LinkedHashMap<String, ParameterDataBuilder> builders = new LinkedHashMap<>(); | ||
|
|
||
| while (rs.next()) { | ||
| String paramName = rs.getString("name"); | ||
| int parameterId = rs.getInt("parameter_id"); | ||
| String type = rs.getString("type"); | ||
| ParameterData.Kind kind = ParameterData.Kind.valueOf(rs.getString("kind")); | ||
| String defaultValue = rs.getString("default_value"); | ||
| String description = rs.getString("description"); | ||
| boolean optional = rs.getBoolean("optional"); | ||
| String importStatements = rs.getString("import_statements"); | ||
|
|
||
| // Member type data | ||
| String memberType = rs.getString("member_type"); | ||
| String memberKind = rs.getString("member_kind"); | ||
| String memberPackage = rs.getString("member_package"); | ||
|
|
||
| // Get or create the builder for this parameter | ||
| ParameterDataBuilder builder = builders.get(paramName); | ||
| if (builder == null) { | ||
| builder = new ParameterDataBuilder(); | ||
| builder.parameterId = parameterId; | ||
| builder.name = paramName; | ||
| builder.type = type; | ||
| builder.kind = kind; | ||
| builder.defaultValue = defaultValue; | ||
| builder.description = description; | ||
| builder.optional = optional; | ||
| builder.importStatements = importStatements; | ||
| builders.put(paramName, builder); | ||
| } | ||
|
|
||
| // Add member type if present | ||
| if (memberType != null) { | ||
| ParameterMemberTypeData memberData = new ParameterMemberTypeData( | ||
| memberType, memberKind, memberPackage); | ||
| builder.typeMembers.add(memberData); | ||
| } | ||
| } | ||
|
|
||
| // Convert builders to ParameterData | ||
| LinkedHashMap<String, ParameterData> parameterResults = new LinkedHashMap<>(); | ||
| for (ParameterDataBuilder builder : builders.values()) { | ||
| parameterResults.put(builder.name, builder.build()); | ||
| } | ||
| return parameterResults; | ||
|
|
||
| } catch (SQLException e) { | ||
| Logger.getGlobal().severe("Error executing query: " + e.getMessage()); | ||
| return new LinkedHashMap<>(); | ||
| } | ||
| } | ||
|
|
||
| // Helper builder class | ||
| private static class ParameterDataBuilder { | ||
| int parameterId; | ||
| String name; | ||
| String type; | ||
| ParameterData.Kind kind; | ||
| String defaultValue; | ||
| String description; | ||
| boolean optional; | ||
| String importStatements; | ||
| List<ParameterMemberTypeData> typeMembers = new ArrayList<>(); | ||
|
|
||
| ParameterData build() { | ||
| return new ParameterData( | ||
| parameterId, | ||
| name, | ||
| type, | ||
| kind, | ||
| defaultValue, | ||
| description, | ||
| optional, | ||
| importStatements, | ||
| typeMembers | ||
| ); | ||
| } | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.