dnsmgr/app/lib/TaskRunner.php

<?php
namespace app\lib;

use app\lib\NewDb;
use app\lib\CheckUtils;
use app\lib\DnsHelper;
use app\lib\MsgNotice;

class TaskRunner
{
    private $conn;

    private function db()
    {
        if(!$this->conn){
            $this->conn = NewDb::connect();
        }
        return $this->conn;
    }

    private function closeDb()
    {
        if($this->conn){
            $this->conn->close();
        }
    }

    public function execute($row)
    {
        if($row['checktype'] == 2){
            $result = CheckUtils::curl($row['checkurl'], $row['timeout'], $row['main_value']);
        }else if($row['checktype'] == 1){
            $result = CheckUtils::tcp($row['main_value'], $row['tcpport'], $row['timeout']);
        }else{
            $result = CheckUtils::ping($row['main_value']);
        }

        $action = 0;
        if($result['status'] && $row['status']==1){
            if($row['cycle'] <= 1 || $row['errcount'] >= $row['cycle']){
                $this->db()->name('dmtask')->where('id', $row['id'])->update(['status'=>0, 'errcount'=>0, 'switchtime'=>time()]);
                $action = 2;
            }else{
                $this->db()->name('dmtask')->where('id', $row['id'])->inc('errcount')->update();
            }
        }elseif(!$result['status'] && $row['status']==0){
            if($row['cycle'] <= 1 || $row['errcount'] >= $row['cycle']){
                $this->db()->name('dmtask')->where('id', $row['id'])->update(['status'=>1, 'errcount'=>0, 'switchtime'=>time()]);
                $action = 1;
            }else{
                $this->db()->name('dmtask')->where('id', $row['id'])->inc('errcount')->update();
            }
        }elseif($row['errcount'] > 0){
            $this->db()->name('dmtask')->where('id', $row['id'])->update(['errcount'=>0]);
        }

        if($action > 0){
            $drow = $this->db()->name('domain')->alias('A')->join('account B','A.aid = B.id')->where('A.id', $row['did'])->field('A.*,B.type,B.ak,B.sk,B.ext')->find();
            if(!$drow){
                echo '域名不存在（ID：'.$row['did'].'）'."\n";
                $this->closeDb();
                return;
            }
            $row['domain'] = $row['rr'] . '.' . $drow['name'];
        }
        if($action == 1){
            if($row['type'] == 2){
                $dns = DnsHelper::getModel2($drow);
                $recordinfo = json_decode($row['recordinfo'], true);
                $res = $dns->updateDomainRecord($row['recordid'], $row['rr'], getDnsType($row['backup_value']), $row['backup_value'], $recordinfo['Line'], $recordinfo['TTL']);
                if(!$res){
                    $this->db()->name('log')->insert(['uid' => 0, 'domain' => $drow['name'], 'action' => '修改解析失败', 'data' => $dns->getError(), 'addtime' => date("Y-m-d H:i:s")]);
                }
            }elseif($row['type'] == 1){
                $dns = DnsHelper::getModel2($drow);
                $res = $dns->setDomainRecordStatus($row['recordid'], '0');
                if(!$res){
                    $this->db()->name('log')->insert(['uid' => 0, 'domain' => $drow['name'], 'action' => '暂停解析失败', 'data' => $dns->getError(), 'addtime' => date("Y-m-d H:i:s")]);
                }
            }
        }elseif($action == 2){
            if($row['type'] == 2){
                $dns = DnsHelper::getModel2($drow);
                $recordinfo = json_decode($row['recordinfo'], true);
                $res = $dns->updateDomainRecord($row['recordid'], $row['rr'], getDnsType($row['main_value']), $row['main_value'], $recordinfo['Line'], $recordinfo['TTL']);
                if(!$res){
                    $this->db()->name('log')->insert(['uid' => 0, 'domain' => $drow['name'], 'action' => '修改解析失败', 'data' => $dns->getError(), 'addtime' => date("Y-m-d H:i:s")]);
                }
            }elseif($row['type'] == 1){
                $dns = DnsHelper::getModel2($drow);
                $res = $dns->setDomainRecordStatus($row['recordid'], '1');
                if(!$res){
                    $this->db()->name('log')->insert(['uid' => 0, 'domain' => $drow['name'], 'action' => '启用解析失败', 'data' => $dns->getError(), 'addtime' => date("Y-m-d H:i:s")]);
                }
            }
        }else{
            $this->closeDb();
            return;
        }

        $this->db()->name('dmlog')->insert([
            'taskid' => $row['id'],
            'action' => $action,
            'errmsg' => $result['status'] ? null : $result['errmsg'],
            'date' => date('Y-m-d H:i:s')
        ]);
        $this->closeDb();

        MsgNotice::send($action, $row, $result);
    }
}