com.laudandjolynn.mytv.MyTvData.java Source code

Introduction

Here is the source code for com.laudandjolynn.mytv.MyTvData.java

Source

/*******************************************************************************
 * Copyright 2015 htd0324@gmail.com
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package com.laudandjolynn.mytv;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.laudandjolynn.mytv.exception.MyTvException;
import com.laudandjolynn.mytv.utils.Constant;
import com.laudandjolynn.mytv.utils.FileUtils;

/**
 * @author: Laud
 * @email: htd0324@gmail.com
 * @date: 2015330 ?1:41:30
 * @copyright: www.laudandjolynn.com
 */
public class MyTvData {
    private final static Logger logger = LoggerFactory.getLogger(MyTvData.class);
    private boolean dataInited = false;
    private boolean dbInited = false;
    private boolean stationCrawlerInited = false;
    private boolean programCrawlerInited = false;

    public MyTvData() {
        this.loadData();
    }

    /**
     * ?
     */
    private void loadData() {
        logger.debug("load data from my tv data file: " + Constant.MY_TV_DATA_FILE_PATH);
        File file = new File(Constant.MY_TV_DATA_FILE_PATH);
        if (!file.exists()) {
            this.dbInited = false;
            this.dataInited = false;
            this.stationCrawlerInited = false;
            this.programCrawlerInited = false;
            return;
        }
        SAXReader reader = new SAXReader();
        try {
            Document xmlDoc = reader.read(new File(Constant.MY_TV_DATA_FILE_PATH));
            List<?> nodes = xmlDoc.selectNodes("//" + Constant.XML_TAG_DB);
            if (nodes != null && nodes.size() > 0) {
                this.dbInited = Boolean.valueOf(((Element) nodes.get(0)).getText());
            }
            nodes = xmlDoc.selectNodes("//" + Constant.XML_TAG_DATA);
            if (nodes != null && nodes.size() > 0) {
                this.dataInited = Boolean.valueOf(((Element) nodes.get(0)).getText());
            }
            nodes = xmlDoc.selectNodes("//" + Constant.XML_TAG_STATION);
            if (nodes != null && nodes.size() > 0) {
                this.stationCrawlerInited = Boolean.valueOf(((Element) nodes.get(0)).getText());
            }

            nodes = xmlDoc.selectNodes("//" + Constant.XML_TAG_PROGRAM);
            if (nodes != null && nodes.size() > 0) {
                this.programCrawlerInited = Boolean.valueOf(((Element) nodes.get(0)).getText());
            }
        } catch (DocumentException e) {
            logger.debug("can't parse xml file.  -- " + Constant.MY_TV_DATA_FILE_PATH);
            this.dbInited = false;
            this.dataInited = false;
            this.stationCrawlerInited = false;
            this.programCrawlerInited = false;
            file.deleteOnExit();
        }
    }

    public void writeData(String parent, String tag, String value) {
        logger.debug("write data to my tv data file: " + Constant.MY_TV_DATA_FILE_PATH);
        File file = new File(Constant.MY_TV_DATA_FILE_PATH);
        if (!file.exists()) {
            Document doc = DocumentHelper.createDocument();
            doc.addElement(Constant.APP_NAME);
            try {
                FileUtils.writeWithNIO(doc.asXML().getBytes(), Constant.MY_TV_DATA_FILE_PATH);
            } catch (IOException e) {
                throw new MyTvException(
                        "error occur while write data to file. -- " + Constant.MY_TV_DATA_FILE_PATH);
            }
        }
        SAXReader reader = new SAXReader();
        try {
            Document xmlDoc = reader.read(file);
            Element parentElement = xmlDoc.getRootElement();
            if (parent != null) {
                List<?> nodes = xmlDoc.selectNodes("//" + parent);
                if (nodes != null && nodes.size() > 0) {
                    parentElement = (Element) nodes.get(0);
                }
            }
            parentElement.addElement(tag).setText(value);
            try {
                XMLWriter writer = new XMLWriter(new FileWriter(file));
                writer.write(xmlDoc);
                writer.close();
            } catch (IOException e) {
                throw new MyTvException(
                        "error occur while write data to file. -- " + Constant.MY_TV_DATA_FILE_PATH);
            }
        } catch (DocumentException e) {
            String msg = "can't parse xml file. -- " + Constant.MY_TV_DATA_FILE_PATH;
            throw new MyTvException(msg);
        }
    }

    /**
     * db???
     * 
     * @return
     */
    public boolean isDbInited() {
        return this.dbInited;
    }

    /**
     * ????
     * 
     * @return
     */
    public boolean isDataInited() {
        return dataInited;
    }

    /**
     * ????
     * @return
     */
    public boolean isProgramCrawlerInited() {
        return programCrawlerInited;
    }

    /**
     * ?????
     * @return
     */
    public boolean isStationCrawlerInited() {
        return stationCrawlerInited;
    }
}