[GitHub] [shardingsphere] totalo commented on a change in pull request #15115: Make It is easy to use with narayana

[GitBox <gi...@apache.org>]

totalo commented on a change in pull request #15115:
URL: https://github.com/apache/shardingsphere/pull/15115#discussion_r793419280



##########
File path: shardingsphere-kernel/shardingsphere-transaction/shardingsphere-transaction-core/src/main/java/org/apache/shardingsphere/transaction/context/TransactionContextsBuilder.java
##########
@@ -61,4 +75,100 @@ private TransactionRule getTransactionRule() {
         Optional<TransactionRule> transactionRule = globalRules.stream().filter(each -> each instanceof TransactionRule).map(each -> (TransactionRule) each).findFirst();
         return transactionRule.orElseGet(() -> new TransactionRule(new DefaultTransactionRuleConfigurationBuilder().build()));
     }
+    
+    private void generateNarayanaConfig(final TransactionRule transactionRule) {
+        Map<Object, Object> result;
+        if (transactionRule.getDefaultType() == TransactionType.XA && transactionRule.getProviderType().equalsIgnoreCase("Narayana")) {
+            result = generateDefaultNarayanaConfig();
+            if (null != transactionRule.getProps()) {
+                swapJdbcStore(transactionRule, result);
+            }
+        } else {
+            return;
+        }
+        String value = NarayanaConfigMapToXml((LinkedHashMap<Object, Object>) result);
+        String path = ClassLoader.getSystemResource("").getPath();
+        System.out.println(path);
+        try (BufferedWriter bufferedWriter = Files.newBufferedWriter(Paths.get(path, "jbossts-properties.xml"))) {
+            bufferedWriter.write(value);
+            bufferedWriter.flush();
+        } catch (final IOException ex) {
+            log.error("generate narayana config file failed.");
+        }
+    }
+
+    private void swapJdbcStore(final TransactionRule transactionRule, Map<Object, Object> config) {
+        Object host = transactionRule.getProps().get("host");
+        Object port = transactionRule.getProps().get("port");
+        Object user = transactionRule.getProps().getProperty("user");
+        Object password = transactionRule.getProps().getProperty("password");
+        Object databaseName = transactionRule.getProps().getProperty("databaseName");
+        if (null != host && null != port && null != user && null != password && null != databaseName) {
+            String jdbcAccess = String.format("com.arjuna.ats.internal.arjuna.objectstore.jdbc.accessors.DynamicDataSourceJDBCAccess;ClassName=com.mysql.cj.jdbc.MysqlDataSource;URL=jdbc:mysql://%s:%d/%s;User=%s;Password=%s", host, port, databaseName, user, password);
+            config.put("ObjectStoreEnvironmentBean.objectStoreType", "com.arjuna.ats.internal.arjuna.objectstore.jdbc.JDBCStore");
+            config.put("ObjectStoreEnvironmentBean.jdbcAccess", jdbcAccess);
+            config.put("ObjectStoreEnvironmentBean.tablePrefix", "Action");
+            config.put("ObjectStoreEnvironmentBean.dropTable", true);
+            config.put("ObjectStoreEnvironmentBean.stateStore.objectStoreType", "com.arjuna.ats.internal.arjuna.objectstore.jdbc.JDBCStore");
+            config.put("ObjectStoreEnvironmentBean.stateStore.jdbcAccess", jdbcAccess);
+            config.put("ObjectStoreEnvironmentBean.stateStore.tablePrefix", "stateStore");
+            config.put("ObjectStoreEnvironmentBean.stateStore.dropTable", true);
+            config.put("ObjectStoreEnvironmentBean.communicationStore.objectStoreType", "com.arjuna.ats.internal.arjuna.objectstore.jdbc.JDBCStore");
+            config.put("ObjectStoreEnvironmentBean.communicationStore.jdbcAccess", jdbcAccess);
+            config.put("ObjectStoreEnvironmentBean.communicationStore.tablePrefix", "Communication");
+            config.put("ObjectStoreEnvironmentBean.communicationStore.dropTable", true);
+        }
+    }
+
+    private Map<Object, Object> generateDefaultNarayanaConfig() {
+        Map<Object, Object> result = new LinkedHashMap<>();
+        result.put("CoordinatorEnvironmentBean.commitOnePhase", "YES");
+        result.put("ObjectStoreEnvironmentBean.transactionSync", "ON");
+        result.put("CoreEnvironmentBean.nodeIdentifier", null == instanceId ? 1 : instanceId);
+        result.put("JTAEnvironmentBean.xaRecoveryNodes", null == instanceId ? 1 : instanceId);
+        List<String> xaResourceOrphanFilterClassNames = new LinkedList<>();
+        xaResourceOrphanFilterClassNames.add("com.arjuna.ats.internal.jta.recovery.arjunacore.JTATransactionLogXAResourceOrphanFilter");
+        xaResourceOrphanFilterClassNames.add("com.arjuna.ats.internal.jta.recovery.arjunacore.JTANodeNameXAResourceOrphanFilter");
+        xaResourceOrphanFilterClassNames.add("com.arjuna.ats.internal.jta.recovery.arjunacore.JTAActionStatusServiceXAResourceOrphanFilter");
+        result.put("JTAEnvironmentBean.xaResourceOrphanFilterClassNames", xaResourceOrphanFilterClassNames);
+        result.put("CoreEnvironmentBean.socketProcessIdPort", 0);
+        List<String> recoveryModuleClassNames = new LinkedList<>();
+        recoveryModuleClassNames.add("com.arjuna.ats.internal.arjuna.recovery.AtomicActionRecoveryModule");
+        recoveryModuleClassNames.add("com.arjuna.ats.internal.jta.recovery.arjunacore.XARecoveryModule");
+        result.put("RecoveryEnvironmentBean.recoveryModuleClassNames", recoveryModuleClassNames);
+        result.put("RecoveryEnvironmentBean.expiryScannerClassNames", "com.arjuna.ats.internal.arjuna.recovery.ExpiredTransactionStatusManagerScanner");
+        result.put("RecoveryEnvironmentBean.recoveryPort", 4712);
+        result.put("RecoveryEnvironmentBean.recoveryAddress", null);
+        result.put("RecoveryEnvironmentBean.transactionStatusManagerPort", 0);
+        result.put("RecoveryEnvironmentBean.transactionStatusManagerAddress", null);
+        result.put("RecoveryEnvironmentBean.recoveryListener", "NO");
+        result.put("RecoveryEnvironmentBean.recoveryBackoffPeriod", 1);
+        return result;
+    }
+
+    private String NarayanaConfigMapToXml(LinkedHashMap<Object, Object> sortedMap) {

Review comment:
       Maybe we can consider using freemark?




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: notifications-unsubscribe@shardingsphere.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org