feat: Support mariadb datasource (#21)

Author: happenedIn

pom.xml

@@ -190,6 +190,12 @@
             <artifactId>hibernate-core</artifactId>
             <version>5.3.7.Final</version>
         </dependency>
+        <dependency>
+            <groupId>org.mariadb.jdbc</groupId>
+            <artifactId>mariadb-java-client</artifactId>
+            <version>3.1.4</version>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>mysql</groupId>
             <artifactId>mysql-connector-java</artifactId>

src/main/java/org/casbin/adapter/HibernateAdapter.java

@@ -59,7 +59,7 @@ private void open() throws SQLException {
     private void createDatabase() {
         Session session = factory.openSession();
         Transaction tx = session.beginTransaction();
-        if (this.databaseProductName.contains("MySQL")) {
+        if (this.databaseProductName.contains("MySQL") || this.databaseProductName.contains("MariaDB")) {
             session.createSQLQuery("CREATE DATABASE IF NOT EXISTS casbin").executeUpdate();
             session.createSQLQuery("USE casbin").executeUpdate();
         } else if (this.databaseProductName.contains("SQLServer")) {
@@ -78,7 +78,7 @@ private void createDatabase() {
     private void createTable() {
         Session session = factory.openSession();
         Transaction tx = session.beginTransaction();
-        if (this.databaseProductName.contains("MySQL")) {
+        if (this.databaseProductName.contains("MySQL") || this.databaseProductName.contains("MariaDB")) {
             session.createSQLQuery("CREATE TABLE IF NOT EXISTS casbin_rule (" +
                     "id INT not NULL primary key," +
                     "ptype VARCHAR(100) not NULL," +
@@ -130,7 +130,7 @@ private void createTable() {
     private void dropTable() {
         Session session = factory.openSession();
         Transaction tx = session.beginTransaction();
-        if (this.databaseProductName.contains("MySQL")) {
+        if (this.databaseProductName.contains("MySQL") || this.databaseProductName.contains("MariaDB")) {
             session.createSQLQuery("DROP TABLE IF EXISTS casbin_rule").executeUpdate();
         } else if (this.databaseProductName.contains("Oracle")) {
             session.createSQLQuery("declare " +
@@ -356,6 +356,8 @@ private void setDatabaseProductName() {
                 this.databaseProductName = "Oracle";
             } else if (this.driver.contains("sqlserver")) {
                 this.databaseProductName = "SQLServer";
+            } else if (this.driver.contains("mariadb")) {
+                this.databaseProductName = "MariaDB";
             }
         }
     }

src/test/java/org/casbin/test/HibernateAdapterDatasourceTest.java

@@ -0,0 +1,36 @@
+package org.casbin.test;
+
+import static org.junit.Assert.assertEquals;
+import java.sql.SQLException;
+import org.casbin.adapter.HibernateAdapter;
+import org.casbin.jcasbin.main.Enforcer;
+import org.junit.Test;
+import org.mariadb.jdbc.MariaDbDataSource;
+
+public class HibernateAdapterDatasourceTest {
+
+    private static final String URL = "jdbc:mariadb://localhost:3306/casbin?serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true&rewriteBatchedStatements=true";
+    private static final String USERNAME = "root";
+    private static final String PASSWORD = "casbin_test";
+
+    @Test
+    public void testInitDBfromMariaDBDatasource() throws SQLException {
+        MariaDbDataSource mariaDbDataSource = new MariaDbDataSource();
+        mariaDbDataSource.setUrl(URL);
+        mariaDbDataSource.setUser(USERNAME);
+        mariaDbDataSource.setPassword(PASSWORD);
+        
+        Enforcer e = new Enforcer("examples/rbac_with_domains_model.conf", new HibernateAdapter(mariaDbDataSource));
+
+        e.savePolicy(); //clear table
+
+        e.addPolicy("admin", "domain1", "data1", "read");
+        e.addGroupingPolicy("alice", "admin", "domain1");
+
+        testDomainEnforce(e, "alice", "domain1", "data1", "read", true);
+    }
+
+    private void testDomainEnforce(Enforcer e, String sub, String dom, String obj, String act, boolean res) {
+        assertEquals(res, e.enforce(sub, dom, obj, act));
+    }
+}