当前位置：Gxlcms > PHP教程 > QuentinZervaas'sDBO

QuentinZervaas'sDBO

时间：2021-07-01 10:21:17 帮助过：37人阅读

《Web 2.0 开发实战》一书的作者，Quentin Zervaas 在书中提到的一个简易PHP数据访问对象。
  1. /**
  2. * DatabaseObject
  3. *
  4. * Abstract class used to easily manipulate data in a database table
  5. * via simple load/save/delete methods
  6. */
  7. abstract class DatabaseObject
  8. {
  9. const TYPE_TIMESTAMP = 1;
  10. const TYPE_BOOLEAN = 2;
  12. protected static $types = array(self::TYPE_TIMESTAMP, self::TYPE_BOOLEAN);
  14. private $_id = null;
  15. private $_properties = array();
  18. protected $_db = null;
  19. protected $_table = '';
  20. protected $_idField = '';
  22. public function __construct(Zend_Db_Adapter_Abstract $db, $table, $idField)
  23. {
  24. $this->_db = $db;
  25. $this->_table = $table;
  26. $this->_idField = $idField;
  27. }
  29. public function load($id, $field = null)
  30. {
  31. if (strlen($field) == 0)
  32. $field = $this->_idField;
  34. if ($field == $this->_idField) {
  35. $id = (int) $id;
  36. if ($id <= 0)
  37. return false;
  38. }
  40. $query = sprintf('select %s from %s where %s = ?',
  41. join(', ', $this->getSelectFields()),
  42. $this->_table,
  43. $field);
  45. $query = $this->_db->quoteInto($query, $id);
  47. return $this->_load($query);
  48. }
  50. protected function getSelectFields($prefix = '')
  51. {
  52. $fields = array($prefix . $this->_idField);
  53. foreach ($this->_properties as $k => $v)
  54. $fields[] = $prefix . $k;
  56. return $fields;
  57. }
  59. protected function _load($query)
  60. {
  61. $result = $this->_db->query($query);
  62. $row = $result->fetch();
  63. if (!$row)
  64. return false;
  66. $this->_init($row);
  68. $this->postLoad();
  70. return true;
  71. }
  73. public function _init($row)
  74. {
  75. foreach ($this->_properties as $k => $v) {
  76. $val = $row[$k];
  78. switch ($v['type']) {
  79. case self::TYPE_TIMESTAMP:
  80. if (!is_null($val))
  81. $val = strtotime($val);
  82. break;
  83. case self::TYPE_BOOLEAN:
  84. $val = (bool) $val;
  85. break;
  86. }
  88. $this->_properties[$k]['value'] = $val;
  89. }
  90. $this->_id = (int) $row[$this->_idField];
  91. }
  94. public function save($useTransactions = true)
  95. {
  96. $update = $this->isSaved();
  98. if ($useTransactions)
  99. $this->_db->beginTransaction();
  101. if ($update)
  102. $commit = $this->preUpdate();
  103. else
  104. $commit = $this->preInsert();
  106. if (!$commit) {
  107. if ($useTransactions)
  108. $this->_db->rollback();
  109. return false;
  110. }
  112. $row = array();
  114. foreach ($this->_properties as $k => $v) {
  115. if ($update && !$v['updated'])
  116. continue;
  118. switch ($v['type']) {
  119. case self::TYPE_TIMESTAMP:
  120. if (!is_null($v['value'])) {
  121. if ($this->_db instanceof Zend_Db_Adapter_Pdo_Pgsql)
  122. $v['value'] = date('Y-m-d H:i:sO', $v['value']);
  123. else
  124. $v['value'] = date('Y-m-d H:i:s', $v['value']);
  125. }
  126. break;
  128. case self::TYPE_BOOLEAN:
  129. $v['value'] = (int) ((bool) $v['value']);
  130. break;
  131. }
  133. $row[$k] = $v['value'];
  134. }
  136. if (count($row) > 0) {
  137. // perform insert/update
  138. if ($update) {
  139. $this->_db->update($this->_table, $row, sprintf('%s = %d', $this->_idField, $this->getId()));
  140. }
  141. else {
  142. $this->_db->insert($this->_table, $row);
  143. $this->_id = $this->_db->lastInsertId($this->_table, $this->_idField);
  144. }
  145. }
  147. // update internal id
  149. if ($commit) {
  150. if ($update)
  151. $commit = $this->postUpdate();
  152. else
  153. $commit = $this->postInsert();
  154. }
  156. if ($useTransactions) {
  157. if ($commit)
  158. $this->_db->commit();
  159. else
  160. $this->_db->rollback();
  161. }
  163. return $commit;
  164. }
  166. public function delete($useTransactions = true)
  167. {
  168. if (!$this->isSaved())
  169. return false;
  171. if ($useTransactions)
  172. $this->_db->beginTransaction();
  174. $commit = $this->preDelete();
  176. if ($commit) {
  177. $this->_db->delete($this->_table, sprintf('%s = %d', $this->_idField, $this->getId()));
  178. }
  179. else {
  180. if ($useTransactions)
  181. $this->_db->rollback();
  182. return false;
  183. }
  185. $commit = $this->postDelete();
  187. $this->_id = null;
  189. if ($useTransactions) {
  190. if ($commit)
  191. $this->_db->commit();
  192. else
  193. $this->_db->rollback();
  194. }
  196. return $commit;
  197. }
  199. public function isSaved()
  200. {
  201. return $this->getId() > 0;
  202. }
  204. public function getId()
  205. {
  206. return (int) $this->_id;
  207. }
  209. public function getDb()
  210. {
  211. return $this->_db;
  212. }
  214. public function __set($name, $value)
  215. {
  216. if (array_key_exists($name, $this->_properties)) {
  217. $this->_properties[$name]['value'] = $value;
  218. $this->_properties[$name]['updated'] = true;
  219. return true;
  220. }
  222. return false;
  223. }
  225. public function __get($name)
  226. {
  227. return array_key_exists($name, $this->_properties) ? $this->_properties[$name]['value'] : null;
  228. }
  230. protected function add($field, $default = null, $type = null)
  231. {
  232. $this->_properties[$field] = array('value' => $default,
  233. 'type' => in_array($type, self::$types) ? $type : null,
  234. 'updated' => false);
  235. }
  237. protected function preInsert()
  238. {
  239. return true;
  240. }
  242. protected function postInsert()
  243. {
  244. return true;
  245. }
  247. protected function preUpdate()
  248. {
  249. return true;
  250. }
  252. protected function postUpdate()
  253. {
  254. return true;
  255. }
  257. protected function preDelete()
  258. {
  259. return true;
  260. }
  262. protected function postDelete()
  263. {
  264. return true;
  265. }
  267. protected function postLoad()
  268. {
  269. return true;
  270. }
  272. public static function BuildMultiple($db, $class, $data)
  273. {
  274. $ret = array();
  276. if (!class_exists($class))
  277. throw new Exception('Undefined class specified: ' . $class);
  279. $testObj = new $class($db);
  281. if (!$testObj instanceof DatabaseObject)
  282. throw new Exception('Class does not extend from DatabaseObject');
  284. foreach ($data as $row) {
  285. $obj = new $class($db);
  286. $obj->_init($row);
  288. $ret[$obj->getId()] = $obj;
  289. }
  291. return $ret;
  292. }
  293. }
  1. class DatabaseObject_User extends DatabaseObject
  2. {
  3. static $userTypes = array('member' => 'Member',
  4. 'administrator' => 'Administrator');
  6. public $profile = null;
  7. public $_newPassword = null;
  9. public function __construct($db)
  10. {
  11. parent::__construct($db, 'users', 'user_id');
  13. $this->add('username');
  14. $this->add('password');
  15. $this->add('user_type', 'member');
  16. $this->add('ts_created', time(), self::TYPE_TIMESTAMP);
  17. $this->add('ts_last_login', null, self::TYPE_TIMESTAMP);
  19. $this->profile = new Profile_User($db);
  20. }
  22. protected function preInsert()
  23. {
  24. $this->_newPassword = Text_Password::create(8);
  25. $this->password = $this->_newPassword;
  26. return true;
  27. }
  29. protected function postLoad()
  30. {
  31. $this->profile->setUserId($this->getId());
  32. $this->profile->load();
  33. }
  35. protected function postInsert()
  36. {
  37. $this->profile->setUserId($this->getId());
  38. $this->profile->save(false);
  40. $this->sendEmail('user-register.tpl');
  41. return true;
  42. }
  44. protected function postUpdate()
  45. {
  46. $this->profile->save(false);
  47. return true;
  48. }
  50. protected function preDelete()
  51. {
  52. $this->profile->delete();
  53. return true;
  54. }
  56. public function sendEmail($tpl)
  57. {
  58. $templater = new Templater();
  59. $templater->user = $this;
  61. // fetch the e-mail body
  62. $body = $templater->render('email/' . $tpl);
  64. // extract the subject from the first line
  65. list($subject, $body) = preg_split('/\r|\n/', $body, 2);
  67. // now set up and send the e-mail
  68. $mail = new Zend_Mail();
  70. // set the to address and the user's full name in the 'to' line
  71. $mail->addTo($this->profile->email,
  72. trim($this->profile->first_name . ' ' .
  73. $this->profile->last_name));
  75. // get the admin 'from' details from the config
  76. $mail->setFrom(Zend_Registry::get('config')->email->from->email,
  77. Zend_Registry::get('config')->email->from->name);
  79. // set the subject and body and send the mail
  80. $mail->setSubject(trim($subject));
  81. $mail->setBodyText(trim($body));
  82. $mail->send();
  83. }
  85. public function createAuthIdentity()
  86. {
  87. $identity = new stdClass;
  88. $identity->user_id = $this->getId();
  89. $identity->username = $this->username;
  90. $identity->user_type = $this->user_type;
  91. $identity->first_name = $this->profile->first_name;
  92. $identity->last_name = $this->profile->last_name;
  93. $identity->email = $this->profile->email;
  95. return $identity;
  96. }
  98. public function loginSuccess()
  99. {
  100. $this->ts_last_login = time();
  101. unset($this->profile->new_password);
  102. unset($this->profile->new_password_ts);
  103. unset($this->profile->new_password_key);
  104. $this->save();
  106. $message = sprintf('Successful login attempt from %s user %s',
  107. $_SERVER['REMOTE_ADDR'],
  108. $this->username);
  110. $logger = Zend_Registry::get('logger');
  111. $logger->notice($message);
  112. }
  113. static public function LoginFailure($username, $code = '')
  114. {
  115. switch ($code) {
  116. case Zend_Auth_Result::FAILURE_IDENTITY_NOT_FOUND:
  117. $reason = 'Unknown username';
  118. break;
  119. case Zend_Auth_Result::FAILURE_IDENTITY_AMBIGUOUS:
  120. $reason = 'Multiple users found with this username';
  121. break;
  122. case Zend_Auth_Result::FAILURE_CREDENTIAL_INVALID:
  123. $reason = 'Invalid password';
  124. break;
  125. default:
  126. $reason = '';
  127. }
  129. $message = sprintf('Failed login attempt from %s user %s',
  130. $_SERVER['REMOTE_ADDR'],
  131. $username);
  133. if (strlen($reason) > 0)
  134. $message .= sprintf(' (%s)', $reason);
  136. $logger = Zend_Registry::get('logger');
  137. $logger->warn($message);
  138. }
  140. public function fetchPassword()
  141. {
  142. if (!$this->isSaved())
  143. return false;
  145. // generate new password properties
  146. $this->_newPassword = Text_Password::create(8);
  147. $this->profile->new_password = md5($this->_newPassword);
  148. $this->profile->new_password_ts = time();
  149. $this->profile->new_password_key = md5(uniqid() .
  150. $this->getId() .
  151. $this->_newPassword);
  153. // save new password to profile and send e-mail
  154. $this->profile->save();
  155. $this->sendEmail('user-fetch-password.tpl');
  157. return true;
  158. }
  160. public function confirmNewPassword($key)
  161. {
  162. // check that valid password reset data is set
  163. if (!isset($this->profile->new_password)
  164. || !isset($this->profile->new_password_ts)
  165. || !isset($this->profile->new_password_key)) {
  167. return false;
  168. }
  170. // check if the password is being confirm within a day
  171. if (time() - $this->profile->new_password_ts > 86400)
  172. return false;
  174. // check that the key is correct
  175. if ($this->profile->new_password_key != $key)
  176. return false;
  178. // everything is valid, now update the account to use the new password
  180. // bypass the local setter as new_password is already an md5
  181. parent::__set('password', $this->profile->new_password);
  183. unset($this->profile->new_password);
  184. unset($this->profile->new_password_ts);
  185. unset($this->profile->new_password_key);
  187. // finally, save the updated user record and the updated profile
  188. return $this->save();
  189. }
  191. public function usernameExists($username)
  192. {
  193. $query = sprintf('select count(*) as num from %s where username = ?',
  194. $this->_table);
  196. $result = $this->_db->fetchOne($query, $username);
  198. return $result > 0;
  199. }
  201. static public function IsValidUsername($username)
  202. {
  203. $validator = new Zend_Validate_Alnum();
  204. return $validator->isValid($username);
  205. }
  207. public function __set($name, $value)
  208. {
  209. switch ($name) {
  210. case 'password':
  211. $value = md5($value);
  212. break;
  214. case 'user_type':
  215. if (!array_key_exists($value, self::$userTypes))
  216. $value = 'member';
  217. break;
  218. }
  220. return parent::__set($name, $value);
  221. }
  222. }
  223. ?>

人气教程排行

本站所有资源全部来源于网络，若本站发布的内容侵害到您的隐私或者利益，请联系我们删除！