@@ -73,12 +73,38 @@ class DB
7373 */
7474 protected $ id'' ;
7575
76+ /**
77+ * 走主库的查寻语句
78+ *
79+ * @var array
80+ */
81+ private $ master'insert ' , 'update ' , 'delete ' ];
82+
83+ /**
84+ * 当前查询主从
85+ *
86+ * @var string
87+ */
88+ private $ masterSlave'' ;
89+
90+ /**
91+ * 数据库配置
92+ *
93+ * @var array
94+ */
95+ private $ dbConfig
96+ 'dbhost ' => '' ,
97+ 'dbname ' => '' ,
98+ 'username ' => '' ,
99+ 'password ' => ''
100+ ];
101+
76102 /**
77103 * 构造函数
78104 */
79105 public function __construct ()
80106 {
81- $ this init ();
107+ // $this->init();
82108 }
83109
84110 /**
@@ -96,20 +122,25 @@ public static function table($tableName = '')
96122 if (! empty ($ prefix
97123 $ dbtableName = $ prefix'_ ' . $ dbtableName ;
98124 }
99- $ dbinit ();
125+ // $db->init();
100126
101127 return $ db
102128 }
103129
104130 /**
105131 * 初始化策略
106132 *
133+ * @param $masterOrSlave 初始化主库还是从库
107134 * @return void
108135 */
109- public function init ()
136+ public function init ($ masterOrSlave = '' )
110137 {
111138 $ configAPP ::$ containergetSingle ('config ' );
112139 $ this dbtype = $ configconfig ['database ' ]['dbtype ' ];
140+ if (! empty ($ masterOrSlave
141+ $ this masterSlave = $ masterOrSlave
142+ }
143+ $ this isMasterOrSlave ();
113144 $ this decide ();
114145 }
115146
@@ -121,14 +152,77 @@ public function init()
121152 public function decide ()
122153 {
123154 $ dbStrategyName$ this dbStrategyMap [$ this dbtype ];
124- $ this dbInstance = APP ::$ containersetSingle (
125- $ this dbtype ,
126- function () use ($ dbStrategyName
127- return new $ dbStrategyName
155+ $ dbConfig$ this dbConfig ;
156+ $ this dbInstance = APP ::$ containergetSingle (
157+ "{$ this dbtype }- {$ this masterSlave }" ,
158+ function () use ($ dbStrategyName$ dbConfig
159+ return new $ dbStrategyName
160+ $ dbConfig'dbhost ' ],
161+ $ dbConfig'dbname ' ],
162+ $ dbConfig'username ' ],
163+ $ dbConfig'password ' ]
164+ );
128165 }
129166 );
130167 }
131168
169+ /**
170+ * 判断走主库还是从库
171+ *
172+ * @return void
173+ */
174+ public function isMasterOrSlave ()
175+ {
176+ if (! empty ($ this masterSlave )) {
177+ $ this initMaster ();
178+ return ;
179+ }
180+ foreach ($ this master as $ v
181+ $ resstripos ($ this sql , $ v
182+ if ($ res0 || $ res
183+ $ this initMaster ();
184+ return ;
185+ }
186+ }
187+ $ this initSlave ();
188+ }
189+
190+ /**
191+ * 初始化主库
192+ */
193+ public function initMaster ()
194+ {
195+ $ configAPP ::$ containergetSingle ('config ' );
196+ $ dbConfig$ configconfig ['database ' ];
197+ $ this dbConfig ['dbhost ' ] = $ dbConfig'dbhost ' ];
198+ $ this dbConfig ['dbname ' ] = $ dbConfig'dbname ' ];
199+ $ this dbConfig ['username ' ] = $ dbConfig'username ' ];
200+ $ this dbConfig ['password ' ] = $ dbConfig'password ' ];
201+
202+ $ this masterSlave = 'master ' ;
203+ }
204+
205+ /**
206+ * 初始化从库
207+ */
208+ public function initSlave ()
209+ {
210+ $ configAPP ::$ containergetSingle ('config ' );
211+ if (! isset ($ configconfig ['database ' ]['slave ' ])) {
212+ $ this initMaster ();
213+ return ;
214+ }
215+ $ slave$ configconfig ['database ' ]['slave ' ];
216+ $ randSlave$ slavearray_rand ($ slave
217+ $ dbConfig$ configconfig ["database-slave- {$ randSlave ];
218+ $ this dbConfig ['dbhost ' ] = $ dbConfig'dbhost ' ];
219+ $ this dbConfig ['dbname ' ] = $ dbConfig'dbname ' ];
220+ $ this dbConfig ['username ' ] = $ dbConfig'username ' ];
221+ $ this dbConfig ['password ' ] = $ dbConfig'password ' ];
222+
223+ $ this masterSlave = "slave- {$ randSlave ;
224+ }
225+
132226 /**
133227 * 查找一条数据
134228 *
@@ -165,6 +259,7 @@ public function findAll($data = [])
165259 public function save ($ data
166260 {
167261 $ this insert ($ data
262+ $ this init ();
168263 $ functionName__FUNCTION__ ;
169264 return $ this dbInstance ->$ functionName$ this
170265 }
@@ -231,6 +326,7 @@ public function sum($data = '')
231326 public function query ($ sql'' )
232327 {
233328 $ this querySql ($ sql
329+ $ this init ();
234330 return $ this dbInstance ->query ($ this
235331 }
236332
@@ -250,6 +346,8 @@ public function buildSql()
250346 if (! empty ($ this limit )) {
251347 $ this sql .= $ this limit ;
252348 }
349+
350+ $ this init ();
253351 }
254352
255353 /**
@@ -259,10 +357,11 @@ public function buildSql()
259357 */
260358 public static function beginTransaction ()
261359 {
262- $ instanceAPP ::$ containersetSingle ('DB ' , function () {
360+ $ instanceAPP ::$ containergetSingle ('DB ' , function () {
263361 return new DB ();
264362 }
265363 );
364+ $ instanceinit ('master ' );
266365 $ instancedbInstance ->beginTransaction ();
267366 }
268367
@@ -273,10 +372,11 @@ public static function beginTransaction()
273372 */
274373 public static function commit ()
275374 {
276- $ instanceAPP ::$ containersetSingle ('DB ' , function () {
375+ $ instanceAPP ::$ containergetSingle ('DB ' , function () {
277376 return new DB ();
278377 }
279378 );
379+ $ instanceinit ('master ' );
280380 $ instancedbInstance ->commit ();
281381 }
282382
@@ -291,6 +391,7 @@ public static function rollBack()
291391 return new DB ();
292392 }
293393 );
394+ $ instanceinit ('master ' );
294395 $ instancedbInstance ->rollBack ();
295396 }
296397
0 commit comments