Clone and add config to turn off split

Author: daipham3198

Adapter/Pdo/CloneMysql.php

@@ -0,0 +1,8 @@
+<?php
+
+namespace CodeFarm\SplitDb\Adapter\Pdo;
+
+class CloneMysql extends \Magento\Framework\DB\Adapter\Pdo\Mysql
+{
+
+}

Adapter/Pdo/Mysql.php

@@ -33,16 +33,21 @@ public function __construct(
             unset($config['excluded_areas']);
         }
 
-        if(isset($config['slaves'])){
+        if(isset($config['slaves']) && isset($config['is_split'])){
             // keep the same slave throughout the request
             $slaveIndex = rand(0, (count($config['slaves']) - 1));
             $slaveConfig = $config['slaves'][$slaveIndex];
             unset($config['slaves']);
-            $slaveConfig = array_merge(
-                $config,
-                $slaveConfig
-            );
-            $this->readConnection = ObjectManager::getInstance()->create(\Magento\Framework\DB\Adapter\Pdo\Mysql::class, [
+            if($config['is_split']){
+                $slaveConfig = array_merge(
+                    $config,
+                    $slaveConfig
+                );
+            }else{
+                $slaveConfig = $config;
+            }
+
+            $this->readConnection = ObjectManager::getInstance()->create(CloneMysql::class, [
                 'string' => $string,
                 'dateTime' => $dateTime,
                 'logger' => $logger,
@@ -52,7 +57,7 @@ public function __construct(
             ]);
         }else{
             // create a read connection with the same credentials as the writer
-            $this->readConnection = ObjectManager::getInstance()->create(\Magento\Framework\DB\Adapter\Pdo\Mysql::class, [
+            $this->readConnection = ObjectManager::getInstance()->create(CloneMysql::class, [
                 'string' => $string,
                 'dateTime' => $dateTime,
                 'logger' => $logger,
@@ -106,7 +111,7 @@ protected function canUseReader($sql)
             'GET_LOCK'
         ];
         foreach($writerSqlIdentifiers as $writerSqlIdentifier){
-            if(stripos($sql, $writerSqlIdentifier) !== false){
+            if(stripos(substr($sql, 0 , 20), $writerSqlIdentifier) !== false){
                 return false;
             }
         }

Console/Command/ChangeDbMode.php

@@ -36,10 +36,17 @@ protected function execute(InputInterface $input, OutputInterface $output)
     {
         if ($mode = $input->getOption(self::MODE)) {
             if($mode == 'default' || $mode == 'split'){
+                $isActive = ($mode == 'split');
                 $write = ObjectManager::getInstance()->create(\Magento\Framework\App\DeploymentConfig\Writer::class);
                 $write->saveConfig([
                     ConfigFilePool::APP_ENV => [
-                        'DB_MODE' => $mode
+                        'db' => [
+                            'connection' => [
+                                'default' => [
+                                    'is_split' => $isActive
+                                ]
+                            ]
+                        ]
                     ]
                 ]);
             } else {