db相关调整

2023-03-10 15:48:42 +08:00
parent bdf9b6dc5e
commit 3b0403052b
2 changed files with 52 additions and 1 deletions
@@ -19,7 +19,7 @@
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
-package fun.asgc.neutrino.proxy.server.base.rest;
+package fun.asgc.neutrino.proxy.server.base.db;

 import com.google.common.collect.Lists;
 import fun.asgc.neutrino.core.db.template.JdbcTemplate;
@@ -0,0 +1,51 @@
+package fun.asgc.neutrino.proxy.server.base.db;
+
+import com.alibaba.druid.pool.DruidDataSource;
+import fun.asgc.neutrino.core.db.template.JdbcTemplate;
+import fun.asgc.neutrino.proxy.server.base.rest.config.DbConfig;
+import fun.asgc.neutrino.proxy.server.constant.DbTypeEnum;
+import org.noear.solon.annotation.Bean;
+import org.noear.solon.annotation.Configuration;
+import org.noear.solon.annotation.Inject;
+import org.sqlite.SQLiteConfig;
+import org.sqlite.SQLiteDataSource;
+
+import javax.sql.DataSource;
+
+/**
+ * @author: aoshiguchen
+ * @date: 2023/3/10
+ */
+@Configuration
+public class DbConfiguration {
+
+    @Bean(value = "dataSource", typed = true)
+    public DataSource dataSource(@Inject DbConfig dbConfig) {
+        DbTypeEnum dbTypeEnum = DbTypeEnum.of(dbConfig.getType());
+        if (DbTypeEnum.SQLITE == dbTypeEnum) {
+            SQLiteDataSource dataSource = new SQLiteDataSource();
+            dataSource.setUrl(dbConfig.getUrl());
+            dataSource.setJournalMode(SQLiteConfig.JournalMode.WAL.getValue());
+            return dataSource;
+        } else if (DbTypeEnum.MYSQL == dbTypeEnum) {
+            DruidDataSource dataSource = new DruidDataSource();
+            dataSource.setDriverClassName(dbConfig.getDriverClass());
+            dataSource.setUrl(dbConfig.getUrl());
+            dataSource.setInitialSize(5);
+            dataSource.setMinIdle(5);
+            dataSource.setMaxActive(20);
+            dataSource.setMaxWait(60000);
+            dataSource.setPoolPreparedStatements(true);
+            dataSource.setUsername(dbConfig.getUsername());
+            dataSource.setPassword(dbConfig.getPassword());
+            return dataSource;
+        }
+
+        return null;
+    }
+
+    @Bean
+    public JdbcTemplate jdbcTemplate(@Inject("dataSource") DataSource dataSource) {
+        return new JdbcTemplate(dataSource);
+    }
+}