php pdo封装类代码(支持事务)

2. /**
3. * PDO数据库
4. * @copyright By GOOGLE
5. */
6. class pdo_db
7. {
8. /**
9. * PDO实例
10. * @var PDO
11. */
12. protected $_db;
13. /**
14. * PDO准备语句
15. * @var PDOStatement
16. */
17. protected $_stmt;
18. /**
19. * 最后的SQL语句
20. * @var string
21. */
22. protected $_sql;
23. /**
24. * 配置信息 $config=array('dsn'=>xxx,'name'=>xxx,'password'=>xxx,'option'=>xxx)
25. * @var array
26. */
27. protected $_config;
28. /**
29. * 构造函数
30. * @param array $config
31. */
32. public function __construct($config)
33. {
34. $this->_config = $config;
35. }
36. /**
37. * 连接数据库
38. * @return void
39. */
40. public function connect()
41. {
42. $this->_db = new PDO($this->_config['dsn'], $this->_config['name'], $this->_config['password'], $this->_config['option']);
43. //默认把结果序列化成stdClass
44. // $this->_db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_OBJ);
45. $this->_db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
46. //自己写代码捕获Exception
47. $this->_db->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_SILENT);
48. }
49. /**
50. * 断开连接
51. * @return void
52. */
53. public function disConnect()
54. {
55. $this->_db = null;
56. $this->_stmt = null;
57. }
58. /**
59. * 执行sql，返回新加入的id
60. * @param string $statement
61. * @return string
62. */
63. public function exec($statement)
64. {
65. if ($this->_db->exec($statement)){
66. $this->_sql = $statement;
67. return $this->lastId();
68. }
69. $this->errorMessage();
70. }
71. /**
72. * 查询sql
73. * @param string $statement
74. * @return pdo_db
75. */
76. public function query($statement)
77. {
78. $res = $this->_db->query($statement);
79. if ($res){
80. $this->_stmt = $res;
81. $this->_sql = $statement;
82. return $this;
83. }
84. $this->errorMessage();
85. }
86. /**
87. * 序列化一次数据
88. * @return mixed
89. */
90. public function fetchOne()
91. {
92. return $this->_stmt->fetch();
93. }
94. /**
95. * 序列化所有数据
96. * @return array
97. */
98. public function fetchAll()
99. {
100. return $this->_stmt->fetchAll();
101. }
102. /**
103. * 最后添加的id
104. * @return string
105. */
106. public function lastId()
107. {
108. return $this->_db->lastInsertId();
109. }
110. /**
111. * 影响的行数
112. * @return int
113. */
114. public function affectRows()
115. {
116. return $this->_stmt->rowCount();
117. }
118. /**
119. * 预备语句
120. * @param string $statement
121. * @return pdo_db
122. */
123. public function prepare($statement)
124. {
125. $res = $this->_db->prepare($statement);
126. if ($res){
127. $this->_stmt = $res;
128. $this->_sql = $statement;
129. return $this;
130. }
131. $this->errorMessage();
132. }
133. /**
134. * 绑定数据
135. * @param array $array
136. * @return pdo_db
137. */
138. public function bindArray($array)
139. {
140. foreach ($array as $k => $v){
141. if (is_array($v)){
142. //array的有效结构 array('value'=>xxx,'type'=>PDO::PARAM_XXX)
143. $this->_stmt->bindValue($k + 1, $v['value'], $v['type']);
144. } else{
145. $this->_stmt->bindValue($k + 1, $v, PDO::PARAM_STR);
146. }
147. }
148. return $this;
149. }
150. /**
151. * 执行预备语句
152. * @return bool
153. */
154. public function execute()
155. {
156. if ($this->_stmt->execute()){
157. return true;
158. }
159. $this->errorMessage();
160. }
161. /**
162. * 开启事务
163. * @return bool
164. */
165. public function beginTransaction()
166. {
167. return $this->_db->beginTransaction();
168. }
169. /**
170. * 执行事务
171. * @return bool
172. */
173. public function commitTransaction()
174. {
175. return $this->_db->commit();
176. }
177. /**
178. * 回滚事务
179. * @return bool
180. */
181. public function rollbackTransaction()
182. {
183. return $this->_db->rollBack();
184. }
185. /**
186. * 抛出错误
187. * @throws Error
188. * @return void
189. */
190. public function errorMessage()
191. {
192. return;
193. $msg = $this->_db->errorInfo();
194. throw new Exception('数据库错误：' . $msg[2]);
195. }
196. //---------------------
197. /**
198. * 单例实例
199. * @var pdo_db
200. */
201. protected static $_instance;
202. /**
203. * 默认数据库
204. * @static
205. * @param array $config
206. * @return pdo_db
207. */
208. public static function instance($config)
209. {
210. if (!self::$_instance instanceof pdo_db){
211. self::$_instance = new pdo_db($config);
212. self::$_instance->connect();
213. }
214. return self::$_instance;
215. }
216. //----------------------
217. /**
218. * 获取PDO支持的数据库
219. * @static
220. * @return array
221. */
222. public static function getSupportDriver(){
223. return PDO::getAvailableDrivers();
224. }
225. /**
226. * 获取数据库的版本信息
227. * @return array
228. */
229. public function getDriverVersion(){
230. $name = $this->_db->getAttribute(PDO::ATTR_DRIVER_NAME);
231. return array($name=>$this->_db->getAttribute(PDO::ATTR_CLIENT_VERSION));
232. }
233. }

复制代码