howardjones/network-weathermap

<?php
// Sample Pluggable datasource for PHP Weathermap 0.9
// - read a pair of values from a database, and return it

// TARGET dbplug:databasename:username:pass:hostkey

namespace Weathermap\Plugins\Datasources;

use Weathermap\Core\MapUtility;
use Weathermap\Core\Map;
use Weathermap\Core\MapDataItem;
use PDO;
use PDOException;

/**
 * Example of collecting data from a SQL database
 *
 * @package Weathermap\Plugins\Datasources
 */
class DatabaseSample extends Base
{

    public function __construct()
    {
        parent::__construct();

        $this->regexpsHandled = array(
            '/^dbplug:([^:]+)$/'
        );
        $this->name = "dbsample";
    }

    public function init(&$map)
    {
        if (!function_exists("mysql_connect")) {
            return false;
        }

        return true;
    }

    /**
     * @param string $targetString
     * @param Map $map
     * @param MapDataItem $mapItem
     * @return array
     */
    public function readData($targetString, &$map, &$mapItem)
    {
        $pdo = null;
        $this->data[IN] = null;
        $this->data[OUT] = null;


        if (preg_match('/^dbplug:([^:]+)$/', $targetString, $matches)) {
            $databaseUser = $map->getHint('dbplug_dbuser');
            $databasePassword = $map->getHint('dbplug_dbpass');
            $databaseName = $map->getHint('dbplug_dbname');
            $databaseHostname = $map->getHint('dbplug_dbhost');


            try {
                # MySQL with PDO_MYSQL
                $pdo = new PDO("mysql:host=$databaseHostname;dbname=$databaseName", $databaseUser, $databasePassword);
                $pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
            } catch (PDOException $e) {
                MapUtility::warn($e->getMessage());
            }

            if ($pdo) {
                $statement = $pdo->prepare("SELECT infield, outfield FROM tablename WHERE host=? LIMIT 1");
                $result = $statement->execute(array($matches[1]));

                if (!$result) {
                    MapUtility::warn("dbsample ReadData: Invalid query: " . $pdo->errorCode() . "\n");
                } else {
                    $row = $statement->fetch(PDO::FETCH_ASSOC);
                    $this->data[IN] = $row['infield'];
                    $this->data[OUT] = $row['outfield'];
                }

                $this->dataTime = time();
            }
        }

        return $this->returnData();
    }
}

// vim:ts=4:sw=4: