[#43] Mysql Master/Slave Replication 적용

src/main/java/com/flab/makedel/annotation/SetDataSource.java

@@ -0,0 +1,18 @@
+package com.flab.makedel.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.METHOD)
+public @interface SetDataSource {
+
+    DataSourceType dataSourceType();
+
+    enum DataSourceType {
+        MASTER, SLAVE;
+    }
+
+}

src/main/java/com/flab/makedel/aop/SetDataSourceAspect.java

@@ -0,0 +1,27 @@
+package com.flab.makedel.aop;
+
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
+import com.flab.makedel.exception.WrongDataSourceException;
+import com.flab.makedel.routingdatasource.RoutingDataSourceManager;
+import java.sql.SQLException;
+import org.aspectj.lang.annotation.Aspect;
+import org.aspectj.lang.annotation.Before;
+import org.springframework.stereotype.Component;
+
+@Aspect
+@Component
+public class SetDataSourceAspect {
+
+    @Before("@annotation(com.flab.makedel.annotation.SetDataSource) && @annotation(target)")
+    public void setDataSource(SetDataSource target) throws WrongDataSourceException {
+
+        if (target.dataSourceType() == DataSourceType.MASTER
+            || target.dataSourceType() == DataSourceType.SLAVE) {
+            RoutingDataSourceManager.setCurrentDataSourceName(target.dataSourceType());
+        } else {
+            throw new WrongDataSourceException("Wrong DataSource Type : Should Check Exception");
+        }
+
+    }
+}

src/main/java/com/flab/makedel/config/DataSourceConfig.java

@@ -0,0 +1,86 @@
+package com.flab.makedel.config;
+
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
+import com.flab.makedel.routingdatasource.RoutingDataSourceManager;
+import java.util.HashMap;
+import java.util.Map;
+import javax.sql.DataSource;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.boot.jdbc.DataSourceBuilder;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.PropertySource;
+import org.springframework.jdbc.datasource.DataSourceTransactionManager;
+import org.springframework.jdbc.datasource.LazyConnectionDataSourceProxy;
+import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
+import org.springframework.transaction.PlatformTransactionManager;
+import org.springframework.transaction.support.TransactionSynchronizationManager;
+
+@Configuration
+@PropertySource("classpath:/application-dev.properties")
+public class DataSourceConfig {
+
+    @Bean
+    @ConfigurationProperties(prefix = "spring.datasource.master")
+    public DataSource masterDataSource() {
+        return DataSourceBuilder.create().build();
+    }
+
+    @Bean
+    @ConfigurationProperties(prefix = "spring.datasource.slave")
+    public DataSource slaveDataSource() {
+        return DataSourceBuilder.create().build();
+    }
+
+    @Bean
+    public DataSource routingDataSource(
+        @Qualifier(value = "masterDataSource") DataSource masterDataSource,
+        @Qualifier(value = "slaveDataSource") DataSource slaveDataSource) {
+
+        AbstractRoutingDataSource routingDataSource = new AbstractRoutingDataSource() {
+            @Override
+            protected Object determineCurrentLookupKey() {
+                DataSourceType dataSourceType = RoutingDataSourceManager.getCurrentDataSourceName();
+
+                if (TransactionSynchronizationManager
+                    .isActualTransactionActive()) {
+                    boolean readOnly = TransactionSynchronizationManager
+                        .isCurrentTransactionReadOnly();
+                    if (readOnly) {
+                        dataSourceType = DataSourceType.SLAVE;
+                    } else {
+                        dataSourceType = DataSourceType.MASTER;
+                    }
+                }
+
+                RoutingDataSourceManager.removeCurrentDataSourceName();
+                return dataSourceType;
+            }
+        };
+
+        Map<Object, Object> targetDataSources = new HashMap<>();
+
+        targetDataSources.put(DataSourceType.MASTER, masterDataSource);
+        targetDataSources.put(DataSourceType.SLAVE, slaveDataSource);
+
+        routingDataSource.setTargetDataSources(targetDataSources);
+        routingDataSource.setDefaultTargetDataSource(masterDataSource);
+
+        return routingDataSource;
+    }
+
+    @Bean
+    public DataSource lazyRoutingDataSource(
+        @Qualifier(value = "routingDataSource") DataSource routingDataSource) {
+        return new LazyConnectionDataSourceProxy(routingDataSource);
+    }
+
+    @Bean
+    public PlatformTransactionManager transactionManager(
+        @Qualifier(value = "lazyRoutingDataSource") DataSource lazyRoutingDataSource) {
+        DataSourceTransactionManager transactionManager = new DataSourceTransactionManager();
+        transactionManager.setDataSource(lazyRoutingDataSource);
+        return transactionManager;
+    }
+}

