package org.openslx.imagemaster.db.mappers;

import java.sql.SQLException;
import java.util.List;
import java.util.concurrent.ArrayBlockingQueue;

import org.apache.log4j.Logger;
import org.openslx.filetransfer.FileRange;
import org.openslx.filetransfer.util.ChunkStatus;
import org.openslx.filetransfer.util.FileChunk;
import org.openslx.imagemaster.db.Database;
import org.openslx.imagemaster.db.MysqlConnection;
import org.openslx.imagemaster.db.MysqlStatement;

public class DbImageBlock
{

	private static final Logger LOGGER = Logger.getLogger( DbImageBlock.class );

	private static AsyncThread asyncBlockUpdate = null;

	private static synchronized void initAsyncThread()
	{
		if ( asyncBlockUpdate == null ) {
			asyncBlockUpdate = new AsyncThread();
			asyncBlockUpdate.start();
		}
	}

	public static void asyncUpdate( String imageVersionId, FileChunk chunk ) throws InterruptedException
	{
		initAsyncThread();
		asyncBlockUpdate.put( new ChunkUpdate( imageVersionId, chunk.range, chunk.getStatus() != ChunkStatus.COMPLETE ) );
	}

	private static class AsyncThread extends Thread
	{
		private final ArrayBlockingQueue<ChunkUpdate> queue = new ArrayBlockingQueue<>( 100 );

		public void put( ChunkUpdate chunk ) throws InterruptedException
		{
			queue.put( chunk );
		}

		@Override
		public void run()
		{
			try {
				while ( !interrupted() ) {
					ChunkUpdate chunk = queue.take();
					Thread.sleep( 100 );
					try ( MysqlConnection connection = Database.getConnection() ) {
						MysqlStatement stmt = connection.prepareStatement( "UPDATE imageblock SET ismissing = :ismissing"
								+ " WHERE imageversionid = :imageversionid AND startbyte = :startbyte AND blocksize = :blocksize" );
						do {
							stmt.setBoolean( "ismissing", chunk.isMissing );
							stmt.setString( "imageversionid", chunk.imageVersionId );
							stmt.setLong( "startbyte", chunk.range.startOffset );
							stmt.setInt( "blocksize", chunk.range.getLength() );
							stmt.executeUpdate();
							chunk = queue.poll();
						} while ( chunk != null );
						connection.commit();
					} catch ( SQLException e ) {
						LOGGER.error( "Query failed in DbImageBlock.AsyncThread.run()", e );
						continue;
					}
					Thread.sleep( 2000 );
				}
			} catch ( InterruptedException e ) {
				LOGGER.debug( "async thread interrupted" );
				interrupt();
			}
		}
	}

	private static class ChunkUpdate
	{
		public final String imageVersionId;
		public final FileRange range;
		public final boolean isMissing;

		public ChunkUpdate( String imageVersionId, FileRange range, boolean isMissing )
		{
			this.imageVersionId = imageVersionId;
			this.range = range;
			this.isMissing = isMissing;
		}
	}

	public static void insertChunkList( String imageVersionId, List<FileChunk> all, boolean missing ) throws SQLException
	{
		try ( MysqlConnection connection = Database.getConnection() ) {
			MysqlStatement stmt = connection.prepareStatement( "INSERT IGNORE INTO imageblock"
					+ " (imageversionid, startbyte, blocksize, blocksha1, ismissing) VALUES"
					+ " (:imageversionid, :startbyte, :blocksize, :blocksha1, :ismissing)" );
			stmt.setString( "imageversionid", imageVersionId );
			stmt.setBoolean( "ismissing", missing );
			for ( FileChunk chunk : all ) {
				stmt.setLong( "startbyte", chunk.range.startOffset );
				stmt.setInt( "blocksize", chunk.range.getLength() );
				stmt.setBinary( "blocksha1", chunk.getSha1Sum() );
				stmt.executeUpdate();
			}
			connection.commit();
		} catch ( SQLException e ) {
			LOGGER.error( "Query failed in DbImageBlock.insertChunkList()", e );
			throw e;
		}
	}

}