芝麻web文件管理V1.00

编辑当前文件:/home/mgatv524/www/midiatech/lib/XTR/StatsArchiveTask.php

<?php
/*
 * Spring Signage Ltd - http://www.springsignage.com
 * Copyright (C) 2016 Spring Signage Ltd
 * (StatsArchiveTask.php)
 */

namespace Xibo\XTR;
use Jenssegers\Date\Date;
use Xibo\Entity\User;
use Xibo\Exception\NotFoundException;
use Xibo\Exception\TaskRunException;
use Xibo\Factory\MediaFactory;
use Xibo\Factory\UserFactory;
use Xibo\Helper\Random;

/**
 * Class StatsArchiveTask
 * @package Xibo\XTR
 */
class StatsArchiveTask implements TaskInterface
{
    use TaskTrait;

/** @var  User */
    private $archiveOwner;

/** @var MediaFactory */
    private $mediaFactory;

/** @var UserFactory */
    private $userFactory;

/** @inheritdoc */
    public function setFactories($container)
    {
        $this->userFactory = $container->get('userFactory');
        $this->mediaFactory = $container->get('mediaFactory');
        return $this;
    }

/** @inheritdoc */
    public function run()
    {
        $this->archiveStats();
        $this->tidyStats();
    }

public function archiveStats()
    {
        $this->runMessage = '# ' . __('Stats Archive') . PHP_EOL . PHP_EOL;

if ($this->getOption('archiveStats', "Off") == "On") {
            // Archive tasks by week.
            $periodSizeInDays = $this->getOption('periodSizeInDays', 7);
            $maxPeriods = $this->getOption('maxPeriods', 4);
            $periodsToKeep = $this->getOption('periodsToKeep', 1);
            $this->setArchiveOwner();

// Get the earliest
            $earliestDate = $this->timeSeriesStore->getEarliestDate();

if (count($earliestDate) <= 0) {
                $this->runMessage = __('Nothing to archive');
                return;
            }

/** @var Date $earliestDate */
            $earliestDate = $this->date->parse($earliestDate['minDate'], 'U')->setTime(0, 0, 0);

// Take the earliest date and roll forward until the current time
            /** @var Date $now */
            $now = $this->date->parse()->subDay($periodSizeInDays * $periodsToKeep)->setTime(0, 0, 0);
            $i = 0;

while ($earliestDate < $now && $i < $maxPeriods) {
                $i++;

$this->log->debug('Running archive number ' . $i);

// Push forward
                $fromDt = $earliestDate->copy();
                $earliestDate->addDays($periodSizeInDays);

$this->exportStatsToLibrary($fromDt, $earliestDate);
                $this->store->commitIfNecessary();
            }

$this->runMessage .= ' - ' . __('Done') . PHP_EOL . PHP_EOL;
        } else {
            $this->runMessage .= ' - ' . __('Disabled') . PHP_EOL . PHP_EOL;
        }
    }

/**
     * Export stats to the library
     * @param Date $fromDt
     * @param Date $toDt
     */
    private function exportStatsToLibrary($fromDt, $toDt)
    {
        $this->runMessage .= ' - ' . $this->date->getLocalDate($fromDt) . ' / ' . $this->date->getLocalDate($toDt) . PHP_EOL;

$resultSet = $this->timeSeriesStore->getStats([
            'fromDt'=> $fromDt,
            'toDt'=> $toDt,
        ]);

// Create a temporary file for this
        $fileName = tempnam(sys_get_temp_dir(), 'stats');

$out = fopen($fileName, 'w');
        fputcsv($out, ['Stat Date', 'Type', 'FromDT', 'ToDT', 'Layout', 'Display', 'Media', 'Tag', 'Duration', 'Count', 'DisplayId', 'LayoutId', 'WidgetId', 'MediaId']);

while ($row = $resultSet->getNextRow() ) {

if ($this->timeSeriesStore->getEngine() == 'mongodb') {

$statDate = isset($row['statDate']) ? $this->date->parse($row['statDate']->toDateTime()->format('U'), 'U')->format('Y-m-d H:i:s') : null;
                $start = $this->date->parse($row['start']->toDateTime()->format('U'), 'U')->format('Y-m-d H:i:s');
                $end = $this->date->parse($row['end']->toDateTime()->format('U'), 'U')->format('Y-m-d H:i:s');
            } else {

$statDate = isset($row['statDate']) ?$this->date->parse($row['statDate'], 'U')->format('Y-m-d H:i:s') : null;
                $start = $this->date->parse($row['start'], 'U')->format('Y-m-d H:i:s');
                $end = $this->date->parse($row['end'], 'U')->format('Y-m-d H:i:s');
            }

// Read the columns
            fputcsv($out, [
                $statDate,
                $this->sanitizer->string($row['type']),
                $start,
                $end,
                isset($row['layout']) ? $this->sanitizer->string($row['layout']) :'',
                isset($row['display']) ? $this->sanitizer->string($row['display']) :'',
                isset($row['media']) ? $this->sanitizer->string($row['media']) :'',
                isset($row['tag']) ? $this->sanitizer->string($row['tag']) :'',
                $this->sanitizer->string($row['duration']),
                $this->sanitizer->string($row['count']),
                $this->sanitizer->int($row['displayId']),
                isset($row['layoutId']) ? $this->sanitizer->int($row['layoutId']) :'',
                isset($row['widgetId']) ? $this->sanitizer->int($row['widgetId']) :'',
                isset($row['mediaId']) ? $this->sanitizer->int($row['mediaId']) :'',
            ]);
        }

fclose($out);

// Create a ZIP file and add our temporary file
        $zipName = $this->config->getSetting('LIBRARY_LOCATION') . 'temp/stats.csv.zip';
        $zip = new \ZipArchive();
        $result = $zip->open($zipName, \ZipArchive::CREATE | \ZipArchive::OVERWRITE);
        if ($result !== true)
            throw new \InvalidArgumentException(__('Can\'t create ZIP. Error Code: %s', $result));

$zip->addFile($fileName, 'stats.csv');
        $zip->close();

// Remove the CSV file
        unlink($fileName);

// Upload to the library
        $media = $this->mediaFactory->create(
            __('Stats Export %s to %s - %s', $fromDt->format('Y-m-d'), $toDt->format('Y-m-d'), Random::generateString(5)),
            'stats.csv.zip',
            'genericfile',
            $this->archiveOwner->getId()
        );
        $media->save();

// Set max attempts to -1 so that we continue deleting until we've removed all of the stats that we've exported
        $options = [
            'maxAttempts' => -1,
            'statsDeleteSleep' => 1,
            'limit' => 1000
        ];

// Delete the stats, incrementally
        $this->timeSeriesStore->deleteStats($toDt, $fromDt, $options);
    }

/**
     * Set the archive owner
     * @throws TaskRunException
     */
    private function setArchiveOwner()
    {
        $archiveOwner = $this->getOption('archiveOwner', null);

if ($archiveOwner == null) {
            $admins = $this->userFactory->getSuperAdmins();

if (count($admins) <= 0)
                throw new TaskRunException(__('No super admins to use as the archive owner, please set one in the configuration.'));

$this->archiveOwner = $admins[0];

} else {
            try {
                $this->archiveOwner = $this->userFactory->getByName($archiveOwner);
            } catch (NotFoundException $e) {
                throw new TaskRunException(__('Archive Owner not found'));
            }
        }
    }

/**
     * Tidy Stats
     */
    private function tidyStats()
    {
        $this->runMessage .= '## ' . __('Tidy Stats') . PHP_EOL;

if ($this->config->getSetting('MAINTENANCE_STAT_MAXAGE') != 0) {

$maxage = Date::now()->subDays(intval($this->config->getSetting('MAINTENANCE_STAT_MAXAGE')));
            $maxAttempts = $this->getOption('statsDeleteMaxAttempts', 10);
            $statsDeleteSleep = $this->getOption('statsDeleteSleep', 3);

$options = [
                'maxAttempts' => $maxAttempts,
                'statsDeleteSleep' => $statsDeleteSleep,
                'limit' => 10000 // Note: for mongo we dont use $options['limit'] anymore
            ];

try {
                $result = $this->timeSeriesStore->deleteStats($maxage, null, $options);
                if ($result > 0) {
                    $this->runMessage .= ' - ' . __('Done.') . PHP_EOL . PHP_EOL;
                }
            } catch (\Exception $exception) {
                $this->runMessage .= ' - ' . __('Error.') . PHP_EOL . PHP_EOL;
            }
        } else {
            $this->runMessage .= ' - ' . __('Disabled') . PHP_EOL . PHP_EOL;
        }
    }
}