<?php
// PukiWiki - Yet another WikiWikiWeb clone
-// $Id: counter.inc.php,v 1.19 2007/02/04 11:14:44 henoheno Exp $
-// Copyright (C)
-// 2002-2005, 2007 PukiWiki Developers Team
+// counter.inc.php
+// Copyright
+// 2002-2019 PukiWiki Development Team
// 2002 Y.MASUI GPL2 http://masui.net/pukiwiki/ masui@masui.net
// License: GPL2
//
// Counter file's suffix
define('PLUGIN_COUNTER_SUFFIX', '.count');
+// Use Database (1) or not (0)
+define('PLUGIN_COUNTER_USE_DB', 0);
+// Database Connection setting
+define('PLUGIN_COUNTER_DB_CONNECT_STRING', 'sqlite:counter/counter.db');
+define('PLUGIN_COUNTER_DB_USERNAME', '');
+define('PLUGIN_COUNTER_DB_PASSWORD', '');
+global $plugin_counter_db_options;
+$plugin_counter_db_options = null;
+// For MySQL
+// $plugin_counter_db_options = array(PDO::MYSQL_ATTR_INIT_COMMAND =>
+// "SET NAMES utf8mb4 COLLATE utf8mb4_bin");
+
+define('PLUGIN_COUNTER_DB_TABLE_NAME_PREFIX', '');
+
+if (PLUGIN_COUNTER_USE_DB) {
+ ini_set('default_socket_timeout', 2);
+}
// Report one
function plugin_counter_inline()
// Return a summary
function plugin_counter_get_count($page)
{
- global $vars;
+ global $vars, $plugin_counter_db_options;
static $counters = array();
static $default;
+ $page_counter_t = PLUGIN_COUNTER_DB_TABLE_NAME_PREFIX . 'page_counter';
if (! isset($default))
$default = array(
// Set default
$counters[$page] = $default;
$modify = FALSE;
+ $c = & $counters[$page];
- // Open
- $file = COUNTER_DIR . encode($page) . PLUGIN_COUNTER_SUFFIX;
- pkwk_touch_file($file);
- $fp = fopen($file, 'r+')
- or die('counter.inc.php: Cannot open COUNTER_DIR/' . basename($file));
- set_file_buffer($fp, 0);
- flock($fp, LOCK_EX);
- rewind($fp);
-
- // Read
- foreach (array_keys($default) as $key) {
- // Update
- $counters[$page][$key] = rtrim(fgets($fp, 256));
- if (feof($fp)) break;
+ if (PLUGIN_COUNTER_USE_DB) {
+ if (SOURCE_ENCODING !== 'UTF-8') {
+ die('counter.inc.php: Database counter is only available in UTF-8 mode');
+ }
+ $is_new_page = false;
+ try {
+ $pdo = new PDO(PLUGIN_COUNTER_DB_CONNECT_STRING,
+ PLUGIN_COUNTER_DB_USERNAME, PLUGIN_COUNTER_DB_PASSWORD,
+ $plugin_counter_db_options);
+ $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+ $pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
+ $pdo->setAttribute(PDO::ATTR_TIMEOUT, 5);
+ $stmt = $pdo->prepare(
+"SELECT total, update_date,
+ today_viewcount, yesterday_viewcount, remote_addr
+ FROM $page_counter_t
+ WHERE page_name = ?"
+ );
+ $stmt->execute(array($page));
+ $r = $stmt->fetch();
+ if ($r === false) {
+ $is_new_page = true;
+ } else {
+ $c['ip'] = $r['remote_addr'];
+ $c['date'] = $r['update_date'];
+ $c['yesterday'] = intval($r['yesterday_viewcount']);
+ $c['today'] = intval($r['today_viewcount']);
+ $c['total'] = intval($r['total']);
+ $stmt->closeCursor();
+ }
+ } catch (Exception $e) {
+ // Error occurred
+ $db_error = '(DBError)';
+ return array(
+ 'total' => $db_error,
+ 'date' => $db_error,
+ 'today' => $db_error,
+ 'yesterday' => $db_error,
+ 'ip' => $db_error);
+ }
+ } else {
+ // Open
+ $file = COUNTER_DIR . encode($page) . PLUGIN_COUNTER_SUFFIX;
+ pkwk_touch_file($file);
+ $fp = fopen($file, 'r+')
+ or die('counter.inc.php: Cannot open COUNTER_DIR/' . basename($file));
+ set_file_buffer($fp, 0);
+ flock($fp, LOCK_EX);
+ rewind($fp);
+
+ // Read
+ foreach (array_keys($default) as $key) {
+ // Update
+ $c[$key] = rtrim(fgets($fp, 256));
+ if (feof($fp)) break;
+ }
}
// Anothoer day?
- if ($counters[$page]['date'] != $default['date']) {
+ $remote_addr = $_SERVER['REMOTE_ADDR'];
+ $count_up = FALSE;
+ if ($c['date'] != $default['date']) {
$modify = TRUE;
- $is_yesterday = ($counters[$page]['date'] == get_date('Y/m/d', UTIME - 24 * 60 * 60));
- $counters[$page]['ip'] = $_SERVER['REMOTE_ADDR'];
- $counters[$page]['date'] = $default['date'];
- $counters[$page]['yesterday'] = $is_yesterday ? $counters[$page]['today'] : 0;
- $counters[$page]['today'] = 1;
- $counters[$page]['total']++;
- } else if ($counters[$page]['ip'] != $_SERVER['REMOTE_ADDR']) {
+ $is_yesterday = ($c['date'] == get_date('Y/m/d', UTIME - 24 * 60 * 60));
+ $c[$page]['ip'] = $remote_addr;
+ $c['date'] = $default['date'];
+ $c['yesterday'] = $is_yesterday ? $c['today'] : 0;
+ $c['today'] = 1;
+ $c['total']++;
+ $count_up = TRUE;
+ } else if ($c['ip'] != $remote_addr) {
// Not the same host
$modify = TRUE;
- $counters[$page]['ip'] = $_SERVER['REMOTE_ADDR'];
- $counters[$page]['today']++;
- $counters[$page]['total']++;
+ $c['ip'] = $remote_addr;
+ $c['today']++;
+ $c['total']++;
+ $count_up = TRUE;
}
- // Modify
- if ($modify && $vars['cmd'] == 'read') {
- rewind($fp);
- ftruncate($fp, 0);
- foreach (array_keys($default) as $key)
- fputs($fp, $counters[$page][$key] . "\n");
+ if (PLUGIN_COUNTER_USE_DB) {
+ if ($modify && $vars['cmd'] == 'read') {
+ try {
+ if ($is_new_page) {
+ // Insert
+ $add_stmt = $pdo->prepare(
+"INSERT INTO $page_counter_t
+ (page_name, total, update_date, today_viewcount,
+ yesterday_viewcount, remote_addr)
+ VALUES (?, ?, ?, ?, ?, ?)"
+ );
+ $r_add = $add_stmt->execute(array($page, $c['total'],
+ $c['date'], $c['today'], $c['yesterday'], $c['ip']));
+ } else if ($count_up) {
+ // Update on counting up 'total'
+ $upd_stmt = $pdo->prepare(
+"UPDATE $page_counter_t
+ SET total = total + 1,
+ update_date = ?,
+ today_viewcount = ?,
+ yesterday_viewcount = ?,
+ remote_addr = ?
+ WHERE page_name = ?"
+ );
+ $r_upd = $upd_stmt->execute(array($c['date'],
+ $c['today'], $c['yesterday'], $c['ip'], $page));
+ }
+ } catch (PDOException $e) {
+ foreach (array_keys($c) as $key) {
+ $c[$key] .= '(DBError)';
+ }
+ }
+ }
+ } else {
+ // Modify
+ if ($modify && $vars['cmd'] == 'read') {
+ rewind($fp);
+ ftruncate($fp, 0);
+ foreach (array_keys($default) as $key)
+ fputs($fp, $c[$key] . "\n");
+ }
+ // Close
+ flock($fp, LOCK_UN);
+ fclose($fp);
}
+ return $c;
+}
+
+function plugin_counter_get_popular_list($today, $except, $max) {
+ if (PLUGIN_COUNTER_USE_DB === 0) {
+ return plugin_counter_get_popular_list_file($today, $except, $max);
+ } else {
+ return plugin_counter_get_popular_list_db($today, $except, $max);
+ }
+}
+
+function plugin_counter_get_popular_list_file($today, $except, $max) {
+ global $whatsnew;
+ $counters = array();
+ $except_quote = str_replace('#', '\#', $except);
+ foreach (get_existpages(COUNTER_DIR, '.count') as $file=>$page) {
+ if (($except != '' && preg_match("#$except_quote#", $page)) ||
+ $page == $whatsnew || check_non_list($page) ||
+ ! is_page($page))
+ continue;
+
+ $array = file(COUNTER_DIR . $file);
+ $count = rtrim($array[0]);
+ $date = rtrim($array[1]);
+ $today_count = rtrim($array[2]);
- // Close
- flock($fp, LOCK_UN);
- fclose($fp);
+ if ($today) {
+ // $pageが数値に見える(たとえばencode('BBS')=424253)とき、
+ // array_splice()によってキー値が変更されてしまうのを防ぐ
+ // ため、キーに '_' を連結する
+ if ($today == $date) $counters['_' . $page] = $today_count;
+ } else {
+ $counters['_' . $page] = $count;
+ }
+ }
+
+ asort($counters, SORT_NUMERIC);
+
+ // BugTrack2/106: Only variables can be passed by reference from PHP 5.0.5
+ $counters = array_reverse($counters, TRUE); // with array_splice()
+ $counters = array_splice($counters, 0, $max);
+ return $counters;
+}
+
+function plugin_counter_get_popular_list_db($today, $except, $max) {
+ global $whatsnew;
+ $page_counter_t = PLUGIN_COUNTER_DB_TABLE_NAME_PREFIX . 'page_counter';
+ if ($today) {
+ $order_by_c = 'today_viewcount';
+ } else {
+ $order_by_c = 'total';
+ }
+ $counters = array();
+ try {
+ $pdo = new PDO(PLUGIN_COUNTER_DB_CONNECT_STRING,
+ PLUGIN_COUNTER_DB_USERNAME, PLUGIN_COUNTER_DB_PASSWORD,
+ $plugin_counter_db_options);
+ $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+ $pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
+ $pdo->setAttribute(PDO::ATTR_TIMEOUT, 5);
+ if ($today) {
+ $stmt = $pdo->prepare(
+"SELECT page_name, total, update_date,
+ today_viewcount, yesterday_viewcount
+ FROM $page_counter_t
+ WHERE update_date = ?
+ ORDER BY $order_by_c DESC
+ LIMIT ?"
+ );
+ } else {
+ $stmt = $pdo->prepare(
+"SELECT page_name, total, update_date,
+ today_viewcount, yesterday_viewcount
+ FROM $page_counter_t
+ ORDER BY $order_by_c DESC
+ LIMIT ?"
+ );
+ }
+ $except_quote = str_replace('#', '\#', $except);
+ $limit = $max + 100;
+ if ($today) {
+ $stmt->execute(array($today, $limit));
+ } else {
+ $stmt->execute(array($limit));
+ }
+ foreach ($stmt as $r) {
+ $page = $r['page_name'];
+ if (($except != '' && preg_match("#$except_quote#", $page)) ||
+ $page == $whatsnew || check_non_list($page) ||
+ ! is_page($page)) {
+ continue;
+ }
+ if ($today) {
+ $counters['_' . $page] = $r['today_viewcount'];
+ } else {
+ $counters['_' . $page] = $r['total'];
+ }
+ }
+ $stmt->closeCursor();
+ return array_splice($counters, 0, $max);
+ } catch (Exception $e) {
+ die('counter.inc.php: Error occurred on getting pupular pages');
+ }
+}
+
+function plugin_counter_page_rename($pages) {
+ global $plugin_counter_db_options;
+ if (PLUGIN_COUNTER_USE_DB !== 0) {
+ $page_counter_t = PLUGIN_COUNTER_DB_TABLE_NAME_PREFIX . 'page_counter';
+ $pdo = new PDO(PLUGIN_COUNTER_DB_CONNECT_STRING,
+ PLUGIN_COUNTER_DB_USERNAME, PLUGIN_COUNTER_DB_PASSWORD,
+ $plugin_counter_db_options);
+ $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+ $pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
+ $stmt_delete = $pdo->prepare(
+"DELETE FROM $page_counter_t
+ WHERE page_name = ?"
+ );
+ $stmt_rename = $pdo->prepare(
+"UPDATE $page_counter_t
+ SET page_name = ?
+ WHERE page_name = ?"
+ );
+ foreach ($pages as $old_name=>$new_name) {
+ $stmt_delete->execute(array($new_name));
+ $stmt_rename->execute(array($new_name, $old_name));
+ }
+ }
+}
- return $counters[$page];
+/**
+ * php -r "include 'plugin/counter.inc.php'; plugin_counter_tool_setup_table();"
+ */
+function plugin_counter_tool_setup_table() {
+ global $plugin_counter_db_options;
+ $page_counter_t = PLUGIN_COUNTER_DB_TABLE_NAME_PREFIX . 'page_counter';
+ $pdo = new PDO(PLUGIN_COUNTER_DB_CONNECT_STRING,
+ PLUGIN_COUNTER_DB_USERNAME, PLUGIN_COUNTER_DB_PASSWORD,
+ $plugin_counter_db_options);
+ $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
+ $pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
+ $r = $pdo->exec(
+"CREATE TABLE $page_counter_t (
+ page_name VARCHAR(190) PRIMARY KEY,
+ total INTEGER NOT NULL,
+ update_date VARCHAR(20) NOT NULL,
+ today_viewcount INTEGER NOT NULL,
+ yesterday_viewcount INTEGER NOT NULL,
+ remote_addr VARCHAR(100)
+ )"
+ );
+ echo "OK\n";
}
OSDN Git Service
