public WriterPropertiesBuilder CompressionLevel(ColumnPath path, int compressionLevel)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Compression_Level_By_ColumnPath(_handle.IntPtr, path.Handle.IntPtr, compressionLevel));
GC.KeepAlive(_handle);
GC.KeepAlive(path);
return(this);
}
public WriterPropertiesBuilder DisableStatistics(ColumnPath path)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Disable_Statistics_By_ColumnPath(_handle.IntPtr, path.Handle.IntPtr));
GC.KeepAlive(_handle);
GC.KeepAlive(path);
return(this);
}
public WriterPropertiesBuilder Encoding(ColumnPath path, Encoding encoding)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Encoding_By_ColumnPath(_handle.IntPtr, path.Handle.IntPtr, encoding));
GC.KeepAlive(_handle);
GC.KeepAlive(path);
return(this);
}
public WriterPropertiesBuilder Encryption(FileEncryptionProperties?fileEncryptionProperties)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Encryption(_handle.IntPtr, fileEncryptionProperties?.Handle.IntPtr ?? IntPtr.Zero));
GC.KeepAlive(_handle);
GC.KeepAlive(fileEncryptionProperties);
return(this);
}
public WriterPropertiesBuilder EnableDictionary(ColumnPath path)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Enable_Dictionary_By_ColumnPath(_handle.IntPtr, path.Handle.IntPtr));
GC.KeepAlive(_handle);
GC.KeepAlive(path);
return(this);
}
public FileEncryptionPropertiesBuilder EncryptedColumns(ColumnEncryptionProperties[] columnEncryptionProperties)
{
var handles = columnEncryptionProperties.Select(p => p.Handle.IntPtr).ToArray();
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Encrypted_Columns(_handle.IntPtr, handles, handles.Length));
GC.KeepAlive(_handle);
GC.KeepAlive(columnEncryptionProperties);
return(this);
}
public FileDecryptionPropertiesBuilder AadPrefixVerifier(AadPrefixVerifier aadPrefixVerifier)
{
var gcHandle = aadPrefixVerifier?.CreateGcHandle() ?? IntPtr.Zero;
try
{
ExceptionInfo.Check(FileDecryptionPropertiesBuilder_Aad_Prefix_Verifier(
_handle.IntPtr,
gcHandle,
ParquetSharp.AadPrefixVerifier.FreeGcHandleCallback,
ParquetSharp.AadPrefixVerifier.VerifyFuncCallback));
}
catch
{
ParquetSharp.AadPrefixVerifier.FreeGcHandleCallback(gcHandle);
throw;
}
GC.KeepAlive(_handle);
GC.KeepAlive(aadPrefixVerifier);
return(this);
}
public FileDecryptionPropertiesBuilder KeyRetriever(DecryptionKeyRetriever keyRetriever)
{
var gcHandle = keyRetriever?.CreateGcHandle() ?? IntPtr.Zero;
try
{
ExceptionInfo.Check(FileDecryptionPropertiesBuilder_Key_Retriever(
_handle.IntPtr,
gcHandle,
DecryptionKeyRetriever.FreeGcHandleCallback,
DecryptionKeyRetriever.GetKeyFuncCallback));
}
catch
{
DecryptionKeyRetriever.FreeGcHandleCallback(gcHandle);
throw;
}
GC.KeepAlive(_handle);
GC.KeepAlive(keyRetriever);
return(this);
}
public WriterPropertiesBuilder WriteBatchSize(long writeBatchSize)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Write_Batch_Size(_handle.IntPtr, writeBatchSize));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder Version(ParquetVersion version)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Version(_handle.IntPtr, version));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder MaxRowGroupLength(long maxRowGroupLength)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Max_Row_Group_Length(_handle.IntPtr, maxRowGroupLength));
return(this);
}
public void Close()
{
ExceptionInfo.Check(RowGroupWriter_Close(_handle));
}
public FileEncryptionPropertiesBuilder SetPlaintextFooter()
{
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Set_Plaintext_Footer(_handle.IntPtr));
GC.KeepAlive(_handle);
return(this);
}
public FileEncryptionPropertiesBuilder AadPrefix(string aadPrefix)
{
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Aad_Prefix(_handle.IntPtr, aadPrefix));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder EnableDictionary()
{
ExceptionInfo.Check(WriterPropertiesBuilder_Enable_Dictionary(_handle.IntPtr));
return(this);
}
public ColumnEncryptionPropertiesBuilder KeyId(string keyId)
{
ExceptionInfo.Check(ColumnEncryptionPropertiesBuilder_Key_Id(_handle.IntPtr, keyId));
GC.KeepAlive(_handle);
return(this);
}
public FileEncryptionPropertiesBuilder Algorithm(ParquetCipher parquetCipher)
{
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Algorithm(_handle.IntPtr, parquetCipher));
GC.KeepAlive(_handle);
return(this);
}
public FileEncryptionPropertiesBuilder FooterKeyMetadata(string footerKeyMetadata)
{
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Footer_Key_Metadata(_handle.IntPtr, footerKeyMetadata));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder CompressionLevel(int compressionLevel)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Compression_Level(_handle.IntPtr, compressionLevel));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder Compression(string path, Compression codec)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Compression_By_Path(_handle.IntPtr, path, codec));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder EnableStatistics(string path)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Enable_Statistics_By_Path(_handle.IntPtr, path));
GC.KeepAlive(_handle);
return(this);
}
// Dictionary enable/disable
public WriterPropertiesBuilder DisableDictionary()
{
ExceptionInfo.Check(WriterPropertiesBuilder_Disable_Dictionary(_handle.IntPtr));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder DisableDictionary(string path)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Disable_Dictionary_By_Path(_handle.IntPtr, path));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder Encoding(Encoding encoding)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Encoding(_handle.IntPtr, encoding));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder CreatedBy(string createdBy)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Created_By(_handle.IntPtr, createdBy));
GC.KeepAlive(_handle);
return(this);
}
public FileEncryptionPropertiesBuilder DisableAadPrefixStorage()
{
ExceptionInfo.Check(FileEncryptionPropertiesBuilder_Disable_Aad_Prefix_Storage(_handle.IntPtr));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder EnableStatistics()
{
ExceptionInfo.Check(WriterPropertiesBuilder_Enable_Statistics(_handle.IntPtr));
GC.KeepAlive(_handle);
return(this);
}
public WriterPropertiesBuilder DataPagesize(long pageSize)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Data_Pagesize(_handle.IntPtr, pageSize));
GC.KeepAlive(_handle);
return(this);
}
/// <summary>
/// Close the file writer as well any column or group writers that are still opened.
/// This is the recommended way of closing Parquet files, rather than relying on the Dispose() method,
/// as the latter will gobble exceptions.
/// </summary>
public void Close()
{
ExceptionInfo.Check(ParquetFileWriter_Close(_handle.IntPtr));
GC.KeepAlive(_handle);
}
public WriterPropertiesBuilder DictionaryPagesizeLimit(long dictionaryPagesizeLimit)
{
ExceptionInfo.Check(WriterPropertiesBuilder_Dictionary_Pagesize_Limit(_handle.IntPtr, dictionaryPagesizeLimit));
GC.KeepAlive(_handle);
return(this);
}
