Example #1
0
        MeasureFileHelper ComputeOneFile(Tuple <string, bool, Guid, bool> record, ParallelLoopState state, MeasureFileHelper helper)
        {
            var fullPath = record.Item1;
            var isImage  = record.Item2;
            var fileId   = record.Item3;
            var isNew    = record.Item4;

            var ex = helper.ProcessFile(fullPath, isImage, out var imageHash, out var sha1Hash, out var fileSize, out var fileState);

            if (ex != null)
            {
                OutputOneFileFailed(fullPath, ex.FullyQualifiedErrorId + " - " + ex.Exception.Message);
                exceptions.Add(ex);
            }
            else
            {
                OutputOneFileFinished(fullPath);
            }
            toWrite.Add(new Tuple <Guid, byte[], byte[], int, FileState, bool>(fileId, imageHash?.Coefficients, sha1Hash, fileSize, fileState, isNew));

            return(helper);
        }
        protected override void ProcessRecord()
        {
            if (File == null)
            {
                throw new ArgumentNullException(nameof(File));
            }

            bool isRecomputing;

            if (File.FileState == FileState.New)
            {
                isRecomputing = false;
            }
            else
            {
                if (Recompute.IsPresent)
                {
                    isRecomputing = true;
                }
                else
                {
                    ThrowTerminatingError(new ErrorRecord(new InvalidOperationException("State of this file is not set to New and -" + nameof(Recompute) + " is not set."), "ImageStore Measure File", ErrorCategory.InvalidOperation, File.Id));
                    isRecomputing = true; //Place here for avoiding compiling error. Won't be executed.
                }
            }

            var extension = ExtensionHelper.GetExtensionName(File.ExtensionId, out var extensionIsImage, out var extensionIgnored);

            if (extension == null)
            {
                throw new ArgumentException("Extension cannot be found.", nameof(File));
            }

            if (extensionIgnored)
            {
                throw new ArgumentException("Extension of this file is set to ignored.", nameof(File));
            }

            var folder = FolderHelper.GetFolderPath(File.FolderId, out _);

            if (folder == null)
            {
                throw new ArgumentException("Folder cannot be found.", nameof(File));
            }
            if (!folder.EndsWith(DirectorySeparatorString.Value))
            {
                folder += DirectorySeparatorString.Value;
            }

            Digest imageHash;

            byte[]    sha1Hash;
            int       fileSize;
            FileState fileState;

            using (var computer = new MeasureFileHelper())
            {
                var fullFilePath = FileHelper.GetFullFilePath(folder, File.Path, File.FileName, extension);
                var ex           = computer.ProcessFile(fullFilePath, extensionIsImage, out imageHash, out sha1Hash, out fileSize, out fileState);
                if (ex != null)
                {
                    WriteError(ex);
                }
            }

            var newFile = new ImageStoreFile(File.Id, File.FolderId, File.Path, File.FileName, File.ExtensionId)
            {
                ImageHashDigest        = imageHash,
                Sha1Hash               = sha1Hash,
                FileSize               = fileSize,
                FileState              = fileState,
                ImageComparedThreshold = 0
            };

            if (UpdateFileCmdlet.UpdateRecord(newFile) <= 0)
            {
                WriteError(new ErrorRecord(new InvalidOperationException("Cannot update this file."), "ImageStore - Measuring", ErrorCategory.WriteError, null));
            }

            if (isRecomputing)
            {
                var connection = DatabaseConnection.Current;
                using (var commandToDeleteSame = new SqlCommand("Delete from [SameFile] Where [FileId]=@Id"))
                    using (var commandToDeleteSimilar = new SqlCommand("Delete from [SimilarFile] Where [File1Id]=@Id or [File2Id]=@Id"))
                    {
                        commandToDeleteSame.Connection = connection;
                        commandToDeleteSame.Parameters.Add(new SqlParameter("@Id", System.Data.SqlDbType.UniqueIdentifier)
                        {
                            Value = File.Id
                        });
                        commandToDeleteSame.ExecuteNonQuery();

                        commandToDeleteSimilar.Connection = connection;
                        commandToDeleteSimilar.Parameters.Add(new SqlParameter("@Id", System.Data.SqlDbType.UniqueIdentifier)
                        {
                            Value = File.Id
                        });
                        commandToDeleteSimilar.ExecuteNonQuery();
                    }
            }
            WriteObject(newFile);
        }
Example #3
0
 void DisposeHelper(MeasureFileHelper helper) => helper.Dispose();