Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
SnowflakeUnloadFunction
public static async Task<HttpResponseMessage> Run(
[HttpTrigger(AuthorizationLevel.Function, "post", Route = "v1/unload")]
HttpRequestMessage req,
ILogger logger,
ExecutionContext context)
{
ServiceProvider serviceProvider = Initializer.Initialize(context);
string body = await req.Content.ReadAsStringAsync().ConfigureAwait(false);
UnloadCommand command = JsonConvert.DeserializeObject<UnloadCommand>(body);
if (!command.Validate(out string message))
{
return req.CreateResponse(HttpStatusCode.BadRequest, message);
}
IConfiguration config = serviceProvider.GetService<IConfiguration>();
var client = new SnowflakeClient(config["ConnectionString"]);
try
{
client.Unload(command.Stage, command.Query, command.FilePrefix, command.Warehouse, command.Database, command.Schema, command.SingleFile, command.Overwrite);
}
catch (Exception e)
{
logger.LogError(e, "Error processing request");
return req.CreateResponse(HttpStatusCode.InternalServerError, e.Message);
}
return req.CreateResponse(HttpStatusCode.OK);
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment