Here is some quick code on how to create a partition function using inherit tables on PostgreSql, is simple but hopefully this will save some poor soul the documentation reading and for me save me some time in the future, currently things been very busy working in Ms Sql and PostgreSql doing several conversions from one DB to the other.
1a.-Create sequence
1b.- Create table
2.-Create function
3.-create trigger
-- FUNCTION:
TableSchemaNameHere.TableName_partition_insert()
-- DROP FUNCTION TableSchemaNameHere.event_partiont_insert();
CREATE SEQUENCE seq_TableSchemaNameHere.TableNameHere_TableColumn
minvalue 1 nomaxvalue increment by 1;
create table TableSchemaNameHere.Tablename
(
PKID int DEFAULT nextval('seq_TableSchemaNameHere.TableNameHere_TableColumn') NOT NULL,
insertdate timestamp(6) without time zone default now(),
time timestamp without time zone,
constraint pk_ primary key (pkid)
)
CREATE or Replace FUNCTION TableSchemaNameHere.TableName_partiont_insert()
RETURNS trigger
LANGUAGE 'plpgsql'
COST 100.0
VOLATILE NOT LEAKPROOF
AS $BODY$
declare
declare
_partition_date varchar(50);
_partition varchar(80);
_IsNewChild boolean;
_current_day int;
_first_day varchar(2) := '01';
_last_day varchar(2);
_pick_day varchar(2);
_tablename varchar(80);
begin
-- get current day
_current_day := date_part('day',new.time);
-- find last day of the month function
_last_day := date_part ('day',TableSchemaNameHere.last_day (new.time::date));
-- get YYYY_MM format
_partition_date := to_char (NEW.time,'YYYY_MM');
-- find if first or second sequence
if _current_day < 16
then
_pick_day := _first_day;
_partition_date := _partition_date||'_'|| _first_day;
else
_pick_day = _last_day;
_partition_date := _partition_date||'_'|| _last_day;
end if;
_partition := TG_TABLE_NAME || '_' || _partition_date;
_tablename := _partition;
if not exists ( select 1 from pg_tables where schemaname= TG_TABLE_SCHEMA and tablename= _partition)
then
RAISE NOTICE 'A partition has been created %',TG_TABLE_SCHEMA ||'.'|| _partition;
_IsNewChild = TRUE;
EXECUTE 'create table '|| TG_TABLE_SCHEMA ||'.'|| _partition || ' (check( to_char (time,''YYYY_MM'||'_'|| _pick_day||''') = '''|| to_char (NEW.time,'YYYY_MM')||'_'||_pick_day||''')) INHERITS ( TableSchemaNameHere.TableNamehere);';
end if;
execute 'insert into '|| TG_TABLE_SCHEMA || '.' || _partition || ' SELECT(' || TG_TABLE_SCHEMA || '.' || TG_TABLE_NAME || ' ' || quote_literal(NEW) || ').* RETURNING eventid;';
if (_IsNewChild)
then
-- Add primary key
EXECUTE format('ALTER TABLE '||TG_TABLE_SCHEMA||'.%s ADD PRIMARY KEY(pkid)', _partition);
-- Assign owner of of inherited table
EXECUTE format('ALTER TABLE '||TG_TABLE_SCHEMA||'.%s OWNER TO db_owner', _partition);
-- Add FK promo_batch_id
EXECUTE format('ALTER TABLE '||TG_TABLE_SCHEMA||'.%s ADD CONSTRAINT FK_%s FOREIGN KEY(columnId) REFERENCES TableSchemaNameHere.fkid (fkid) MATCH FULL',_tablename, _tablename);
-- Need to define indexes for inherited tables
EXECUTE format('CREATE INDEX ix_%s_EventTypeID ON '||TG_TABLE_SCHEMA||'.%s USING btree ("indexkeycolumn")', _tablename, _tablename);
end if;
RETURN NULL;
end
$BODY$;
----Create trigger
CREATE TRIGGER tr_TableName_Partition
BEFORE INSERT ON TableSchemaNameHere.TableNameHere
FOR EACH ROW EXECUTE PROCEDURE myschema.server_partition_function();
"I hear and I forget. I see and I remember. I do and I understand."
Confucius
Jorge Novo Development Solutions,Ideas, and Crazy Thoughts.
Email: ETLDEVDBA(at)gmail.com
Tweeter: @ETLDEVDBA
June 28, 2017
May 31, 2017
Get objects from AWS S3
Continue to work on pulling and pushing objects into S3 aws so here is a quick and dirty code cannibalize around...
#region Help: Introduction to the script task
/* The Script Task allows you to perform virtually any operation that can be accomplished in
* a .Net application within the context of an Integration Services control flow.
*
* Expand the other regions which have "Help" prefixes for examples of specific ways to use
* Integration Services features within this script task. */
#endregion
#region Namespaces
using System;
using System.Data;
using Microsoft.SqlServer.Dts. Runtime;
using System.Windows.Forms;
using System.IO;
using Amazon.S3;
using Amazon.S3.Model;
using Amazon;
#endregion
namespace ST_ 951ab847fc194587b642a51853cb75 36
{
/// <summary>
/// ScriptMain is the entry point class of the script. Do not change the name, attributes,
/// or parent of this class.
/// </summary>
[Microsoft.SqlServer.Dts. Tasks.ScriptTask.SSISScriptTas kEntryPointAttribute]
public partial class ScriptMain : Microsoft.SqlServer.Dts.Tasks. ScriptTask.VSTARTScriptObjectM odelBase
{
#region Help: Using Integration Services variables and parameters in a script
/* To use a variable in this script, first ensure that the variable has been added to
* either the list contained in the ReadOnlyVariables property or the list contained in
* the ReadWriteVariables property of this script task, according to whether or not your
* code needs to write to the variable. To add the variable, save this script, close this instance of
* Visual Studio, and update the ReadOnlyVariables and
* ReadWriteVariables properties in the Script Transformation Editor window.
* To use a parameter in this script, follow the same steps. Parameters are always read-only.
*
* Example of reading from a variable:
* DateTime startTime = (DateTime) Dts.Variables["System:: StartTime"].Value;
*
* Example of writing to a variable:
* Dts.Variables["User:: myStringVariable"].Value = "new value";
*
* Example of reading from a package parameter:
* int batchId = (int) Dts.Variables["$Package:: batchId"].Value;
*
* Example of reading from a project parameter:
* int batchId = (int) Dts.Variables["$Project:: batchId"].Value;
*
* Example of reading from a sensitive project parameter:
* int batchId = (int) Dts.Variables["$Project:: batchId"].GetSensitiveValue();
* */
#endregion
#region Help: Firing Integration Services events from a script
/* This script task can fire events for logging purposes.
*
* Example of firing an error event:
* Dts.Events.FireError(18, "Process Values", "Bad value", "", 0);
*
* Example of firing an information event:
* Dts.Events.FireInformation(3, "Process Values", "Processing has started", "", 0, ref fireAgain)
*
* Example of firing a warning event:
* Dts.Events.FireWarning(14, "Process Values", "No values received for input", "", 0);
* */
#endregion
#region Help: Using Integration Services connection managers in a script
/* Some types of connection managers can be used in this script task. See the topic
* "Working with Connection Managers Programatically" for details.
*
* Example of using an ADO.Net connection manager:
* object rawConnection = Dts.Connections["Sales DB"].AcquireConnection(Dts. Transaction);
* SqlConnection myADONETConnection = (SqlConnection)rawConnection;
* //Use the connection in some code here, then release the connection
* Dts.Connections["Sales DB"].ReleaseConnection( rawConnection);
*
* Example of using a File connection manager
* object rawConnection = Dts.Connections["Prices.zip"]. AcquireConnection(Dts. Transaction);
* string filePath = (string)rawConnection;
* //Use the connection in some code here, then release the connection
* Dts.Connections["Prices.zip"]. ReleaseConnection( rawConnection);
* */
#endregion
/// <summary>
/// This method is called when this script task executes in the control flow.
/// Before returning from this method, set the value of Dts.TaskResult to indicate success or failure.
/// To open Help, press F1.
/// </summary>
public void Main()
{
AppDomain.CurrentDomain. AssemblyResolve += new ResolveEventHandler( CurrentDomain_AssemblyResolve) ;
// TODO: Add your code here
GetBucketFiles();
Dts.TaskResult = (int)ScriptResults.Success;
}
private void GetBucketFiles()
{
String ak = “AmazonKey”
String sk = "AmazonSecretKey";
string BUCKET_NAME = @"BUCKETNAME";
AmazonS3Client client = new AmazonS3Client(ak,sk,Amazon.Re gionEndpoint.USEast1);
// List all objects
ListObjectsRequest listRequest = new ListObjectsRequest
{
BucketName = BUCKET_NAME,
MaxKeys = 1000
};
ListObjectsResponse listResponse;
do
{
// Get a list of objects
listResponse = client.ListObjects( listRequest);
foreach (S3Object obj in listResponse.S3Objects)
{
if (obj.Key.EndsWith(".log"))
{
String FileName = obj.Key;
GetObjectRequest request = new GetObjectRequest
{
BucketName = BUCKET_NAME,
Key = FileName
};
using (GetObjectResponse response = client.GetObject(request))
{
string dest = Path.Combine(@"DirectoryPath", FileName.Replace(@"DirectoryKey","" ));
if (!File.Exists(dest))
{
}
}
}
}
} while (listResponse.IsTruncated);
}
#region ScriptResults declaration
/// <summary>
/// This enum provides a convenient shorthand within the scope of this class for setting the
/// result of the script.
///
/// This code was generated automatically.
/// </summary>
enum ScriptResults
{
Success = Microsoft.SqlServer.Dts. Runtime.DTSExecResult.Success,
Failure = Microsoft.SqlServer.Dts. Runtime.DTSExecResult.Failure
};
#endregion
static System.Reflection.Assembly CurrentDomain_AssemblyResolve( object sender, ResolveEventArgsargs)
{
if (args.Name.Contains("AWSSDK. Core"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly. LoadFile(System.IO.Path. Combine(path,"AWSSDK.Core.dll"));
}
/*if (args.Name.Contains("AWSSDK. KeyManagementService"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly. LoadFile(System.IO.Path. Combine(path, "AWSSDK.KeyManagementService. dll"));
}*/
if (args.Name.Contains("AWSSDK. S3"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly. LoadFile(System.IO.Path. Combine(path,"AWSSDK.S3.dll"));
}
return null;
}
}
}
March 14, 2017
Streaming Data Into AWS S3 using KMS Encryption
Cheers.
#region Help: Introduction to the script task
/* The Script Task allows you to perform virtually any operation that can be accomplished in
* a .Net application within the context of an Integration Services control flow.
*
* Expand the other regions which have "Help" prefixes for examples of specific ways to use
* Integration Services features within this script task. */
#endregion
#region Namespaces
using System;
using System.Data;
using System.IO;
using System.Text;
using System.Data.OleDb;
using System.Windows.Forms;
using Amazon.KeyManagementService;
using Amazon.KeyManagementService.Model;
using Amazon.S3;
using Amazon.S3.Model;
using System.Security.Cryptography;
using System.Threading.Tasks;
using Microsoft.SqlServer.Dts.Runtime;
#endregion
namespace ST_69f7c19bae0d4e359409f7ed3cdecd0f
{
/// <summary>
/// ScriptMain is the entry point class of the script. Do not change the name, attributes,
/// or parent of this class.
/// </summary>
[Microsoft.SqlServer.Dts.Tasks.ScriptTask.SSISScriptTaskEntryPointAttribute]
public partial class ScriptMain : Microsoft.SqlServer.Dts.Tasks.ScriptTask.VSTARTScriptObjectModelBase
{
#region Help: Using Integration Services variables and parameters in a script
/* To use a variable in this script, first ensure that the variable has been added to
* either the list contained in the ReadOnlyVariables property or the list contained in
* the ReadWriteVariables property of this script task, according to whether or not your
* code needs to write to the variable. To add the variable, save this script, close this instance of
* Visual Studio, and update the ReadOnlyVariables and
* ReadWriteVariables properties in the Script Transformation Editor window.
* To use a parameter in this script, follow the same steps. Parameters are always read-only.
*
* Example of reading from a variable:
* DateTime startTime = (DateTime) Dts.Variables["System::StartTime"].Value;
*
* Example of writing to a variable:
* Dts.Variables["User::myStringVariable"].Value = "new value";
*
* Example of reading from a package parameter:
* int batchId = (int) Dts.Variables["$Package::batchId"].Value;
*
* Example of reading from a project parameter:
* int batchId = (int) Dts.Variables["$Project::batchId"].Value;
*
* Example of reading from a sensitive project parameter:
* int batchId = (int) Dts.Variables["$Project::batchId"].GetSensitiveValue();
* */
#endregion
#region Help: Firing Integration Services events from a script
/* This script task can fire events for logging purposes.
*
* Example of firing an error event:
* Dts.Events.FireError(18, "Process Values", "Bad value", "", 0);
*
* Example of firing an information event:
* Dts.Events.FireInformation(3, "Process Values", "Processing has started", "", 0, ref fireAgain)
*
* Example of firing a warning event:
* Dts.Events.FireWarning(14, "Process Values", "No values received for input", "", 0);
* */
#endregion
#region Help: Using Integration Services connection managers in a script
/* Some types of connection managers can be used in this script task. See the topic
* "Working with Connection Managers Programatically" for details.
*
* Example of using an ADO.Net connection manager:
* object rawConnection = Dts.Connections["Sales DB"].AcquireConnection(Dts.Transaction);
* SqlConnection myADONETConnection = (SqlConnection)rawConnection;
* //Use the connection in some code here, then release the connection
* Dts.Connections["Sales DB"].ReleaseConnection(rawConnection);
*
* Example of using a File connection manager
* object rawConnection = Dts.Connections["Prices.zip"].AcquireConnection(Dts.Transaction);
* string filePath = (string)rawConnection;
* //Use the connection in some code here, then release the connection
* Dts.Connections["Prices.zip"].ReleaseConnection(rawConnection);
* */
#endregion
/// <summary>
/// This method is called when this script task executes in the control flow.
/// Before returning from this method, set the value of Dts.TaskResult to indicate success or failure.
/// To open Help, press F1.
/// </summary>
public void Main()
{
try
{
/*
User::objRawData,User::strAmazonKey,User::strAmazonSecretKey,
* User::strARN,User::strBucketDirectory,
* User::strBucketName,User::strFileName
* */
AppDomain.CurrentDomain.AssemblyResolve += new ResolveEventHandler(CurrentDomain_AssemblyResolve);
//Variables
var tbl = GetJsonRawData(Dts.Variables["User::objRawData"].Value); // Table Object
int val = (int)Dts.Variables["User::intLoopController"].Value; // File Interactiption
var FileName = Dts.Variables["User::strFileName"].Value.ToString()+val.ToString()+".txt"; // File Name
String BucketDirectory = Dts.Variables["User::strBucketDirectory"].Value.ToString();// Bucket Directory
if (!String.IsNullOrEmpty(BucketDirectory))
{
FileName = BucketDirectory + "/" + FileName;
}
String AmazonKey = Dts.Variables["User::strAmazonKey"].Value.ToString();
String AmazonSecKey = Dts.Variables["User::strAmazonSecretKey"].Value.ToString();
String ARN = Dts.Variables["User::strARN"].Value.ToString();
String BucketName = Dts.Variables["User::strBucketName"].Value.ToString();
Stream input = GetRawData(tbl);
Encrypt(input, AmazonKey, AmazonSecKey, ARN,FileName,BucketName);
input = null;
Dts.Variables["User::intLoopController"].Value = val + 1;
Dts.TaskResult = (int)ScriptResults.Success;
}
catch (Exception e)
{
Dts.Events.FireError(1, e.TargetSite.ToString(), e.Message.ToString(), "", 0);
Dts.TaskResult = (int)ScriptResults.Failure;
}
}
private DataTable GetJsonRawData(object DataSetData)
{
OleDbDataAdapter oleDb = new OleDbDataAdapter();
DataTable dt = new DataTable();
oleDb.Fill(dt, DataSetData);
return dt;
}
private Stream GetRawData(DataTable dt)
{
StringBuilder rawdata = new StringBuilder();
try
{
if (dt != null)
{
foreach (DataRow dr in dt.Rows)
{
foreach (DataColumn dc in dt.Columns)
{
byte[] json = (byte[])dr[dc];
rawdata.Append(Encoding.UTF8.GetString(json));
}
rawdata.Append(Environment.NewLine);
}
}
return new MemoryStream(Encoding.UTF8.GetBytes(rawdata.ToString()));
}
catch (Exception e)
{
Dts.Events.FireError(1, e.TargetSite.ToString(), e.Message.ToString(), "", 0);
Dts.TaskResult = (int)ScriptResults.Failure;
return new MemoryStream();
}
}
/*
* //string dk = Convert.ToBase64String(dataKey.CiphertextBlob.ToArray());
//algorithm.Key = dataKey.Plaintext.ToArray();
//string k = Convert.ToBase64String(dataKey.Plaintext.ToArray());
//byte[] Iv = ASCIIEncoding.UTF8.GetBytes(System.Guid.NewGuid().ToString().Substring(0, 12));
//string siv = Convert.ToBase64String(algorithm.IV);
//byte[] iv = Encoding.UTF8.GetBytes("0123456789123456");
//algorithm.IV = iv;
**/
private void Encrypt(Stream input, string AmazonKey,string AmazonSecretKey,string arn,String s3FileName,String s3BucketName)
{
try
{
string meta_datakey = "x-amz-meta-datakey";
string meta_iv = "x-amz-meta-iv";
string meta_encoder = "x-amz-meta-encoder";
string meta_encrypt = "x-amz-meta-clientside-algorithm";
string ProxyHost = Dts.Variables["User::strProxyHost"].Value.ToString();
Int32 ProxyPort =(Int32) Dts.Variables["User::intProxyPort"].Value;
using (var output = new MemoryStream())
{
var conf = new AmazonKeyManagementServiceConfig();
conf.ProxyHost = ProxyHost;
conf.ProxyPort = ProxyPort;
conf.RegionEndpoint = Amazon.RegionEndpoint.USEast1;
using (var kms = new AmazonKeyManagementServiceClient(AmazonKey, AmazonSecretKey, conf))
using (var algorithm = Aes.Create())
{
var dataKey = kms.GenerateDataKey(new GenerateDataKeyRequest
{
KeyId = arn,
KeySpec = DataKeySpec.AES_256,
});
using (var cryptoStream = new CryptoStream(
output,
algorithm.CreateEncryptor(dataKey.Plaintext.ToArray(), algorithm.IV),
CryptoStreamMode.Write))
{
input.CopyTo(cryptoStream);
cryptoStream.FlushFinalBlock();
var s3conf = new AmazonS3Config();
s3conf.ProxyHost = ProxyHost;
s3conf.ProxyPort = ProxyPort;
s3conf.RegionEndpoint = Amazon.RegionEndpoint.USEast1;
using (var s3client = new AmazonS3Client(AmazonKey, AmazonSecretKey, s3conf))
{
PutObjectRequest request = (new PutObjectRequest()
{
BucketName = s3BucketName,
Key = s3FileName,
InputStream = output
});
request.Metadata.Add(meta_encrypt, "AES_256");
request.Metadata.Add(meta_encoder,"Base 64");
request.Metadata.Add(meta_iv,Convert.ToBase64String(algorithm.IV));
request.Metadata.Add(meta_datakey,Convert.ToBase64String(dataKey.Plaintext.ToArray()));
s3client.PutObject(request);
}
}
}
}
}
catch (Exception e)
{
Dts.Events.FireError(1, e.TargetSite.ToString(), e.Message.ToString(), "", 0);
Dts.TaskResult = (int)ScriptResults.Failure;
}
}
void Decrypt(Stream output, string AmazonKey, string AmazonSecretKey, string arn, String s3FileName, String s3BucketName)
{
try
{
using (var input = new MemoryStream())
using (var kms = new AmazonKeyManagementServiceClient(AmazonKey, AmazonSecretKey, Amazon.RegionEndpoint.USEast1))
using (var algorithm = Aes.Create())
{
var length = input.ReadByte();
var buffer = new byte[length];
input.Read(buffer, 0, length);
var decryptedData = kms.Decrypt(new DecryptRequest
{
CiphertextBlob = new MemoryStream(buffer)
});
algorithm.Key = decryptedData.Plaintext.ToArray();
var iv = algorithm.IV;
input.Read(iv, 0, iv.Length);
algorithm.IV = iv;
using (var cryptoStream = new CryptoStream(input,
algorithm.CreateDecryptor(), CryptoStreamMode.Read))
{
cryptoStream.CopyTo(output);
}
}
}
catch (Exception e)
{
Dts.Events.FireError(1, e.TargetSite.ToString(), e.Message.ToString(), "", 0);
Dts.TaskResult = (int)ScriptResults.Failure;
}
}
static System.Reflection.Assembly CurrentDomain_AssemblyResolve(object sender, ResolveEventArgs args)
{
if (args.Name.Contains("AWSSDK.Core"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly.LoadFile(System.IO.Path.Combine(path, "AWSSDK.Core.dll"));
}
if (args.Name.Contains("AWSSDK.KeyManagementService"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly.LoadFile(System.IO.Path.Combine(path, "AWSSDK.KeyManagementService.dll"));
}
if (args.Name.Contains("AWSSDK.S3"))
{
string path = @"C:\Program Files (x86)\AWS SDK for .NET\bin\Net45\";
return System.Reflection.Assembly.LoadFile(System.IO.Path.Combine(path, "AWSSDK.S3.dll"));
}
return null;
}
#region ScriptResults declaration
/// <summary>
/// This enum provides a convenient shorthand within the scope of this class for setting the
/// result of the script.
///
/// This code was generated automatically.
/// </summary>
enum ScriptResults
{
Success = Microsoft.SqlServer.Dts.Runtime.DTSExecResult.Success,
Failure = Microsoft.SqlServer.Dts.Runtime.DTSExecResult.Failure
};
#endregion
}
}
Subscribe to:
Posts (Atom)


