2 # Cache for article titles and ids linked from one source
4 # These are used in incrementalSetup()
5 define ('LINKCACHE_GOOD', 0);
6 define ('LINKCACHE_BAD', 1);
7 define ('LINKCACHE_IMAGE', 2);
11 /* private */ var $mGoodLinks, $mBadLinks, $mActive;
12 /* private */ var $mImageLinks;
13 /* private */ var $mPreFilled, $mOldGoodLinks, $mOldBadLinks;
17 $this->mActive
= true;
18 $this->mPreFilled
= false;
19 $this->mGoodLinks
= array();
20 $this->mBadLinks
= array();
21 $this->mImageLinks
= array();
22 $this->mOldGoodLinks
= array();
23 $this->mOldBadLinks
= array();
26 function getGoodLinkID( $title )
28 if ( array_key_exists( $title, $this->mGoodLinks
) ) {
29 return $this->mGoodLinks
[$title];
35 function isBadLink( $title )
37 return in_array( $title, $this->mBadLinks
);
40 function addGoodLink( $id, $title )
42 if ( $this->mActive
) {
43 $this->mGoodLinks
[$title] = $id;
47 function addBadLink( $title )
49 if ( $this->mActive
&& ( ! $this->isBadLink( $title ) ) ) {
50 array_push( $this->mBadLinks
, $title );
54 function addImageLink( $title )
56 if ( $this->mActive
) { $this->mImageLinks
[$title] = 1; }
59 function clearBadLink( $title )
61 $index = array_search( $title, $this->mBadLinks
);
62 if ( isset( $index ) ) {
63 unset( $this->mBadLinks
[$index] );
65 global $wgMemc, $wgDBname;
66 $wgMemc->delete( "$wgDBname:lc:title:$title" );
69 function suspend() { $this->mActive
= false; }
70 function resume() { $this->mActive
= true; }
71 function getGoodLinks() { return $this->mGoodLinks
; }
72 function getBadLinks() { return $this->mBadLinks
; }
73 function getImageLinks() { return $this->mImageLinks
; }
75 function addLink( $title )
77 return $this->addLinkObj( Title
::newFromDBkey( $title ) );
80 function addLinkObj( &$nt )
82 $title = $nt->getDBkey();
83 if ( $this->isBadLink( $title ) ) { return 0; }
84 $id = $this->getGoodLinkID( $title );
85 if ( 0 != $id ) { return $id; }
87 global $wgMemc, $wgDBname;
88 $fname = "LinkCache::addLink-checkdatabase";
89 wfProfileIn( $fname );
91 $ns = $nt->getNamespace();
94 wfProfileOut( $fname );
98 $id = $wgMemc->get( $key = "$wgDBname:lc:title:$title" );
100 $sql = "SELECT HIGH_PRIORITY cur_id FROM cur WHERE cur_namespace=" .
101 "{$ns} AND cur_title='" . wfStrencode( $title ) . "'";
102 $res = wfQuery( $sql, DB_READ
, "LinkCache::addLink" );
104 if ( 0 == wfNumRows( $res ) ) {
107 $s = wfFetchObject( $res );
110 $wgMemc->add( $key, $id, time()+
3600 );
112 if ( 0 == $id ) { $this->addBadLink( $title ); }
113 else { $this->addGoodLink( $id, $title ); }
114 wfProfileOut( $fname );
118 function preFill( &$fromtitle )
120 $fname = "LinkCache::preFill";
121 wfProfileIn( $fname );
122 # Note -- $fromtitle is a Title *object*
123 $dbkeyfrom = wfStrencode( $fromtitle->getPrefixedDBKey() );
124 $sql = "SELECT HIGH_PRIORITY cur_id,cur_namespace,cur_title
126 WHERE cur_id=l_to AND l_from='{$dbkeyfrom}'";
127 $res = wfQuery( $sql, DB_READ
, "LinkCache::preFill" );
128 while( $s = wfFetchObject( $res ) ) {
129 $this->addGoodLink( $s->cur_id
,
130 Title
::makeName( $s->cur_namespace
, $s->cur_title
)
135 $id = $fromtitle->getArticleID();
138 $sql = "SELECT HIGH_PRIORITY bl_to
140 WHERE bl_from='{$id}'";
141 $res = wfQuery( $sql, DB_READ
, "LinkCache::preFill" );
142 while( $s = wfFetchObject( $res ) ) {
143 $this->addBadLink( $s->bl_to
);
146 $this->mOldBadLinks
= $this->mBadLinks
;
147 $this->mOldGoodLinks
= $this->mGoodLinks
;
148 $this->mPreFilled
= true;
150 wfProfileOut( $fname );
153 function getGoodAdditions()
155 return array_diff( $this->mGoodLinks
, $this->mOldGoodLinks
);
158 function getBadAdditions()
160 return array_values( array_diff( $this->mBadLinks
, $this->mOldBadLinks
) );
163 function getImageAdditions()
165 return array_diff_assoc( $this->mImageLinks
, $this->mOldImageLinks
);
168 function getGoodDeletions()
170 return array_diff( $this->mOldGoodLinks
, $this->mGoodLinks
);
173 function getBadDeletions()
175 return array_values( array_diff( $this->mOldBadLinks
, $this->mBadLinks
) );
178 function getImageDeletions()
180 return array_diff_assoc( $this->mOldImageLinks
, $this->mImageLinks
);
183 # Parameters: $which is one of the LINKCACHE_xxx constants, $del and $add are
184 # the incremental update arrays which will be filled. Returns whether or not it's
185 # worth doing the incremental version. For example, if [[List of mathematical topics]]
186 # was blanked, it would take a long, long time to do incrementally.
187 function incrementalSetup( $which, &$del, &$add )
189 if ( ! $this->mPreFilled
) {
195 $old =& $this->mOldGoodLinks
;
196 $cur =& $this->mGoodLinks
;
197 $del = $this->getGoodDeletions();
198 $add = $this->getGoodAdditions();
201 $old =& $this->mOldBadLinks
;
202 $cur =& $this->mBadLinks
;
203 $del = $this->getBadDeletions();
204 $add = $this->getBadAdditions();
206 default: # LINKCACHE_IMAGE
213 # Clears cache but leaves old preFill copies alone
216 $this->mGoodLinks
= array();
217 $this->mBadLinks
= array();
218 $this->mImageLinks
= array();