net.sf.xsd2pgschema.implement.ChkSumStatThrd Maven / Gradle / Ivy

Go to download
/*
    xsd2pgschema - Database replication tool based on XML Schema
    Copyright 2018 Masashi Yokochi

    https://sourceforge.net/projects/xsd2pgschema/

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 net.sf.xsd2pgschema.implement;

import java.io.IOException;
import java.nio.file.Path;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.concurrent.LinkedBlockingQueue;

import net.sf.xsd2pgschema.option.PgSchemaOption;
import net.sf.xsd2pgschema.option.XmlFileFilter;
import net.sf.xsd2pgschema.xmlutil.XmlParser;

/**
 * Thread function for chksumstat.
 *
 * @author yokochi
 */
public class ChkSumStatThrd implements Runnable {

	/** The PostgreSQL data model option. */
	private PgSchemaOption option;

	/** The XML file filter. */
	private XmlFileFilter xml_file_filter;

	/** The XML file queue. */
	private LinkedBlockingQueue xml_file_queue;

	/** The set of new document id while synchronization. */
	private HashSet sync_new_doc_rows;

	/** The set of updating document id while synchronization. */
	private HashSet sync_up_doc_rows;

	/** The set of deleting document id while synchronization (key=document id, value=check sum file path). */
	private HashMap sync_del_doc_rows;

	/** The instance of message digest for check sum. */
	private MessageDigest md_chk_sum;

	/**
	 * Instance of ChkSumStatThrd.
	 *
	 * @param xml_file_filter XML file filter
	 * @param xml_file_queue XML file queue
	 * @param sync_new_doc_rows set of new document id while synchronization
	 * @param sync_up_doc_rows set of updating document id while synchronization
	 * @param sync_del_doc_rows set of deleting document id while synchronization
	 * @param option PostgreSQL data model option
	 * @throws NoSuchAlgorithmException the no such algorithm exception
	 */
	public ChkSumStatThrd(final XmlFileFilter xml_file_filter, final LinkedBlockingQueue xml_file_queue, HashSet sync_new_doc_rows, HashSet sync_up_doc_rows, HashMap sync_del_doc_rows, PgSchemaOption option) throws NoSuchAlgorithmException {

		this.xml_file_filter = xml_file_filter;
		this.xml_file_queue = xml_file_queue;

		this.sync_new_doc_rows = sync_new_doc_rows;
		this.sync_up_doc_rows = sync_up_doc_rows;
		this.sync_del_doc_rows = sync_del_doc_rows;

		this.option = option;

		md_chk_sum = MessageDigest.getInstance(option.check_sum_algorithm);

	}

	/* (non-Javadoc)
	 * @see java.lang.Runnable#run()
	 */
	@Override
	public void run() {

		Path xml_file_path, has_path;
		XmlParser xml_parser;
		String document_id;

		while ((xml_file_path = xml_file_queue.poll()) != null) {

			xml_parser = new XmlParser(xml_file_path, xml_file_filter);

			document_id = xml_parser.document_id;

			synchronized (sync_del_doc_rows) {
				has_path = sync_del_doc_rows.remove(document_id);
			}

			if (has_path != null) {

				try {

					if (xml_parser.identify(option, md_chk_sum))
						continue;

				} catch (IOException e) {
					e.printStackTrace();
				}

				synchronized (sync_up_doc_rows) {
					sync_up_doc_rows.add(document_id);
				}

			}

			else {

				if (!option.sync_dry_run) {

					try {

						xml_parser.identify(option, md_chk_sum);

					} catch (IOException e) {
						e.printStackTrace();
					}

				}

				synchronized (sync_new_doc_rows) {
					sync_new_doc_rows.add(document_id);
				}

			}

		}

	}

}