src/main/java/com/flab/makedel/config/DatabaseConfig.java

@@ -5,6 +5,7 @@
 import org.mybatis.spring.SqlSessionFactoryBean;
 import org.mybatis.spring.SqlSessionTemplate;
 import org.mybatis.spring.annotation.MapperScan;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.ApplicationContext;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
@@ -33,9 +34,11 @@ public class DatabaseConfig {
     private final ApplicationContext applicationContext;
 
     @Bean
-    public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
+    public SqlSessionFactory sqlSessionFactory(
+        @Qualifier(value = "lazyRoutingDataSource") DataSource lazyRoutingDataSource)
+        throws Exception {
         SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
-        sqlSessionFactoryBean.setDataSource(dataSource);
+        sqlSessionFactoryBean.setDataSource(lazyRoutingDataSource);
         sqlSessionFactoryBean.setMapperLocations(
             applicationContext.getResources("classpath:/mapper/**/*.xml"));
         return sqlSessionFactoryBean.getObject();

src/main/java/com/flab/makedel/Exception/DuplicatedIdException.java → src/main/java/com/flab/makedel/exception/DuplicatedIdException.java

@@ -1,4 +1,4 @@
-package com.flab.makedel.Exception;
+package com.flab.makedel.exception;
 
 public class DuplicatedIdException extends RuntimeException {
 

src/main/java/com/flab/makedel/Exception/NotExistIdException.java → src/main/java/com/flab/makedel/exception/NotExistIdException.java

@@ -1,4 +1,4 @@
-package com.flab.makedel.Exception;
+package com.flab.makedel.exception;
 
 public class NotExistIdException extends RuntimeException {
 

src/main/java/com/flab/makedel/exception/WrongDataSourceException.java

@@ -0,0 +1,10 @@
+package com.flab.makedel.exception;
+
+import java.sql.SQLException;
+
+public class WrongDataSourceException extends SQLException {
+
+    public WrongDataSourceException(String message) {
+        super(message);
+    }
+}

src/main/java/com/flab/makedel/mapper/MenuMapper.java

@@ -1,14 +1,19 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.MenuDTO;
 import java.util.List;
 
 public interface MenuMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertMenu(MenuDTO menu);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void deleteMenu(long menuId);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     List<MenuDTO> selectStoreMenu(long storeId);
 
 }

src/main/java/com/flab/makedel/mapper/OptionMapper.java

@@ -1,5 +1,7 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.OptionDTO;
 import java.util.List;
 
@@ -15,10 +17,13 @@
 
 public interface OptionMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertOptionList(List<OptionDTO> optionList);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void deleteOptionList(List<OptionDTO> optionList);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     List<OptionDTO> selectOptionList(long menuId);
 
 }

src/main/java/com/flab/makedel/mapper/OrderMapper.java

@@ -1,5 +1,7 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.OrderDTO;
 import com.flab.makedel.dto.OrderDTO.OrderStatus;
 import com.flab.makedel.dto.OrderDetailDTO;
@@ -9,18 +11,25 @@
 
 public interface OrderMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertOrder(OrderDTO orderDTO);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void completeOrder(long totalPrice, long orderId, OrderStatus orderStatus);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     OrderReceiptDTO selectOrderReceipt(long orderId);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     List<OrderStoreDetailDTO> selectDetailStoreOrder(long storeId);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void approveOrder(long orderId, OrderStatus orderStatus);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void updateStandbyOrderToDelivering(long orderId, String riderId, OrderStatus orderStatus);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void finishDeliveringOrder(long orderId, OrderStatus orderStatus);
 
 }

src/main/java/com/flab/makedel/mapper/OrderMenuMapper.java

@@ -1,10 +1,13 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.OrderMenuDTO;
 import java.util.List;
 
 public interface OrderMenuMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertOrderMenu(List<OrderMenuDTO> orderMenuList);
 
 }

