squidex/backend/src/Squidex.Domain.Apps.Entities.MongoDb/FullText/MongoIndexOutput.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System;
using System.IO;
using System.Linq;
using Lucene.Net.Store;
using MongoDB.Bson;
using MongoDB.Driver;
using MongoDB.Driver.GridFS;

namespace Squidex.Domain.Apps.Entities.MongoDb.FullText
{
    public sealed class MongoIndexOutput : IndexOutput
    {
        private readonly IndexOutput cacheOutput;
        private readonly MongoDirectory indexDirectory;
        private readonly string indexFileName;
        private bool isFlushed;
        private bool isWritten;

        public override long Length
        {
            get { return cacheOutput.Length; }
        }

        public override long Checksum
        {
            get { return cacheOutput.Checksum; }
        }

        public MongoIndexOutput(MongoDirectory indexDirectory, IOContext context, string indexFileName)
        {
            this.indexDirectory = indexDirectory;
            this.indexFileName = indexFileName;

            cacheOutput = indexDirectory.CacheDirectory.CreateOutput(indexFileName, context);
        }

        protected override void Dispose(bool disposing)
        {
            if (disposing)
            {
                Flush();

                cacheOutput.Dispose();

                if (isWritten && isFlushed)
                {
                    var fileInfo = new FileInfo(indexDirectory.GetFullPath(indexFileName));

                    using (var fs = new FileStream(indexDirectory.GetFullPath(indexFileName), FileMode.Open, FileAccess.Read))
                    {
                        var fullName = indexDirectory.GetFullName(indexFileName);

                        var options = new GridFSUploadOptions
                        {
                            Metadata = new BsonDocument
                            {
                                ["WrittenTime"] = fileInfo.LastWriteTimeUtc
                            }
                        };

                        try
                        {
                            indexDirectory.Bucket.UploadFromStream(fullName, indexFileName, fs, options);
                        }
                        catch (MongoBulkWriteException ex) when (ex.WriteErrors.Any(x => x.Code == 11000))
                        {
                            indexDirectory.Bucket.Delete(fullName);
                            indexDirectory.Bucket.UploadFromStream(fullName, indexFileName, fs, options);
                        }
                    }
                }
            }
        }

        public override long GetFilePointer()
        {
            return cacheOutput.GetFilePointer();
        }

        public override void Flush()
        {
            cacheOutput.Flush();

            isFlushed = true;
        }

        public override void WriteByte(byte b)
        {
            cacheOutput.WriteByte(b);

            isWritten = true;
        }

        public override void WriteBytes(byte[] b, int offset, int length)
        {
            cacheOutput.WriteBytes(b, offset, length);

            isWritten = true;
        }

        [Obsolete]
        public override void Seek(long pos)
        {
            cacheOutput.Seek(pos);
        }
    }
}