Example #1
0
        public override string UpdateDatabaseRecord(DatabaseRecord record, long etag)
        {
            var deletionInProgressStatus = HardDelete ? DeletionInProgressStatus.HardDelete
                : DeletionInProgressStatus.SoftDelete;

            if (record.DeletionInProgress == null)
            {
                record.DeletionInProgress = new Dictionary <string, DeletionInProgressStatus>();
            }

            if (FromNodes != null && FromNodes.Length > 0)
            {
                foreach (var node in FromNodes)
                {
                    if (record.Topology.RelevantFor(node) == false)
                    {
                        DatabaseDoesNotExistException.ThrowWithMessage(record.DatabaseName, $"Request to delete database from node '{node}' failed.");
                    }

                    // rehabs will be removed only once the replication sent all the documents to the mentor
                    if (record.Topology.Rehabs.Contains(node) == false)
                    {
                        record.Topology.RemoveFromTopology(node);
                    }

                    if (UpdateReplicationFactor)
                    {
                        record.Topology.ReplicationFactor--;
                    }
                    if (ClusterNodes.Contains(node))
                    {
                        record.DeletionInProgress[node] = deletionInProgressStatus;
                    }
                }
            }
            else
            {
                var allNodes = record.Topology.Members.Select(m => m)
                               .Concat(record.Topology.Promotables.Select(p => p))
                               .Concat(record.Topology.Rehabs.Select(r => r));

                foreach (var node in allNodes)
                {
                    if (ClusterNodes.Contains(node))
                    {
                        record.DeletionInProgress[node] = deletionInProgressStatus;
                    }
                }

                record.Topology = new DatabaseTopology
                {
                    Stamp             = record.Topology.Stamp,
                    ReplicationFactor = 0
                };
            }

            record.Topology.Stamp.Index = etag;

            return(null);
        }
Example #2
0
        public Task CreateDatabase(RequestHandlerContext context)
        {
            var databaseName = context.RouteMatch.GetCapture();

            // todo: think if we need to pass this check to the landlord
            if (context.RavenServer.ServerStore.IsPassive())
            {
                throw new NodeIsPassiveException($"Can't perform actions on the database '{databaseName}' while the node is passive.");
            }

            if (context.RavenServer.ServerStore.IdleDatabases.TryGetValue(databaseName.Value, out var replicationsDictionary))
            {
                if (context.HttpContext.Request.Query.TryGetValue("from-outgoing", out var dbId) && context.HttpContext.Request.Query.TryGetValue("etag", out var replicationEtag))
                {
                    var hasChanges = false;
                    var etag       = Convert.ToInt64(replicationEtag);

                    if (replicationsDictionary.TryGetValue(dbId, out var storedEtag))
                    {
                        if (storedEtag < etag)
                        {
                            hasChanges = true;
                        }
                    }
                    else
                    {
                        if (etag > 0)
                        {
                            hasChanges = true;
                        }
                    }

                    if (hasChanges == false)
                    {
                        throw new DatabaseIdleException($"Replication attempt doesn't have changes to database {databaseName.Value}, which is currently idle.");
                    }
                }
            }

            var databasesLandlord = context.RavenServer.ServerStore.DatabasesLandlord;
            var database          = databasesLandlord.TryGetOrCreateResourceStore(databaseName);

            if (database.IsCompletedSuccessfully)
            {
                context.Database = database.Result;

                if (context.Database == null)
                {
                    DatabaseDoesNotExistException.Throw(databaseName.Value);
                }

                // ReSharper disable once PossibleNullReferenceException
                if (context.Database.DatabaseShutdownCompleted.IsSet)
                {
                    using (context.RavenServer.ServerStore.ContextPool.AllocateOperationContext(out TransactionOperationContext ctx))
                        using (ctx.OpenReadTransaction())
                        {
                            if (context.RavenServer.ServerStore.Cluster.DatabaseExists(ctx, databaseName.Value))
                            {
                                // db got disabled during loading
                                throw new DatabaseDisabledException($"Cannot complete the request, because {databaseName.Value} has been disabled.");
                            }
                        }

                    // db got deleted during loading
                    DatabaseDoesNotExistException.ThrowWithMessage(databaseName.Value, "Cannot complete the request.");
                }

                return(context.Database.DatabaseShutdown.IsCancellationRequested == false
                    ? Task.CompletedTask
                    : UnlikelyWaitForDatabaseToUnload(context, context.Database, databasesLandlord, databaseName));
            }

            return(UnlikelyWaitForDatabaseToLoad(context, database, databasesLandlord, databaseName));
        }
        public override string UpdateDatabaseRecord(DatabaseRecord record, long etag)
        {
            var deletionInProgressStatus = HardDelete ? DeletionInProgressStatus.HardDelete
                : DeletionInProgressStatus.SoftDelete;

            if (MentorChangeVector != null)
            {
                if (record.DeletionInProgressChangeVector == null)
                {
                    record.DeletionInProgressChangeVector = new Dictionary <string, string>();
                }

                foreach (var entry in MentorChangeVector)
                {
                    var key   = entry.Key;
                    var value = entry.Value;

                    record.DeletionInProgressChangeVector[key] = value;
                }
            }

            if (record.DeletionInProgress == null)
            {
                record.DeletionInProgress = new Dictionary <string, DeletionInProgressStatus>();
            }
            if (FromNodes != null && FromNodes.Length > 0)
            {
                foreach (var node in FromNodes)
                {
                    if (record.Topology.RelevantFor(node) == false)
                    {
                        DatabaseDoesNotExistException.ThrowWithMessage(record.DatabaseName, $"Request to delete database from node '{node}' failed.");
                    }
                    if (UpdateReplicationFactor)
                    {
                        record.Topology.RemoveFromTopology(node);
                        record.Topology.ReplicationFactor--;
                    }
                    if (ClusterNodes.Contains(node))
                    {
                        record.DeletionInProgress[node] = deletionInProgressStatus;
                    }
                }
            }
            else
            {
                var allNodes = record.Topology.Members.Select(m => m)
                               .Concat(record.Topology.Promotables.Select(p => p))
                               .Concat(record.Topology.Rehabs.Select(r => r));

                foreach (var node in allNodes)
                {
                    if (ClusterNodes.Contains(node))
                    {
                        record.DeletionInProgress[node] = deletionInProgressStatus;
                    }
                }

                record.Topology.ReplicationFactor = 0;
                record.Topology = new DatabaseTopology
                {
                    Stamp = record.Topology.Stamp
                };
            }
            return(null);
        }