src/main/java/com/flab/makedel/mapper/OrderMenuOptionMapper.java

@@ -1,10 +1,13 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.OrderMenuOptionDTO;
 import java.util.List;
 
 public interface OrderMenuOptionMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertOrderMenuOption(List<OrderMenuOptionDTO> orderMenuOptionList);
 
 }

src/main/java/com/flab/makedel/mapper/PayMapper.java

@@ -1,9 +1,12 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.PayDTO;
 
 public interface PayMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertPay(PayDTO payDTO);
 
 }

src/main/java/com/flab/makedel/mapper/StoreCategoryMapper.java

@@ -1,5 +1,7 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.StoreCategoryDTO;
 import java.util.List;
 
@@ -11,6 +13,7 @@
 
 public interface StoreCategoryMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     List<StoreCategoryDTO> selectCategoryList();
 
 }

src/main/java/com/flab/makedel/mapper/StoreMapper.java

@@ -1,23 +1,32 @@
 package com.flab.makedel.mapper;
 
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.StoreInfoDTO;
 import com.flab.makedel.dto.StoreDTO;
 import java.util.List;
 
 public interface StoreMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertStore(StoreDTO store);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     List<StoreDTO> selectStoreList(String ownerId);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     boolean isMyStore(long storeId, String ownerId);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     StoreDTO selectStore(long storeId, String ownerId);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void closeMyStore(long storeId);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void openMyStore(long storeId);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     StoreInfoDTO selectStoreInfo(long storeId);
 
 }

src/main/java/com/flab/makedel/mapper/UserMapper.java

@@ -1,23 +1,31 @@
 package com.flab.makedel.mapper;
 
 import com.flab.makedel.annotation.LoginCheck.UserLevel;
+import com.flab.makedel.annotation.SetDataSource;
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
 import com.flab.makedel.dto.UserDTO;
 import com.flab.makedel.dto.UserInfoDTO;
 import org.apache.ibatis.annotations.Mapper;
 
 @Mapper
 public interface UserMapper {
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     boolean isExistsId(String id);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     UserInfoDTO selectUserInfo(String id);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void insertUser(UserDTO user);
 
+    @SetDataSource(dataSourceType = DataSourceType.SLAVE)
     UserDTO selectUserById(String id);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void deleteUser(String id);
 
+    @SetDataSource(dataSourceType = DataSourceType.MASTER)
     void updateUserPassword(String id, String newPassword);
 
 }

src/main/java/com/flab/makedel/routingdatasource/RoutingDataSourceManager.java

@@ -0,0 +1,20 @@
+package com.flab.makedel.routingdatasource;
+
+import com.flab.makedel.annotation.SetDataSource.DataSourceType;
+
+public class RoutingDataSourceManager {
+
+    private static final ThreadLocal<DataSourceType> currentDataSourceName = new ThreadLocal<>();
+
+    public static void setCurrentDataSourceName(DataSourceType dataSourceType) {
+        currentDataSourceName.set(dataSourceType);
+    }
+
+    public static DataSourceType getCurrentDataSourceName() {
+        return currentDataSourceName.get();
+    }
+
+    public static void removeCurrentDataSourceName() {
+        currentDataSourceName.remove();
+    }
+}