2025-01-07 09:40:39 +01:00
// Copyright (C) 2025, The Duplicati Team
2024-02-28 15:45:30 +01:00
// https://duplicati.com, hello@duplicati.com
2025-05-19 13:19:24 +02:00
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
2024-02-28 15:45:30 +01:00
// Software is furnished to do so, subject to the following conditions:
2025-05-19 13:19:24 +02:00
//
// The above copyright notice and this permission notice shall be included in
2024-02-28 15:45:30 +01:00
// all copies or substantial portions of the Software.
2025-05-19 13:19:24 +02:00
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
2024-02-28 15:45:30 +01:00
// DEALINGS IN THE SOFTWARE.
2025-05-19 21:59:44 +02:00
2025-04-03 11:31:36 +02:00
#nullable enable
2019-01-25 23:37:57 +01:00
using System ;
using System.Collections.Generic ;
using System.Linq ;
2025-05-19 13:19:24 +02:00
using System.Threading.Tasks ;
2025-05-19 15:43:55 +02:00
using Duplicati.Library.Utility ;
2025-05-19 13:19:24 +02:00
using Microsoft.Data.Sqlite ;
2019-01-25 23:37:57 +01:00
namespace Duplicati.Library.Main.Database
{
internal class LocalBackupDatabase : LocalDatabase
{
/// <summary>
/// The tag used for logging
/// </summary>
private static readonly string LOGTAG = Logging . Log . LogTagFromType < LocalBackupDatabase >();
2025-05-19 13:19:24 +02:00
private SqliteCommand m_findblockCommand = null !;
private SqliteCommand m_findblocksetCommand = null !;
private SqliteCommand m_findfilesetCommand = null !;
private SqliteCommand m_findmetadatasetCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_insertblockCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_insertfileCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_insertblocksetCommand = null !;
private SqliteCommand m_insertblocksetentryCommand = null !;
private SqliteCommand m_insertblocklistHashesCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_insertmetadatasetCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_findfileCommand = null !;
private SqliteCommand m_selectfilelastmodifiedCommand = null !;
private SqliteCommand m_selectfilelastmodifiedWithSizeCommand = null !;
private SqliteCommand m_selectfileHashCommand = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
private SqliteCommand m_insertfileOperationCommand = null !;
private SqliteCommand m_selectfilemetadatahashandsizeCommand = null !;
private SqliteCommand m_getfirstfilesetwithblockinblockset = null !;
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
private HashSet < string > m_blocklistHashes = [];
2019-08-05 20:14:05 -04:00
2025-04-06 08:41:59 +02:00
/// <summary>
/// The temporary table with deleted blocks that can be re-used; null if not table is used
/// </summary>
private string? m_tempDeletedBlockTable ;
/// <summary>
/// The in-mmeory lookup for deleted blocks; null if in-memory lookup is not used
/// </summary>
private Dictionary < string , Dictionary < long , long >>? m_deletedBlockLookup ;
/// <summary>
/// The command used to move deleted blocks to the main block table; null if not used
/// </summary>
2025-05-19 13:19:24 +02:00
private SqliteCommand ? m_moveblockfromdeletedCommand ;
2025-04-06 08:41:59 +02:00
/// <summary>
/// The command used to find blocks in the deleted blocks table; null if not used
/// </summary>
2025-05-19 13:19:24 +02:00
private SqliteCommand ? m_findindeletedCommand ;
2025-04-06 08:41:59 +02:00
2019-01-25 23:37:57 +01:00
private long m_filesetId ;
2025-05-19 13:19:24 +02:00
private bool m_logQueries ;
2019-01-25 23:37:57 +01:00
2025-05-21 07:24:36 +02:00
public static async Task < LocalBackupDatabase > CreateAsync ( string path , Options options , LocalBackupDatabase ? dbnew = null )
2019-01-25 23:37:57 +01:00
{
2025-05-21 07:24:36 +02:00
dbnew ??= new LocalBackupDatabase ();
2025-05-19 13:19:24 +02:00
2025-05-21 07:24:36 +02:00
dbnew = ( LocalBackupDatabase ) await CreateLocalDatabaseAsync ( path , "Backup" , false , options . SqlitePageCache , dbnew );
dbnew . ShouldCloseConnection = true ;
2025-05-19 13:19:24 +02:00
2025-05-21 07:24:36 +02:00
return await CreateAsync ( dbnew , options );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-21 07:24:36 +02:00
public static async Task < LocalBackupDatabase > CreateAsync ( LocalDatabase dbparent , Options options , LocalBackupDatabase ? dbnew = null )
2019-01-25 23:37:57 +01:00
{
2025-05-21 07:24:36 +02:00
dbnew ??= new LocalBackupDatabase ();
2025-05-19 13:19:24 +02:00
dbnew = ( LocalBackupDatabase ) await CreateLocalDatabaseAsync ( dbparent , dbnew );
dbnew . m_logQueries = options . ProfileAllDatabaseQueries ;
2025-05-19 21:59:44 +02:00
dbnew . m_findblockCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT ""ID""
FROM ""Block""
WHERE
2025-05-22 13:34:29 +02:00
""Hash"" = @Hash
2025-05-19 21:59:44 +02:00
AND ""Size"" = @Size
" );
dbnew . m_findblocksetCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT ""ID""
FROM ""Blockset""
WHERE
""Fullhash"" = @Fullhash
AND ""Length"" = @Length
" );
dbnew . m_findmetadatasetCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT ""A"".""ID""
FROM
""Metadataset"" A,
""BlocksetEntry"" B,
""Block"" C
WHERE
""A"".""BlocksetID"" = ""B"".""BlocksetID""
AND ""B"".""BlockID"" = ""C"".""ID""
AND ""C"".""Hash"" = @Hash
AND ""C"".""Size"" = @Size
" );
dbnew . m_findfilesetCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT ""ID""
FROM ""FileLookup""
WHERE
""BlocksetID"" = @BlocksetId
AND ""MetadataID"" = @MetadataId
AND ""Path"" = @Path
AND ""PrefixID"" = @PrefixId
" );
dbnew . m_insertblockCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""Block"" (
""Hash"",
""VolumeID"",
""Size""
)
VALUES (
@Hash,
@VolumeId,
@Size
);
SELECT last_insert_rowid();
" );
dbnew . m_insertfileOperationCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""FilesetEntry"" (
""FilesetID"",
""FileID"",
""Lastmodified""
)
VALUES (
@FilesetId,
@FileId,
@LastModified
)
" );
dbnew . m_insertfileCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""FileLookup"" (
""PrefixID"",
""Path"",
""BlocksetID"",
""MetadataID""
)
VALUES (
@PrefixId,
@Path,
@BlocksetId,
@MetadataId
);
SELECT last_insert_rowid();
" );
dbnew . m_insertblocksetCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""Blockset"" (
""Length"",
""FullHash""
)
VALUES (
@Length,
@Fullhash
);
SELECT last_insert_rowid();" );
dbnew . m_insertblocksetentryCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""BlocksetEntry"" (
""BlocksetID"",
""Index"",
""BlockID""
)
SELECT
@BlocksetId AS A,
@Index AS B,
""ID""
FROM ""Block""
WHERE ""Hash"" = @Hash
AND ""Size"" = @Size
" );
dbnew . m_insertblocklistHashesCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""BlocklistHash"" (
""BlocksetID"",
""Index"",
""Hash""
)
VALUES (
@BlocksetId,
@Index,
@Hash
)
" );
dbnew . m_insertmetadatasetCommand = await dbnew . Connection . CreateCommandAsync ( @"
INSERT INTO ""Metadataset"" (""BlocksetID"")
VALUES (@BlocksetId);
SELECT last_insert_rowid();
" );
dbnew . m_selectfilelastmodifiedCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT
""A"".""ID"",
""B"".""LastModified""
FROM (
SELECT ""ID""
FROM ""FileLookup""
WHERE ""PrefixID"" = @PrefixId
AND ""Path"" = @Path
) ""A""
CROSS JOIN ""FilesetEntry"" ""B""
WHERE
""A"".""ID"" = ""B"".""FileID""
AND ""B"".""FilesetID"" = @FilesetId
" );
dbnew . m_selectfilelastmodifiedWithSizeCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT
""C"".""ID"",
""C"".""LastModified"",
""D"".""Length""
FROM
(
SELECT
""A"".""ID"",
""B"".""LastModified"",
""A"".""BlocksetID""
FROM (
SELECT
""ID"",
""BlocksetID""
FROM ""FileLookup""
WHERE
""PrefixID"" = @PrefixId
AND ""Path"" = @Path
) ""A""
CROSS JOIN ""FilesetEntry"" ""B""
WHERE
""A"".""ID"" = ""B"".""FileID""
AND ""B"".""FilesetID"" = @FilesetId
) AS ""C"",
""Blockset"" AS ""D""
WHERE ""C"".""BlocksetID"" == ""D"".""ID""
" );
dbnew . m_selectfilemetadatahashandsizeCommand = await dbnew . Connection . CreateCommandAsync ( @"
SELECT
""Blockset"".""Length"",
""Blockset"".""FullHash""
FROM
""Blockset"",
""Metadataset"",
""File""
WHERE
""File"".""ID"" = @FileId
AND ""Blockset"".""ID"" = ""Metadataset"".""BlocksetID""
AND ""Metadataset"".""ID"" = ""File"".""MetadataID""
" );
2025-04-06 08:41:59 +02:00
// Experimental toggling of the deleted block cache
// If the value is less than zero, the lookup is disabled
// meaning that deleted blocks are never reused (same as 2.1.0.5 and earlier)
// A value of zero disables the in-memory cache, always using a temporary table
// Any other value is the size of the in-memory cache
// If the number of deleted blocks exceed the cache size, a temporary table is used
var deletedBlockCacheSize = Environment . GetEnvironmentVariable ( "DUPLICATI_DELETEDBLOCKCACHESIZE" );
if (! long . TryParse ( deletedBlockCacheSize , out var deletedBlockCacheSizeLong ))
deletedBlockCacheSizeLong = 10000 ;
if ( deletedBlockCacheSizeLong >= 0 )
{
2025-05-19 13:19:24 +02:00
using ( var cmd = dbnew . Connection . CreateCommand ())
2025-04-06 08:41:59 +02:00
{
2025-05-19 13:19:24 +02:00
dbnew . m_tempDeletedBlockTable = "DeletedBlock-" + Library . Utility . Utility . ByteArrayAsHexString ( Guid . NewGuid (). ToByteArray ());
2025-05-20 06:07:45 +02:00
await cmd . SetCommandAndParameters ( $@"
CREATE TEMPORARY TABLE ""{dbnew.m_tempDeletedBlockTable}"" AS
SELECT
MAX(""ID"") AS ""ID"",
""Hash"",
""Size""
FROM ""DeletedBlock""
WHERE ""VolumeID"" IN (
SELECT ""ID""
FROM ""RemoteVolume""
WHERE ""State"" NOT IN (@States)
)
GROUP BY
""Hash"",
""Size""
" )
2025-05-22 13:24:00 +02:00
. ExpandInClauseParameterAsync ( "@States" , [
2025-05-20 06:07:45 +02:00
RemoteVolumeState . Deleted ,
RemoteVolumeState . Deleting
])
. ExecuteNonQueryAsync ();
2025-04-06 08:41:59 +02:00
2025-05-20 06:07:45 +02:00
var deletedBlocks = await cmd . ExecuteScalarInt64Async ( @ $"
SELECT COUNT(*)
FROM ""{dbnew.m_tempDeletedBlockTable}""
" , 0 );
2025-04-06 08:41:59 +02:00
// There are no deleted blocks, so we can drop the table
if ( deletedBlocks == 0 )
{
2025-05-19 15:43:55 +02:00
await cmd . ExecuteNonQueryAsync ( $@"DROP TABLE ""{dbnew.m_tempDeletedBlockTable}""" );
2025-05-19 13:19:24 +02:00
dbnew . m_tempDeletedBlockTable = null ;
2025-04-06 08:41:59 +02:00
}
// The deleted blocks are small enough to fit in memory
else if ( deletedBlocks <= deletedBlockCacheSizeLong )
{
2025-05-19 13:19:24 +02:00
dbnew . m_deletedBlockLookup = new Dictionary < string , Dictionary < long , long >>();
2025-05-20 06:07:45 +02:00
cmd . SetCommandAndParameters ( @ $"
SELECT
""ID"",
""Hash"",
""Size""
FROM ""{dbnew.m_tempDeletedBlockTable}""
" );
using ( var reader = await cmd . ExecuteReaderAsync ())
2025-05-19 15:43:55 +02:00
while ( await reader . ReadAsync ())
2025-04-06 08:41:59 +02:00
{
var id = reader . ConvertValueToInt64 ( 0 );
var hash = reader . ConvertValueToString ( 1 ) ?? throw new Exception ( "Hash is null" );
var size = reader . ConvertValueToInt64 ( 2 );
2025-05-19 13:19:24 +02:00
if (! dbnew . m_deletedBlockLookup . TryGetValue ( hash , out var sizes ))
dbnew . m_deletedBlockLookup [ hash ] = sizes = new Dictionary < long , long >();
2025-04-06 08:41:59 +02:00
sizes [ size ] = id ;
}
2025-05-19 15:43:55 +02:00
await cmd . ExecuteNonQueryAsync ( $@"DROP TABLE ""{dbnew.m_tempDeletedBlockTable}""" );
2025-05-19 13:19:24 +02:00
dbnew . m_tempDeletedBlockTable = null ;
2025-04-06 08:41:59 +02:00
}
// The deleted blocks are too large to fit in memory, so we use a temporary table
else
{
2025-05-20 06:07:45 +02:00
await cmd . ExecuteNonQueryAsync ( $@"
CREATE UNIQUE INDEX ""unique_{dbnew.m_tempDeletedBlockTable}""
ON ""{dbnew.m_tempDeletedBlockTable}"" (
""Hash"",
""Size""
)
" );
dbnew . m_findindeletedCommand = await dbnew . Connection . CreateCommandAsync ( $@"
SELECT ""ID""
FROM ""{dbnew.m_tempDeletedBlockTable}""
WHERE
""Hash"" = @Hash
AND ""Size"" = @Size
" );
dbnew . m_moveblockfromdeletedCommand = await dbnew . Connection . CreateCommandAsync ( @ $"
INSERT INTO ""Block"" (
""Hash"",
""Size"",
""VolumeID""
)
SELECT
""Hash"",
""Size"",
""VolumeID""
FROM ""DeletedBlock""
WHERE ""ID"" = @DeletedBlockId LIMIT 1;
DELETE FROM ""DeletedBlock""
WHERE ""ID"" = @DeletedBlockId;
DELETE FROM {dbnew.m_tempDeletedBlockTable}
WHERE ""ID"" = @DeletedBlockId;
SELECT last_insert_rowid()
" );
2025-04-06 08:41:59 +02:00
}
if ( deletedBlocks > 0 )
{
2025-05-20 06:07:45 +02:00
dbnew . m_moveblockfromdeletedCommand = await dbnew . m_connection . CreateCommandAsync ( @ $"
INSERT INTO ""Block"" (
""Hash"",
""Size"",
""VolumeID""
)
SELECT
""Hash"",
""Size"",
""VolumeID""
FROM ""DeletedBlock""
WHERE ""ID"" = @DeletedBlockId LIMIT 1;
DELETE FROM ""DeletedBlock""
WHERE ""ID"" = @DeletedBlockId;
SELECT last_insert_rowid()
" );
2025-04-06 08:41:59 +02:00
}
}
}
2019-01-25 23:37:57 +01:00
// Allow users to test on real-world data
// to get feedback on potential performance
int . TryParse ( Environment . GetEnvironmentVariable ( "TEST_QUERY_VERSION" ), out var testqueryversion );
if ( testqueryversion != 0 )
Logging . Log . WriteWarningMessage ( LOGTAG , "TestFileQuery" , null , "Using performance test query version {0} as the TEST_QUERY_VERSION environment variable is set" , testqueryversion );
2025-05-19 13:19:24 +02:00
// The original query (v==1) finds the most recent entry of the file in question,
2019-01-25 23:37:57 +01:00
// but it requires some large joins to extract the required information.
// To speed it up, we use a slightly simpler approach that only looks at the
// previous fileset, and uses information here.
// If there is a case where a file is sometimes there and sometimes not
// (i.e. filter file, remove filter) we will not find the file.
2025-05-19 13:19:24 +02:00
// We currently use this faster version,
2019-01-25 23:37:57 +01:00
// but allow users to switch back via an environment variable
// such that we can get performance feedback
2025-03-18 20:30:27 +01:00
string findQuery ;
2019-01-25 23:37:57 +01:00
switch ( testqueryversion )
{
// The query used in Duplicati until 2.0.3.9
case 1 :
2025-05-20 06:07:45 +02:00
findQuery = @"
SELECT
""FileLookup"".""ID"" AS ""FileID"",
""FilesetEntry"".""Lastmodified"",
""FileBlockset"".""Length"",
""MetaBlockset"".""Fullhash"" AS ""Metahash"",
""MetaBlockset"".""Length"" AS ""Metasize""
FROM
""FileLookup"",
""FilesetEntry"",
""Fileset"",
""Blockset"" ""FileBlockset"",
""Metadataset"",
""Blockset"" ""MetaBlockset""
WHERE
""FileLookup"".""PrefixID"" = @PrefixId
AND ""FileLookup"".""Path"" = @Path
AND ""FilesetEntry"".""FileID"" = ""FileLookup"".""ID""
AND ""Fileset"".""ID"" = ""FilesetEntry"".""FilesetID""
AND ""FileBlockset"".""ID"" = ""FileLookup"".""BlocksetID""
AND ""Metadataset"".""ID"" = ""FileLookup"".""MetadataID""
AND ""MetaBlockset"".""ID"" = ""Metadataset"".""BlocksetID""
AND @FilesetId IS NOT NULL
ORDER BY ""Fileset"".""Timestamp"" DESC
LIMIT 1
" ;
2019-01-25 23:37:57 +01:00
break ;
// The fastest reported query in Duplicati 2.0.3.10, but with "LIMIT 1" added
default :
case 2 :
2025-05-20 06:07:45 +02:00
var getLastFileEntryForPath = @"
SELECT
""A"".""ID"",
""B"".""LastModified"",
""A"".""BlocksetID"",
""A"".""MetadataID""
FROM (
SELECT
""ID"",
""BlocksetID"",
""MetadataID""
FROM ""FileLookup""
WHERE
""PrefixID"" = @PrefixId
AND ""Path"" = @Path
) ""A""
CROSS JOIN ""FilesetEntry"" ""B""
WHERE
""A"".""ID"" = ""B"".""FileID""
AND ""B"".""FilesetID"" = @FilesetId
" ;
findQuery = $@"
SELECT
""C"".""ID"" AS ""FileID"",
""C"".""LastModified"",
""D"".""Length"",
""E"".""FullHash"" as ""Metahash"",
""E"".""Length"" AS ""Metasize""
FROM
({getLastFileEntryForPath}) AS ""C"",
""Blockset"" AS ""D"",
""Blockset"" AS ""E"",
""Metadataset"" ""F""
WHERE
""C"".""BlocksetID"" == ""D"".""ID""
AND ""C"".""MetadataID"" == ""F"".""ID""
AND ""F"".""BlocksetID"" = ""E"".""ID""
LIMIT 1
" ;
2019-01-25 23:37:57 +01:00
break ;
// Potentially faster query: https://forum.duplicati.com/t/release-2-0-3-10-canary-2018-08-30/4497/25
case 3 :
2025-05-20 06:07:45 +02:00
findQuery = @"
SELECT
FileLookup.ID as FileID,
FilesetEntry.Lastmodified,
FileBlockset.Length,
MetaBlockset.FullHash AS Metahash,
MetaBlockset.Length as Metasize
FROM FilesetEntry
INNER JOIN Fileset
ON (FileSet.ID = FilesetEntry.FilesetID)
INNER JOIN FileLookup
ON (FileLookup.ID = FilesetEntry.FileID)
INNER JOIN Metadataset
ON (Metadataset.ID = FileLookup.MetadataID)
INNER JOIN Blockset AS MetaBlockset
ON (MetaBlockset.ID = Metadataset.BlocksetID)
LEFT JOIN Blockset AS FileBlockset
ON (FileBlockset.ID = FileLookup.BlocksetID)
WHERE
FileLookup.PrefixID = @PrefixId
AND FileLookup.Path = @Path
AND FilesetID = @FilesetId
LIMIT 1
" ;
2019-01-25 23:37:57 +01:00
break ;
// The slow query used in Duplicati 2.0.3.10, but with "LIMIT 1" added
case 4 :
2025-05-20 06:07:45 +02:00
findQuery = @"
SELECT
""FileLookup"".""ID"" AS ""FileID"",
""FilesetEntry"".""Lastmodified"",
""FileBlockset"".""Length"",
""MetaBlockset"".""Fullhash"" AS ""Metahash"",
""MetaBlockset"".""Length"" AS ""Metasize""
FROM
""FileLookup"",
""FilesetEntry"",
""Fileset"",
""Blockset"" ""FileBlockset"",
""Metadataset"",
""Blockset"" ""MetaBlockset""
WHERE
""FileLookup"".""PrefixID"" = @PrefixId
AND ""FileLookup"".""Path"" = @Path
AND ""Fileset"".""ID"" = @FilesetId
AND ""FilesetEntry"".""FileID"" = ""FileLookup"".""ID""
AND ""Fileset"".""ID"" = ""FilesetEntry"".""FilesetID""
AND ""FileBlockset"".""ID"" = ""FileLookup"".""BlocksetID""
AND ""Metadataset"".""ID"" = ""FileLookup"".""MetadataID""
AND ""MetaBlockset"".""ID"" = ""Metadataset"".""BlocksetID""
LIMIT 1
" ;
2019-01-25 23:37:57 +01:00
break ;
}
2025-05-19 13:19:24 +02:00
dbnew . m_findfileCommand = dbnew . m_connection . CreateCommand ( findQuery );
2019-01-25 23:37:57 +01:00
2025-05-20 06:07:45 +02:00
dbnew . m_selectfileHashCommand = dbnew . m_connection . CreateCommand ( @"
SELECT ""Blockset"".""Fullhash""
FROM ""Blockset"", ""FileLookup""
WHERE
""Blockset"".""ID"" = ""FileLookup"".""BlocksetID""
AND ""FileLookup"".""ID"" = @FileId
" );
dbnew . m_getfirstfilesetwithblockinblockset = dbnew . m_connection . CreateCommand ( @"
SELECT MIN(""FilesetEntry"".""FilesetID"")
FROM ""FilesetEntry""
WHERE ""FilesetEntry"".""FileID"" IN (
SELECT ""File"".""ID""
FROM ""File""
WHERE ""File"".""BlocksetID"" IN(
SELECT ""BlocklistHash"".""BlocksetID""
FROM ""BlocklistHash""
WHERE ""BlocklistHash"".""Hash"" = @Hash
)
)
" );
2019-01-25 23:37:57 +01:00
2025-05-19 13:19:24 +02:00
dbnew . m_blocklistHashes = new HashSet < string >();
return dbnew ;
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2019-01-25 23:37:57 +01:00
/// <summary>
/// Probes to see if a block already exists
/// </summary>
/// <param name="key">The block key</param>
/// <param name="size">The size of the block</param>
/// <returns>True if the block should be added to the current output</returns>
2025-05-19 15:43:55 +02:00
public async Task < long > FindBlockID ( string key , long size )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
return await m_findblockCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Hash" , key )
. SetParameterValue ( "@Size" , size )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries , - 1 );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
/// <summary>
/// Adds a block to the local database, returning a value indicating if the value presents a new block
/// </summary>
/// <param name="key">The block key</param>
/// <param name="size">The size of the block</param>
/// <returns>True if the block should be added to the current output</returns>
2025-05-19 15:43:55 +02:00
public async Task < bool > AddBlock ( string key , long size , long volumeid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var r = await FindBlockID ( key , size );
2019-01-25 23:37:57 +01:00
if ( r == - 1L )
{
2025-04-06 08:41:59 +02:00
if ( m_moveblockfromdeletedCommand != null )
2024-11-01 16:27:06 +01:00
{
2025-04-06 08:41:59 +02:00
if ( m_deletedBlockLookup != null )
{
if ( m_deletedBlockLookup . TryGetValue ( key , out var sizes ))
if ( sizes . TryGetValue ( size , out var id ))
{
2025-05-19 15:43:55 +02:00
await m_moveblockfromdeletedCommand
. SetTransaction ( m_rtr )
2025-04-06 08:41:59 +02:00
. SetParameterValue ( "@DeletedBlockId" , id )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2025-04-06 08:41:59 +02:00
sizes . Remove ( size );
if ( sizes . Count == 0 )
m_deletedBlockLookup . Remove ( key );
return false ;
}
}
else if ( m_findindeletedCommand != null )
{
// No transaction on the temporary table
2025-05-19 15:43:55 +02:00
var id = await m_findindeletedCommand
2025-04-06 08:41:59 +02:00
. SetParameterValue ( "@Hash" , key )
. SetParameterValue ( "@Size" , size )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries , - 1 );
2024-11-01 16:27:06 +01:00
2025-04-06 08:41:59 +02:00
if ( id != - 1 )
{
2025-05-19 15:43:55 +02:00
var c = await m_moveblockfromdeletedCommand
. SetTransaction ( m_rtr )
2025-04-06 08:41:59 +02:00
. SetParameterValue ( "@DeletedBlockId" , id )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2024-11-01 16:27:06 +01:00
2025-04-06 08:41:59 +02:00
if ( c != 2 )
throw new Exception ( $"Failed to move block {key} with size {size}, result count: {c}" );
// We do not clean up the temporary table, as the regular block lookup should now find it
return false ;
}
}
2024-11-01 16:27:06 +01:00
}
2025-05-19 15:43:55 +02:00
var ins = await m_insertblockCommand
. SetTransaction ( m_rtr )
2025-04-06 08:41:59 +02:00
. SetParameterValue ( "@Hash" , key )
. SetParameterValue ( "@VolumeId" , volumeid )
. SetParameterValue ( "@Size" , size )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2025-04-06 08:41:59 +02:00
if ( ins != 1 )
throw new Exception ( $"Failed to insert block {key} with size {size}, result count: {ins}" );
2019-01-25 23:37:57 +01:00
return true ;
}
else
{
//Update lookup cache if required
return false ;
}
}
/// <summary>
/// Adds a blockset to the database, returns a value indicating if the blockset is new
/// </summary>
/// <param name="filehash">The hash of the blockset</param>
/// <param name="size">The size of the blockset</param>
/// <param name="hashes">The list of hashes</param>
/// <param name="blocksetid">The id of the blockset, new or old</param>
/// <returns>True if the blockset was created, false otherwise</returns>
2025-05-19 15:43:55 +02:00
public async Task <( bool , long )> AddBlockset ( string filehash , long size , int blocksize , IEnumerable < string > hashes , IEnumerable < string > blocklistHashes )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
long blocksetid = await m_findblocksetCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Fullhash" , filehash )
. SetParameterValue ( "@Length" , size )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries , null , - 1 );
2019-01-25 23:37:57 +01:00
if ( blocksetid != - 1 )
2025-05-19 15:43:55 +02:00
return ( false , blocksetid ); //Found it
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
blocksetid = await m_insertblocksetCommand
. SetTransaction ( m_rtr )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@Length" , size )
. SetParameterValue ( "@Fullhash" , filehash )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries );
2025-05-19 14:40:00 +02:00
long ix = 0 ;
if ( blocklistHashes != null )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
m_insertblocklistHashesCommand
. SetTransaction ( m_rtr )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@BlocksetId" , blocksetid );
2019-01-25 23:37:57 +01:00
2025-05-19 14:40:00 +02:00
foreach ( var bh in blocklistHashes )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var c = await m_insertblocklistHashesCommand
. SetParameterValue ( "@Index" , ix )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@Hash" , bh )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2025-05-19 14:40:00 +02:00
if ( c != 1 )
throw new Exception ( $"Failed to insert blocklist hash {bh} for blockset {blocksetid}, result count: {c}" );
ix ++;
2019-01-25 23:37:57 +01:00
}
2025-05-19 14:40:00 +02:00
}
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
m_insertblocksetentryCommand
. SetTransaction ( m_rtr )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@BlocksetId" , blocksetid );
2019-01-25 23:37:57 +01:00
2025-05-19 14:40:00 +02:00
ix = 0 ;
long remainsize = size ;
foreach ( var h in hashes )
{
var exsize = remainsize < blocksize ? remainsize : blocksize ;
2025-05-19 15:43:55 +02:00
var c = await m_insertblocksetentryCommand
. SetParameterValue ( "@Index" , ix )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@Hash" , h )
. SetParameterValue ( "@Size" , exsize )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2025-05-19 14:40:00 +02:00
if ( c != 1 )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
Logging . Log . WriteErrorMessage ( LOGTAG , "CheckingErrorsForIssue1400" , null , "Checking errors, related to #1400. Unexpected result count: {0}, expected {1}, hash: {2}, size: {3}, blocksetid: {4}, ix: {5}, fullhash: {6}, fullsize: {7}" , c , 1 , h , exsize , blocksetid , ix , filehash , size );
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var bid = await cmd . SetCommandAndParameters ( @"
SELECT ""ID""
FROM ""Block""
WHERE ""Hash"" = @Hash
" )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@Hash" , h )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
2025-05-19 14:40:00 +02:00
if ( bid == - 1 )
throw new Exception ( $"Could not find any blocks with the given hash: {h}" );
2025-05-19 15:43:55 +02:00
cmd . SetCommandAndParameters ( @"
SELECT ""Size""
FROM ""Block""
WHERE ""Hash"" = @Hash
" )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@Hash" , h );
2025-05-19 15:43:55 +02:00
await foreach ( var rd in cmd . ExecuteReaderEnumerableAsync ())
2025-05-19 14:40:00 +02:00
Logging . Log . WriteErrorMessage ( LOGTAG , "FoundIssue1400Error" , null , "Found block with ID {0} and hash {1} and size {2}" , bid , h , rd . ConvertValueToInt64 ( 0 , - 1 ));
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-19 14:40:00 +02:00
throw new Exception ( $"Unexpected result count: {c}, expected {1}, check log for more messages" );
2019-01-25 23:37:57 +01:00
}
2025-05-19 14:40:00 +02:00
ix ++;
remainsize -= blocksize ;
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
await m_rtr . CommitAsync ();
2025-05-19 14:40:00 +02:00
2025-05-19 15:43:55 +02:00
return ( true , blocksetid );
2019-01-25 23:37:57 +01:00
}
/// <summary>
/// Gets the metadataset ID from the filehash
/// </summary>
/// <returns><c>true</c>, if metadataset found, false if does not exist.</returns>
/// <param name="filehash">The metadata hash.</param>
/// <param name="size">The size of the metadata.</param>
/// <param name="metadataid">The ID of the metadataset.</param>
/// <param name="transaction">An optional transaction.</param>
2025-05-19 15:43:55 +02:00
public async Task <( bool , long )> GetMetadatasetID ( string filehash , long size )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
long metadataid ;
2019-01-25 23:37:57 +01:00
if ( size > 0 )
{
2025-05-19 15:43:55 +02:00
metadataid = await m_findmetadatasetCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Hash" , filehash )
. SetParameterValue ( "@Size" , size )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries , null , - 1 );
return ( metadataid != - 1 , metadataid );
2019-01-25 23:37:57 +01:00
}
metadataid = - 2 ;
2025-05-19 15:43:55 +02:00
return ( false , metadataid );
2019-01-25 23:37:57 +01:00
}
/// <summary>
/// Adds a metadata set to the database, and returns a value indicating if the record was new
/// </summary>
/// <param name="filehash">The metadata hash</param>
/// <param name="size">The size of the metadata</param>
/// <param name="transaction">The transaction to execute under</param>
/// <param name="blocksetid">The id of the blockset to add</param>
/// <param name="metadataid">The id of the metadata set</param>
/// <returns>True if the set was added to the database, false otherwise</returns>
2025-05-19 15:43:55 +02:00
public async Task <( bool , long )> AddMetadataset ( string filehash , long size , long blocksetid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var ( metadatafound , metadataid ) = await GetMetadatasetID ( filehash , size );
if ( metadatafound )
return ( false , metadataid );
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
metadataid = await m_insertmetadatasetCommand
. SetTransaction ( m_rtr )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@BlocksetId" , blocksetid )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries );
2025-05-19 14:40:00 +02:00
m_rtr . Commit ();
2025-05-19 15:43:55 +02:00
return ( true , metadataid );
2019-01-25 23:37:57 +01:00
}
/// <summary>
/// Adds a file record to the database
/// </summary>
/// <param name="pathprefixid">The path prefix ID</param>
/// <param name="filename">The path to the file</param>
/// <param name="lastmodified">The time the file was modified</param>
/// <param name="blocksetID">The ID of the hashkey for the file</param>
/// <param name="metadataID">The ID for the metadata</param>
/// <param name="transaction">The transaction to use for insertion, or null for no transaction</param>
2025-05-19 15:43:55 +02:00
public async Task AddFile ( long pathprefixid , string filename , DateTime lastmodified , long blocksetID , long metadataID )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var fileidobj = await m_findfilesetCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@BlocksetId" , blocksetID )
. SetParameterValue ( "@MetadataId" , metadataID )
. SetParameterValue ( "@Path" , filename )
. SetParameterValue ( "@PrefixId" , pathprefixid )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries );
2019-01-25 23:37:57 +01:00
if ( fileidobj == - 1 )
{
2025-05-19 15:43:55 +02:00
fileidobj = await m_insertfileCommand . SetTransaction ( m_rtr )
2025-05-19 14:40:00 +02:00
. SetParameterValue ( "@PrefixId" , pathprefixid )
. SetParameterValue ( "@Path" , filename )
. SetParameterValue ( "@BlocksetId" , blocksetID )
. SetParameterValue ( "@MetadataId" , metadataID )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ( m_logQueries );
await m_rtr . CommitAsync ();
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
await AddKnownFile ( fileidobj , lastmodified );
2019-01-25 23:37:57 +01:00
}
/// <summary>
/// Adds a file record to the database
/// </summary>
/// <param name="filename">The path to the file</param>
/// <param name="lastmodified">The time the file was modified</param>
/// <param name="blocksetID">The ID of the hashkey for the file</param>
/// <param name="metadataID">The ID for the metadata</param>
/// <param name="transaction">The transaction to use for insertion, or null for no transaction</param>
2025-05-19 15:43:55 +02:00
public async Task AddFile ( string filename , DateTime lastmodified , long blocksetID , long metadataID )
2019-01-25 23:37:57 +01:00
{
var split = SplitIntoPrefixAndName ( filename );
2025-05-19 15:43:55 +02:00
await AddFile (
await GetOrCreatePathPrefix ( split . Key ),
split . Value ,
lastmodified ,
blocksetID ,
metadataID
);
2019-01-25 23:37:57 +01:00
}
2025-04-03 11:31:36 +02:00
/// <summary>
/// Adds a known file to the fileset
/// </summary>
/// <param name="fileid">Id of the file</param>
/// <param name="lastmodified">The time the file was modified</param>
/// <param name="transaction">The transaction to use for insertion, or null for no transaction</param>
2025-05-19 15:43:55 +02:00
public async Task AddKnownFile ( long fileid , DateTime lastmodified )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await m_insertfileOperationCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , m_filesetId )
. SetParameterValue ( "@FileId" , fileid )
. SetParameterValue ( "@LastModified" , lastmodified . ToUniversalTime (). Ticks )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task AddDirectoryEntry ( string path , long metadataID , DateTime lastmodified )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await AddFile ( path , lastmodified , FOLDER_BLOCKSET_ID , metadataID );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-19 15:43:55 +02:00
public async Task AddSymlinkEntry ( string path , long metadataID , DateTime lastmodified )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await AddFile ( path , lastmodified , SYMLINK_BLOCKSET_ID , metadataID );
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task <( long , DateTime , long )> GetFileLastModified ( long prefixid , string path , long filesetid , bool includeLength )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
DateTime oldModified ;
long length ;
2019-01-25 23:37:57 +01:00
if ( includeLength )
{
2025-05-19 15:43:55 +02:00
m_selectfilelastmodifiedWithSizeCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@PrefixId" , prefixid )
. SetParameterValue ( "@Path" , path )
. SetParameterValue ( "@FilesetId" , filesetid );
2025-05-19 15:43:55 +02:00
using ( var rd = await m_selectfilelastmodifiedWithSizeCommand . ExecuteReaderAsync ( m_logQueries , null ))
if ( await rd . ReadAsync ())
2019-01-25 23:37:57 +01:00
{
oldModified = new DateTime ( rd . ConvertValueToInt64 ( 1 ), DateTimeKind . Utc );
length = rd . ConvertValueToInt64 ( 2 );
2025-05-19 15:43:55 +02:00
return ( rd . ConvertValueToInt64 ( 0 ), oldModified , length );
2019-01-25 23:37:57 +01:00
}
}
else
{
2025-05-19 14:40:00 +02:00
m_selectfilelastmodifiedCommand . SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@PrefixId" , prefixid )
. SetParameterValue ( "@Path" , path )
. SetParameterValue ( "@FilesetId" , filesetid );
2025-05-19 15:43:55 +02:00
using ( var rd = await m_selectfilelastmodifiedCommand . ExecuteReaderAsync ( m_logQueries , null ))
if ( await rd . ReadAsync ())
2019-01-25 23:37:57 +01:00
{
length = - 1 ;
oldModified = new DateTime ( rd . ConvertValueToInt64 ( 1 ), DateTimeKind . Utc );
2025-05-19 15:43:55 +02:00
return ( rd . ConvertValueToInt64 ( 0 ), oldModified , length );
2019-01-25 23:37:57 +01:00
}
}
2025-05-19 15:43:55 +02:00
2019-01-25 23:37:57 +01:00
oldModified = new DateTime ( 0 , DateTimeKind . Utc );
length = - 1 ;
2025-05-19 15:43:55 +02:00
return (- 1 , oldModified , length );
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task <( long , DateTime , long , string? , long )> GetFileEntry ( long prefixid , string path , long filesetid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
DateTime oldModified ;
long lastFileSize ;
string? oldMetahash ;
long oldMetasize ;
m_findfileCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@PrefixId" , prefixid )
. SetParameterValue ( "@Path" , path )
. SetParameterValue ( "@FilesetId" , filesetid );
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
using ( var rd = await m_findfileCommand . ExecuteReaderAsync ())
if ( await rd . ReadAsync ())
2019-01-25 23:37:57 +01:00
{
oldModified = new DateTime ( rd . ConvertValueToInt64 ( 1 ), DateTimeKind . Utc );
2025-04-03 15:46:20 +02:00
lastFileSize = rd . ConvertValueToInt64 ( 2 );
oldMetahash = rd . ConvertValueToString ( 3 );
oldMetasize = rd . ConvertValueToInt64 ( 4 );
2025-05-19 15:43:55 +02:00
return (
rd . ConvertValueToInt64 ( 0 ),
oldModified ,
lastFileSize ,
oldMetahash ,
oldMetasize
);
2019-01-25 23:37:57 +01:00
}
else
{
oldModified = new DateTime ( 0 , DateTimeKind . Utc );
lastFileSize = - 1 ;
oldMetahash = null ;
oldMetasize = - 1 ;
2025-05-19 15:43:55 +02:00
return (
- 1 ,
oldModified ,
lastFileSize ,
oldMetahash ,
oldMetasize
);
2019-01-25 23:37:57 +01:00
}
}
2025-05-19 15:43:55 +02:00
public async Task <( long Size , string MetadataHash )?> GetMetadataHashAndSizeForFile ( long fileid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
m_selectfilemetadatahashandsizeCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FileId" , fileid );
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
using ( var rd = await m_selectfilemetadatahashandsizeCommand . ExecuteReaderAsync ())
if ( await rd . ReadAsync ())
return (
rd . ConvertValueToInt64 ( 0 ),
rd . ConvertValueToString ( 1 ) ?? throw new InvalidOperationException ( "Metadata hash is null" )
);
2019-01-25 23:37:57 +01:00
return null ;
}
2025-05-19 15:43:55 +02:00
public async Task < string? > GetFileHash ( long fileid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var r = await m_selectfileHashCommand
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FileId" , fileid )
2025-05-19 15:43:55 +02:00
. ExecuteScalarAsync ( m_logQueries , null );
2019-01-25 23:37:57 +01:00
if ( r == null || r == DBNull . Value )
return null ;
2019-08-05 20:14:05 -04:00
2019-01-25 23:37:57 +01:00
return r . ToString ();
}
2019-08-05 20:14:05 -04:00
public override void Dispose ()
2025-05-19 15:43:55 +02:00
{
2025-05-23 08:20:00 +02:00
this . DisposeAsync (). Await ();
2025-05-19 15:43:55 +02:00
}
public override async Task DisposeAsync ()
2019-01-25 23:37:57 +01:00
{
2025-04-06 08:41:59 +02:00
if (! string . IsNullOrWhiteSpace ( m_tempDeletedBlockTable ))
try
{
using ( var cmd = m_connection . CreateCommand ())
2025-05-19 15:43:55 +02:00
await cmd . ExecuteNonQueryAsync ( $@"DROP TABLE ""{m_tempDeletedBlockTable}""" );
2025-04-06 08:41:59 +02:00
}
catch ( Exception ex )
{
Logging . Log . WriteWarningMessage ( LOGTAG , "DropTempTableFailed" , ex , "Failed to drop temporary table {0}: {1}" , m_tempDeletedBlockTable , ex . Message );
}
2025-05-23 08:20:00 +02:00
await base . DisposeAsync ();
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task < long > GetLastWrittenDBlockVolumeSize ()
2025-04-23 15:39:32 +02:00
{
2025-05-19 14:40:00 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2025-05-19 15:43:55 +02:00
return await cmd . SetCommandAndParameters ( @"
SELECT ""Size""
FROM ""RemoteVolume""
WHERE
""State"" = @State
AND ""Type"" = @Type
ORDER BY ""ID"" DESC
LIMIT 1
" )
2025-04-23 15:39:32 +02:00
. SetParameterValue ( "@State" , RemoteVolumeState . Uploaded . ToString ())
. SetParameterValue ( "@Type" , RemoteVolumeType . Blocks . ToString ())
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
2025-04-23 15:39:32 +02:00
}
2025-05-19 15:43:55 +02:00
private async Task < long > GetPreviousFilesetID ( SqliteCommand cmd )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
return await GetPreviousFilesetID ( cmd , OperationTimestamp , m_filesetId );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-19 15:43:55 +02:00
private async Task < long > GetPreviousFilesetID ( SqliteCommand cmd , DateTime timestamp , long filesetid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
return await cmd
. SetTransaction ( m_rtr )
. SetCommandAndParameters ( @"
SELECT ""ID""
FROM ""Fileset""
WHERE
""Timestamp"" < @Timestamp
AND ""ID"" != @FilesetId
ORDER BY ""Timestamp"" DESC
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Timestamp" , Library . Utility . Utility . NormalizeDateTimeToEpochSeconds ( timestamp ))
. SetParameterValue ( "@FilesetId" , filesetid )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
internal async Task < Tuple < long , long >> GetLastBackupFileCountAndSize ()
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var lastFilesetId = await cmd . ExecuteScalarInt64Async ( @"
SELECT ""ID""
FROM ""Fileset""
ORDER BY ""Timestamp"" DESC
LIMIT 1
" );
var count = await cmd . SetCommandAndParameters ( @"
SELECT COUNT(*)
FROM ""FileLookup""
INNER JOIN ""FilesetEntry""
ON ""FileLookup"".""ID"" = ""FilesetEntry"".""FileID""
WHERE
""FilesetEntry"".""FilesetID"" = @FilesetId
AND ""FileLookup"".""BlocksetID"" NOT IN (
@FolderBlocksetId,
@SymlinkBlocksetId
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , lastFilesetId )
. SetParameterValue ( "@FolderBlocksetId" , FOLDER_BLOCKSET_ID )
. SetParameterValue ( "@SymlinkBlocksetId" , SYMLINK_BLOCKSET_ID )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
var size = await cmd . SetCommandAndParameters ( @"
SELECT SUM(""Blockset"".""Length"")
FROM
""FileLookup"",
""FilesetEntry"",
""Blockset""
WHERE
""FileLookup"".""ID"" = ""FilesetEntry"".""FileID"" AND
""FileLookup"".""BlocksetID"" = ""Blockset"".""ID"" AND
""FilesetEntry"".""FilesetID"" = @FilesetId AND
""FileLookup"".""BlocksetID"" NOT IN (
@FolderBlocksetId,
@SymlinkBlocksetId
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , lastFilesetId )
. SetParameterValue ( "@FolderBlocksetId" , FOLDER_BLOCKSET_ID )
. SetParameterValue ( "@SymlinkBlocksetId" , SYMLINK_BLOCKSET_ID )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
2019-01-25 23:37:57 +01:00
return new Tuple < long , long >( count , size );
}
}
2025-05-19 15:43:55 +02:00
internal async Task UpdateChangeStatistics ( BackupResults results )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var prevFileSetId = await GetPreviousFilesetID ( cmd );
2025-05-20 06:07:45 +02:00
await ChangeStatistics . UpdateChangeStatistics ( cmd , results , m_filesetId , prevFileSetId );
2019-01-25 23:37:57 +01:00
}
}
/// <summary>
2025-05-19 13:19:24 +02:00
/// Populates FilesetEntry table with files from previous fileset, which aren't
2019-01-25 23:37:57 +01:00
/// yet part of the new fileset, and which aren't on the (optional) list of <c>deleted</c> paths.
/// </summary>
/// <param name="transaction">Transaction</param>
/// <param name="deleted">List of deleted paths, or null</param>
2025-05-19 15:43:55 +02:00
public async Task AppendFilesFromPreviousSet ( IEnumerable < string >? deleted = null )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await AppendFilesFromPreviousSet ( deleted , m_filesetId , - 1 , OperationTimestamp );
2019-01-25 23:37:57 +01:00
}
/// <summary>
2025-05-19 13:19:24 +02:00
/// Populates FilesetEntry table with files from previous fileset, which aren't
2019-01-25 23:37:57 +01:00
/// yet part of the new fileset, and which aren't on the (optional) list of <c>deleted</c> paths.
/// </summary>
/// <param name="transaction">Transaction</param>
/// <param name="deleted">List of deleted paths, or null</param>
/// <param name="filesetid">Current file-set ID</param>
/// <param name="prevId">Source file-set ID</param>
/// <param name="timestamp">If <c>filesetid</c> == -1, used to locate previous file-set</param>
2025-05-19 15:43:55 +02:00
public async Task AppendFilesFromPreviousSet ( IEnumerable < string >? deleted , long filesetid , long prevId , DateTime timestamp )
2019-01-25 23:37:57 +01:00
{
2019-08-05 20:14:05 -04:00
using ( var cmd = m_connection . CreateCommand ())
using ( var cmdDelete = m_connection . CreateCommand ())
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
long lastFilesetId = prevId < 0 ? await GetPreviousFilesetID ( cmd , timestamp , filesetid ) : prevId ;
await cmd . SetTransaction ( m_rtr )
. SetCommandAndParameters ( @"
INSERT INTO ""FilesetEntry"" (
""FilesetID"",
""FileID"",
""Lastmodified""
)
SELECT
@CurrentFilesetId AS ""FilesetID"",
""FileID"",
""Lastmodified""
FROM (
SELECT DISTINCT
""FilesetID"",
""FileID"",
""Lastmodified""
FROM ""FilesetEntry""
WHERE
""FilesetID"" = @PreviousFilesetId
AND ""FileID"" NOT IN (
SELECT ""FileID""
FROM ""FilesetEntry""
WHERE ""FilesetID"" = @CurrentFilesetId
)
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@CurrentFilesetId" , filesetid )
. SetParameterValue ( "@PreviousFilesetId" , lastFilesetId )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ( m_logQueries );
2019-01-25 23:37:57 +01:00
if ( deleted != null )
{
2025-05-19 15:43:55 +02:00
using var tmplist = await TemporaryDbValueList . CreateAsync ( this , deleted );
await ( await cmdDelete . SetTransaction ( m_rtr )
. SetCommandAndParameters ( @"
DELETE FROM ""FilesetEntry""
WHERE
""FilesetID"" = @FilesetId
AND ""FileID"" IN (
SELECT ""ID""
FROM ""File""
WHERE ""Path"" IN (@Paths)
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , filesetid )
2025-05-19 15:43:55 +02:00
. ExpandInClauseParameterAsync ( "@Paths" , tmplist ))
. ExecuteNonQueryAsync ( m_logQueries );
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
await m_rtr . CommitAsync ();
2019-01-25 23:37:57 +01:00
}
}
/// <summary>
2025-05-19 13:19:24 +02:00
/// Populates FilesetEntry table with files from previous fileset, which aren't
/// yet part of the new fileset, and which aren't excluded by the (optional) exclusion
2019-01-25 23:37:57 +01:00
/// predicate.
/// </summary>
/// <param name="transaction">Transaction</param>
/// <param name="exclusionPredicate">Optional exclusion predicate (true = exclude file)</param>
2025-05-19 15:43:55 +02:00
public async Task AppendFilesFromPreviousSetWithPredicate ( Func < string , long , bool > exclusionPredicate )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await AppendFilesFromPreviousSetWithPredicate ( exclusionPredicate , m_filesetId , - 1 , OperationTimestamp );
2019-01-25 23:37:57 +01:00
}
/// <summary>
2025-05-19 13:19:24 +02:00
/// Populates FilesetEntry table with files from previous fileset, which aren't
/// yet part of the new fileset, and which aren't excluded by the (optional) exclusion
2019-01-25 23:37:57 +01:00
/// predicate.
/// </summary>
/// <param name="transaction">Transaction</param>
/// <param name="exclusionPredicate">Optional exclusion predicate (true = exclude file)</param>
/// <param name="fileSetId">Current fileset ID</param>
/// <param name="prevFileSetId">Source fileset ID</param>
/// <param name="timestamp">If <c>prevFileSetId</c> == -1, used to locate previous fileset</param>
2025-05-19 15:43:55 +02:00
public async Task AppendFilesFromPreviousSetWithPredicate ( Func < string , long , bool > exclusionPredicate , long fileSetId , long prevFileSetId , DateTime timestamp )
2019-01-25 23:37:57 +01:00
{
2022-02-26 20:24:14 -07:00
if ( exclusionPredicate == null )
{
2025-05-19 15:43:55 +02:00
await AppendFilesFromPreviousSet ( null , fileSetId , prevFileSetId , timestamp );
2020-08-18 11:43:23 +02:00
return ;
}
2019-01-25 23:37:57 +01:00
using ( var cmd = m_connection . CreateCommand ())
2020-08-17 17:27:45 +02:00
using ( var cmdDelete = m_connection . CreateCommand ())
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
long lastFilesetId = prevFileSetId < 0 ? await GetPreviousFilesetID ( cmd , timestamp , fileSetId ) : prevFileSetId ;
2020-08-17 18:06:38 +02:00
2020-08-18 11:43:23 +02:00
// copy entries from previous file set into a temporary table, except those file IDs already added by the current backup
2022-02-26 20:24:14 -07:00
var tempFileSetTable = "FilesetEntry-" + Library . Utility . Utility . ByteArrayAsHexString ( Guid . NewGuid (). ToByteArray ());
2025-05-19 15:43:55 +02:00
await cmd
. SetTransaction ( m_rtr )
. SetCommandAndParameters ( $@"
CREATE TEMPORARY TABLE ""{tempFileSetTable}"" AS
SELECT
""FileID"",
""Lastmodified""
FROM (
SELECT DISTINCT
""FilesetID"",
""FileID"",
""Lastmodified""
FROM ""FilesetEntry""
WHERE
""FilesetID"" = @PreviousFilesetId
AND ""FileID"" NOT IN (
SELECT ""FileID""
FROM ""FilesetEntry""
WHERE ""FilesetID"" = @CurrentFilesetId
)
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@PreviousFilesetId" , lastFilesetId )
. SetParameterValue ( "@CurrentFilesetId" , fileSetId )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ();
2020-08-17 18:06:38 +02:00
2025-05-19 13:19:24 +02:00
// now we need to remove, from the above, any entries that were enumerated by the
2020-08-18 11:43:23 +02:00
// UNC-driven backup
2025-05-19 14:40:00 +02:00
cmdDelete . SetTransaction ( m_rtr )
2025-05-19 15:43:55 +02:00
. SetCommandAndParameters ( $@"
DELETE FROM ""{tempFileSetTable}""
WHERE ""FileID"" = @FileId
" );
2019-01-25 23:37:57 +01:00
2020-08-18 11:43:23 +02:00
// enumerate files from new temporary file set, and remove any entries handled by UNC
2025-05-19 15:43:55 +02:00
cmd . SetCommandAndParameters ( $@"
SELECT
f.""Path"",
fs.""FileID"",
fs.""Lastmodified"",
COALESCE(bs.""Length"", -1)
FROM (
SELECT DISTINCT
""FileID"",
""Lastmodified""
FROM ""{tempFileSetTable}""
) AS fs
LEFT JOIN ""File"" AS f
ON fs.""FileID"" = f.""ID""
LEFT JOIN ""Blockset"" AS bs
ON f.""BlocksetID"" = bs.""ID"";
" );
await foreach ( var row in cmd . ExecuteReaderEnumerableAsync ())
2019-01-25 23:37:57 +01:00
{
2025-04-03 11:31:36 +02:00
var path = row . ConvertValueToString ( 0 ) ?? throw new Exception ( "Unexpected null value for path" );
var size = row . ConvertValueToInt64 ( 3 );
2020-08-17 18:06:38 +02:00
2020-08-17 17:27:45 +02:00
if ( exclusionPredicate ( path , size ))
2025-05-19 15:43:55 +02:00
await cmdDelete . SetParameterValue ( "@FileId" , row . ConvertValueToInt64 ( 1 ))
. ExecuteNonQueryAsync ();
2019-01-25 23:37:57 +01:00
}
2020-08-18 11:43:23 +02:00
// now copy the temporary table into the FileSetEntry table
2025-05-19 15:43:55 +02:00
await cmd . SetCommandAndParameters ( $@"
INSERT INTO ""FilesetEntry"" (
""FilesetID"",
""FileID"",
""Lastmodified""
)
SELECT
@FilesetId,
""FileID"",
""Lastmodified""
FROM ""{tempFileSetTable}""
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , fileSetId )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ();
2020-08-18 11:43:23 +02:00
2025-05-19 15:43:55 +02:00
await m_rtr . CommitAsync ();
2019-01-25 23:37:57 +01:00
}
2022-02-26 20:24:14 -07:00
}
2019-01-25 23:37:57 +01:00
/// <summary>
/// Creates a timestamped backup operation to correctly associate the fileset with the time it was created.
/// </summary>
/// <param name="volumeid">The ID of the fileset volume to update</param>
/// <param name="timestamp">The timestamp of the operation to create</param>
/// <param name="transaction">An optional external transaction</param>
2025-05-19 15:43:55 +02:00
public override async Task < long > CreateFileset ( long volumeid , DateTime timestamp )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
return m_filesetId = await base . CreateFileset ( volumeid , timestamp );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-19 15:43:55 +02:00
public async Task < IEnumerable < string >> GetTemporaryFilelistVolumeNames ( bool latestOnly )
2022-02-26 20:24:14 -07:00
{
2025-05-19 15:43:55 +02:00
var incompleteFilesetIDs = GetIncompleteFilesets (). OrderBy ( x => x . Value ). Select ( x => x . Key );
2022-02-26 20:24:14 -07:00
2025-05-19 15:43:55 +02:00
if (! await incompleteFilesetIDs . AnyAsync ())
return [];
2022-02-26 20:24:14 -07:00
if ( latestOnly )
2025-05-19 15:43:55 +02:00
incompleteFilesetIDs = new long [] { await incompleteFilesetIDs . LastAsync () }. ToAsyncEnumerable ();
2022-02-26 20:24:14 -07:00
var volumeNames = new List < string >();
2025-05-19 15:43:55 +02:00
await foreach ( var filesetID in incompleteFilesetIDs )
volumeNames . Add (( await GetRemoteVolumeFromFilesetID ( filesetID )). Name );
2022-02-26 20:24:14 -07:00
return volumeNames ;
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async IAsyncEnumerable < string > GetMissingIndexFiles ()
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
using var cmd = m_connection . CreateCommand ( m_rtr )
2025-05-19 15:43:55 +02:00
. SetCommandAndParameters ( @"
SELECT ""Name""
FROM ""RemoteVolume""
WHERE
""Type"" = @Type
AND NOT ""ID"" IN (
SELECT ""BlockVolumeID""
FROM ""IndexBlockLink""
)
AND ""State"" IN (@States)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Type" , RemoteVolumeType . Blocks . ToString ())
2025-05-22 13:24:00 +02:00
. ExpandInClauseParameterAsync ( "@States" , [
2025-05-19 15:43:55 +02:00
RemoteVolumeState . Uploaded . ToString (),
RemoteVolumeState . Verified . ToString ()
]);
2025-04-03 11:31:36 +02:00
2025-05-19 15:43:55 +02:00
using ( var rd = await cmd . ExecuteReaderAsync ())
while ( await rd . ReadAsync ())
2025-04-03 11:31:36 +02:00
yield return rd . ConvertValueToString ( 0 ) ?? throw new Exception ( "Unexpected null value for volume name" );
2019-01-25 23:37:57 +01:00
}
2019-08-05 20:14:05 -04:00
2025-05-19 15:43:55 +02:00
public async Task MoveBlockToVolume ( string blockkey , long size , long sourcevolumeid , long targetvolumeid )
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var c = await cmd . SetCommandAndParameters ( @"
UPDATE ""Block""
SET ""VolumeID"" = @NewVolumeId
WHERE
""Hash"" = @Hash
AND ""Size"" = @Size
AND ""VolumeID"" = @PreviousVolumeId
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@NewVolumeId" , targetvolumeid )
. SetParameterValue ( "@Hash" , blockkey )
. SetParameterValue ( "@Size" , size )
. SetParameterValue ( "@PreviousVolumeId" , sourcevolumeid )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ();
2019-01-25 23:37:57 +01:00
if ( c != 1 )
2025-03-13 23:06:32 +01:00
throw new Exception ( $"Failed to move block {blockkey}:{size} from volume {sourcevolumeid}, count: {c}" );
2019-01-25 23:37:57 +01:00
}
}
2025-05-19 15:43:55 +02:00
public async Task SafeDeleteRemoteVolume ( string name )
2019-01-25 23:37:57 +01:00
{
2025-05-22 15:28:35 +02:00
var volumeid = await GetRemoteVolumeID ( name );
2019-01-25 23:37:57 +01:00
2025-05-19 14:40:00 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var c = await cmd . SetCommandAndParameters ( @"
SELECT COUNT(*)
FROM ""Block""
WHERE ""VolumeID"" = @VolumeId
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@VolumeId" , volumeid )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async (- 1 );
2019-01-25 23:37:57 +01:00
if ( c != 0 )
2025-03-13 23:06:32 +01:00
throw new Exception ( $"Failed to safe-delete volume {name}, blocks: {c}" );
2019-01-25 23:37:57 +01:00
2025-05-19 15:43:55 +02:00
await RemoveRemoteVolume ( name );
2019-08-05 20:14:05 -04:00
}
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task < string []> GetBlocklistHashes ( string name )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
var volumeid = GetRemoteVolumeID ( name );
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
// Grab the strings and return as array to avoid concurrent access to the IEnumerable
2025-05-19 15:43:55 +02:00
cmd . SetCommandAndParameters ( @"
SELECT DISTINCT ""Block"".""Hash""
FROM ""Block""
WHERE
""Block"".""VolumeID"" = @VolumeId
AND ""Block"".""Hash"" IN (
SELECT ""Hash""
FROM ""BlocklistHash""
)
" )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@VolumeId" , volumeid );
2025-05-19 15:43:55 +02:00
return await cmd . ExecuteReaderEnumerableAsync ()
2025-04-03 11:31:36 +02:00
. Select ( x => x . ConvertValueToString ( 0 ) ?? throw new Exception ( "Unexpected null value for blocklist hash" ))
2025-05-19 15:43:55 +02:00
. ToArrayAsync ();
2019-08-05 20:14:05 -04:00
}
2019-01-25 23:37:57 +01:00
}
2025-05-19 15:43:55 +02:00
public async Task < string? > GetFirstPath ()
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var v0 = await cmd . ExecuteScalarAsync ( @"
SELECT ""Path""
FROM ""File""
ORDER BY LENGTH(""Path"") DESC
LIMIT 1
" );
2019-01-25 23:37:57 +01:00
if ( v0 == null || v0 == DBNull . Value )
return null ;
return v0 . ToString ();
}
}
2019-08-05 20:14:05 -04:00
2019-01-25 23:37:57 +01:00
/// <summary>
/// Retrieves change journal data for file set
/// </summary>
/// <param name="fileSetId">Fileset-ID</param>
2025-05-19 15:43:55 +02:00
public async IAsyncEnumerable < Interface . USNJournalDataEntry > GetChangeJournalData ( long fileSetId )
2019-01-25 23:37:57 +01:00
{
var data = new List < Interface . USNJournalDataEntry >();
2025-05-19 15:43:55 +02:00
using var cmd = m_connection . CreateCommand ( @"
SELECT
""VolumeName"",
""JournalID"",
""NextUSN"",
""ConfigHash""
FROM ""ChangeJournalData""
WHERE ""FilesetID"" = @FilesetId
" )
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@FilesetId" , fileSetId );
2025-05-19 15:43:55 +02:00
using ( var rd = await cmd . ExecuteReaderAsync ())
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
while ( await rd . ReadAsync ())
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
yield return new Interface . USNJournalDataEntry
2019-01-25 23:37:57 +01:00
{
Volume = rd . ConvertValueToString ( 0 ),
JournalId = rd . ConvertValueToInt64 ( 1 ),
NextUsn = rd . ConvertValueToInt64 ( 2 ),
ConfigHash = rd . ConvertValueToString ( 3 )
2025-05-19 15:43:55 +02:00
};
2019-01-25 23:37:57 +01:00
}
}
}
/// <summary>
/// Adds NTFS change journal data for file set and volume
/// </summary>
/// <param name="data">Data to add</param>
/// <param name="transaction">An optional external transaction</param>
2025-05-19 15:43:55 +02:00
public async Task CreateChangeJournalData ( IEnumerable < Interface . USNJournalDataEntry > data )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
foreach ( var entry in data )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
using ( var cmd = m_connection . CreateCommand ( m_rtr ))
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
var c = await cmd . SetCommandAndParameters ( @"
INSERT INTO ""ChangeJournalData"" (
""FilesetID"",
""VolumeName"",
""JournalID"",
""NextUSN"",
""ConfigHash""
)
VALUES (
@FilesetId,
@VolumeName,
@JournalId,
@NextUsn,
@ConfigHash
);
" )
. SetParameterValue ( "@FilesetId" , m_filesetId )
. SetParameterValue ( "@VolumeName" , entry . Volume )
. SetParameterValue ( "@JournalId" , entry . JournalId )
. SetParameterValue ( "@NextUsn" , entry . NextUsn )
. SetParameterValue ( "@ConfigHash" , entry . ConfigHash )
. ExecuteNonQueryAsync ();
2019-01-25 23:37:57 +01:00
2025-05-19 14:40:00 +02:00
if ( c != 1 )
throw new Exception ( "Unable to add change journal entry" );
}
2019-01-25 23:37:57 +01:00
}
2025-05-19 14:40:00 +02:00
2025-05-19 15:43:55 +02:00
await m_rtr . CommitAsync ();
2019-01-25 23:37:57 +01:00
}
/// <summary>
/// Adds NTFS change journal data for file set and volume
/// </summary>
/// <param name="data">Data to add</param>
/// <param name="fileSetId">Existing file set to update</param>
/// <param name="transaction">An optional external transaction</param>
2025-05-19 15:43:55 +02:00
public async Task UpdateChangeJournalData ( IEnumerable < Interface . USNJournalDataEntry > data , long fileSetId )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
foreach ( var entry in data )
2019-01-25 23:37:57 +01:00
{
2025-05-19 14:40:00 +02:00
using ( var cmd = m_connection . CreateCommand ())
2019-01-25 23:37:57 +01:00
{
2025-05-19 15:43:55 +02:00
await cmd . SetCommandAndParameters ( @"
UPDATE ""ChangeJournalData""
SET ""NextUSN"" = @NextUsn
WHERE
""FilesetID"" = @FilesetId
AND ""VolumeName"" = @VolumeName
AND ""JournalID"" = @JournalId;
" )
2025-05-19 14:40:00 +02:00
. SetTransaction ( m_rtr )
. SetParameterValue ( "@NextUsn" , entry . NextUsn )
. SetParameterValue ( "@FilesetId" , fileSetId )
. SetParameterValue ( "@VolumeName" , entry . Volume )
. SetParameterValue ( "@JournalId" , entry . JournalId )
2025-05-19 15:43:55 +02:00
. ExecuteNonQueryAsync ();
2019-08-05 20:14:05 -04:00
}
}
2025-05-19 14:40:00 +02:00
2025-05-19 15:43:55 +02:00
await m_rtr . CommitAsync ();
2019-08-05 20:14:05 -04:00
}
2024-10-31 15:30:25 +01:00
/// <summary>
/// Checks if a blocklist hash is known
/// </summary>
/// <param name="hash">The hash to check</param>
/// <param name="transaction">An optional external transaction</param>
/// <returns>True if the hash is known, false otherwise</returns>
2025-05-19 15:43:55 +02:00
public async Task < bool > IsBlocklistHashKnown ( string hash )
2024-10-31 15:30:25 +01:00
{
2025-05-19 15:43:55 +02:00
var res = await m_getfirstfilesetwithblockinblockset
. SetTransaction ( m_rtr )
2025-04-03 11:31:36 +02:00
. SetParameterValue ( "@Hash" , hash )
2025-05-19 15:43:55 +02:00
. ExecuteScalarInt64Async ();
2024-11-04 21:46:12 +01:00
if ( res != - 1 && res != m_filesetId )
2024-10-31 15:30:25 +01:00
return true ;
else
return ! m_blocklistHashes . Add ( hash );
}
2019-01-25 23:37:57 +01:00
}
}