第7周作业10

nononi · nononi · commit 90cc0da11b56 · 2022-04-17T14:42:14.000+08:00
diff --git a/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/datasource/DataSourceConfig.java b/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/datasource/DataSourceConfig.java
@@ -0,0 +1,49 @@
+package io.nononi.shardingspheredemo.datasource;
+
+import org.apache.commons.dbcp2.BasicDataSource;
+
+import org.apache.shardingsphere.api.config.masterslave.MasterSlaveRuleConfiguration;
+import org.apache.shardingsphere.shardingjdbc.api.MasterSlaveDataSourceFactory;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import javax.sql.DataSource;
+import java.sql.SQLException;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Properties;
+
+@Configuration
+public class DataSourceConfig {
+
+    @Bean
+    public DataSource dataSource() throws SQLException {
+        // Configure actual data sources
+        Map<String, DataSource> dataSourceMap = new HashMap<>();
+
+        // Configure master data source
+        BasicDataSource masterDataSource = new BasicDataSource();
+        masterDataSource.setDriverClassName("com.mysql.jdbc.Driver");
+        masterDataSource.setUrl("jdbc:mysql://localhost:3316/db");
+        masterDataSource.setUsername("root");
+        masterDataSource.setPassword("");
+        dataSourceMap.put("ds_master", masterDataSource);
+
+        // Configure the first slave data source
+        BasicDataSource slaveDataSource1 = new BasicDataSource();
+        slaveDataSource1.setDriverClassName("com.mysql.jdbc.Driver");
+        slaveDataSource1.setUrl("jdbc:mysql://localhost:3326/db");
+        slaveDataSource1.setUsername("root");
+        slaveDataSource1.setPassword("");
+        dataSourceMap.put("ds_slave", slaveDataSource1);
+
+
+        // Configure read-write split rule
+        MasterSlaveRuleConfiguration masterSlaveRuleConfig = new MasterSlaveRuleConfiguration("ds_master_slave", "ds_master", Arrays.asList("ds_slave0"));
+
+        // Get data source
+        DataSource dataSource = MasterSlaveDataSourceFactory.createDataSource(dataSourceMap, masterSlaveRuleConfig, new Properties());
+        return dataSource;
+    }
+}
diff --git a/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/service/OrderService.java b/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/service/OrderService.java
@@ -0,0 +1,6 @@
+package io.nononi.shardingspheredemo.service;
+
+public interface OrderService {
+    void insert(int id, String name);
+    void query();
+}
diff --git a/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/service/impl/OrderServiceImpl.java b/06db/work10ofweek7/src/main/java/io/nononi/shardingspheredemo/service/impl/OrderServiceImpl.java
@@ -0,0 +1,51 @@
+package io.nononi.shardingspheredemo.service.impl;
+
+import io.nononi.shardingspheredemo.service.OrderService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+@Component
+public class OrderServiceImpl implements OrderService {
+
+    @Autowired
+    private DataSource dataSource;
+
+    @Override
+    public void insert(int id, String name) {
+        try {
+            Connection conn = dataSource.getConnection();
+            String sql = "INSERT INTO `order` (id,name) VALUES (?,?);";
+            PreparedStatement stmt = conn.prepareStatement(sql);
+            stmt.setInt(1,id);
+            stmt.setString(2,name);
+            stmt.executeUpdate();
+
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+    }
+
+    @Override
+    public void query() {
+        String sql = "SELECT * FROM `order`;";
+        try {
+            Connection conn = dataSource.getConnection();
+            PreparedStatement stmt = conn.prepareStatement(sql);
+            ResultSet resultSet = stmt.executeQuery();
+            while (resultSet.next()) {
+                System.out.print(resultSet.getInt("id") + " ");
+                System.out.println(resultSet.getString("name") + " ");
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+    }
+}
