PHP分割WordPress导出的XML文件

分手后的思念是犯贱 2022-05-27 13:59 219阅读 0赞

//定义生成的XML的大小
    private const XML_SIZE = 500 * 1000;
    
    //定义需要分割的XML文件
    private const XML_SOURCE = DIR_ROOT . 'run/media.xml';
    
    //定义输出文件的前缀
    private const XML_TARGET = DIR_ROOT . 'run/media-';
    
    /**
     * 分割WordPress的XML文件
     */
    public function xml()
    {
        //读取XML源文件
        $from = file_get_contents(self::XML_SOURCE);
    
        //取出 item 前的头部
        $header = mid($from, null, '<item>');
    
        //去除头部之后 的剩余部分
        $from = mid($from, $header);
    
        //尾部
        $end = "</channel></rss>";
    
        $i = 0;
        $output = $header;
    
        //逐个ITEM 处理
        foreach ($this->xmlItem($from) as $item) {
            //试图增加一项
            $new = $output . '<item>' . $item . '</item>';
    
            //如果长度超出限制, 16是尾部的长度
            if (strlen($new) > self::XML_SIZE - 16) {
                $i++;
    
                //输出当前内容
                file_put_contents(self::XML_TARGET . $i . '.xml', $output . $end);
    
                //清空输出缓冲区
                $output = $header;
            }
    
            //当前项输出到缓冲区
            $output .= '<item>' . $item . '</item>';
        }
    
        //生成最后一个输出文件
        $i++;
        file_put_contents(DIR_ROOT . 'run/media-' . $i . '.xml', $output . $end);
    }
    
    /**
     * 遍历获取每个ITEM
     * @param $all string 整个XML文件
     * @return Iterator 遍历器
     */
    private function xmlItem(string $all): Iterator
    {
        //Item计数
        $i = 1;
    
        //当前指针
        $p = 0;
    
        //总长度
        $length = strlen($all);
    
        //整个XML内部搜索
        while (true) {
            //查找起点
            $f = strpos($all, '<item>', $p);
            if ($f === false) {
                //没有了.
                return;
            }
    
            //计算终点
            $t = strpos($all, '</item>', $p + 6);
    
            //取出ITEM内容
            $item = substr($all, $f + 6, $t - $f - 6);
    
            //指针后移
            $p = $t + 7;
    
            //输出进度指示
            echo "ITEMS:" . ($i++) . " $p/$length\r\n";
    
            //返回一个ITEM项
            yield $item;
        }
    }