diff --git a/trunk/class/ShortURL.php b/trunk/class/ShortURL.php index d268c9a..390090d 100644 --- a/trunk/class/ShortURL.php +++ b/trunk/class/ShortURL.php @@ -16,12 +16,13 @@ class ShortURL extends XMLSQL{ if($this->pkAlreadyExists($shortName, 'url')){ return self::STATE_ALREADY_EXIST; }else{ - return $this->insert(array('url'=>$longUrl), rawurlencode($shortName))->into('url')->query(); + return $this->insert(array('url'=>$longUrl,'hit'=>'0'), rawurlencode($shortName))->into('url')->query(); } } public function findThisUrl($shortName){ if($this->pkAlreadyExists(rawurlencode($shortName), 'url')){ + $this->_incrementStatFor($shortName); return $this->select(array('url'))->from('url')->where(rawurlencode($shortName))->query(); }else{ return; @@ -32,4 +33,14 @@ class ShortURL extends XMLSQL{ return $this->select()->from('url')->query(); } + + /** + * Considering the table with $shortname already exist + */ + private function _incrementStatFor($shortName){ + $currentHit = $this->select(array('hit'))->from('url')->where(rawurlencode($shortName))->query(); + $currentHit = $currentHit[0]; + return $this->update('url')->set(array('hit'=>$currentHit+1))->where(rawurlencode($shortName))->query(); + } + } \ No newline at end of file