-
Notifications
You must be signed in to change notification settings - Fork 2.3k
/
BackupStrategyWrapper.java
148 lines (134 loc) · 7.09 KB
/
BackupStrategyWrapper.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
/*
* Copyright (c) 2002-2017 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.backup;
import java.io.File;
import org.neo4j.commandline.admin.CommandFailed;
import org.neo4j.kernel.impl.util.OptionalHostnamePort;
import org.neo4j.io.pagecache.PageCache;
import org.neo4j.kernel.configuration.Config;
import org.neo4j.kernel.lifecycle.LifeSupport;
class BackupStrategyWrapper
{
private final BackupStrategy backupStrategy;
private final BackupCopyService backupCopyService;
private final BackupRecoveryService backupRecoveryService;
private final PageCache pageCache;
private final Config config;
BackupStrategyWrapper( BackupStrategy backupStrategy, BackupCopyService backupCopyService, PageCache pageCache, Config config,
BackupRecoveryService backupRecoveryService )
{
this.backupStrategy = backupStrategy;
this.backupCopyService = backupCopyService;
this.pageCache = pageCache;
this.config = config;
this.backupRecoveryService = backupRecoveryService;
}
/**
* Try to do a backup using the given strategy (ex. BackupProtocol). This covers all stages (starting with incremental and falling back to a a full backup).
* The end result of this method will either be a successful backup or any other return type with the reason why the backup wasn't successful
*
* @param onlineBackupContext the command line arguments, configuration, flags
* @return the ultimate outcome of trying to do a backup with the given strategy
*/
PotentiallyErroneousState<BackupStrategyOutcome> doBackup( OnlineBackupContext onlineBackupContext )
{
LifeSupport lifeSupport = new LifeSupport();
lifeSupport.add( backupStrategy );
lifeSupport.start();
PotentiallyErroneousState<BackupStrategyOutcome> state = performBackupWithoutLifecycle( onlineBackupContext );
lifeSupport.shutdown();
return state;
}
private PotentiallyErroneousState<BackupStrategyOutcome> performBackupWithoutLifecycle( OnlineBackupContext onlineBackupContext )
{
File backupLocation = onlineBackupContext.getResolvedLocationFromName();
final File userSpecifiedBackupLocation = onlineBackupContext.getResolvedLocationFromName();
final OptionalHostnamePort userSpecifiedAddress = onlineBackupContext.getRequiredArguments().getAddress();
final Config config = onlineBackupContext.getConfig();
if ( backupCopyService.backupExists( backupLocation ) )
{
PotentiallyErroneousState<BackupStageOutcome> state =
backupStrategy.performIncrementalBackup( userSpecifiedBackupLocation, config, userSpecifiedAddress );
boolean fullBackupWontWork = BackupStageOutcome.WRONG_PROTOCOL.equals( state.getState() );
boolean incrementalWasSuccessful = BackupStageOutcome.SUCCESS.equals( state.getState() );
if ( fullBackupWontWork || incrementalWasSuccessful )
{
return describeOutcome( state );
}
if ( !onlineBackupContext.getRequiredArguments().isFallbackToFull() )
{
return describeOutcome( state );
}
}
return describeOutcome( fullBackupWithTemporaryFolderResolutions( onlineBackupContext ) );
}
private PotentiallyErroneousState<BackupStageOutcome> fullBackupWithTemporaryFolderResolutions( OnlineBackupContext onlineBackupContext )
{
final File userSpecifiedBackupLocation = onlineBackupContext.getResolvedLocationFromName();
File temporaryFullBackupLocation = backupCopyService.findAnAvailableLocationForNewFullBackup( userSpecifiedBackupLocation );
PotentiallyErroneousState<BackupStageOutcome> state = backupStrategy.performFullBackup( temporaryFullBackupLocation, onlineBackupContext.getConfig(),
onlineBackupContext.getRequiredArguments().getAddress() );
boolean aBackupAlreadyExisted =
userSpecifiedBackupLocation.equals( temporaryFullBackupLocation ); // NOTE temporaryFullBackupLocation can be equal to desired
if ( BackupStageOutcome.SUCCESS.equals( state.getState() ) )
{
backupRecoveryService.recoverWithDatabase( temporaryFullBackupLocation, pageCache, config );
if ( !aBackupAlreadyExisted )
{
try
{
renameTemporaryBackupToExpected( temporaryFullBackupLocation, userSpecifiedBackupLocation );
}
catch ( CommandFailed commandFailed )
{
return new PotentiallyErroneousState<>( BackupStageOutcome.UNRECOVERABLE_FAILURE, commandFailed );
}
}
}
return state;
}
private void renameTemporaryBackupToExpected( File temporaryFullBackupLocation, File userSpecifiedBackupLocation ) throws CommandFailed
{
File newBackupLocationForPreExistingBackup = backupCopyService.findNewBackupLocationForBrokenExisting( userSpecifiedBackupLocation );
backupCopyService.moveBackupLocation( userSpecifiedBackupLocation, newBackupLocationForPreExistingBackup );
backupCopyService.moveBackupLocation( temporaryFullBackupLocation, userSpecifiedBackupLocation );
}
private PotentiallyErroneousState<BackupStrategyOutcome> describeOutcome( PotentiallyErroneousState<BackupStageOutcome> strategyStageOutcome )
{
BackupStageOutcome stageOutcome = strategyStageOutcome.getState();
if ( stageOutcome == BackupStageOutcome.SUCCESS )
{
return new PotentiallyErroneousState<>( BackupStrategyOutcome.SUCCESS, null );
}
if ( stageOutcome == BackupStageOutcome.WRONG_PROTOCOL )
{
return new PotentiallyErroneousState<>( BackupStrategyOutcome.INCORRECT_STRATEGY, strategyStageOutcome.getCause().orElse( null ) );
}
if ( stageOutcome == BackupStageOutcome.FAILURE )
{
return new PotentiallyErroneousState<>( BackupStrategyOutcome.CORRECT_STRATEGY_FAILED, strategyStageOutcome.getCause().orElse( null ) );
}
if ( stageOutcome == BackupStageOutcome.UNRECOVERABLE_FAILURE )
{
return new PotentiallyErroneousState<>( BackupStrategyOutcome.ABSOLUTE_FAILURE, strategyStageOutcome.getCause().orElse( null ) );
}
throw new RuntimeException( "Not all enums covered: " + stageOutcome );
}
}