Return a Task From BeginExecuteNonQuery

Blog as notepad time. Just a little reminder for myself on how to return the result from BeginExecuteNonQuery as a Task<int>

public Task<int> Save(string value)
{
    var taskCompletionSource = new TaskCompletionSource<int>();

    var connection = new SqlConnection(connectionString);
    connection.Open();
    var command = new SqlCommand("uspSaveSomeValue", connection)
    {
        CommandType = CommandType.StoredProcedure
    };
    command.Parameters.AddWithValue("@myparam", value);
    command.BeginExecuteNonQuery(asyncResult =>
    {
        var result = command.EndExecuteNonQuery(asyncResult);
        command.Dispose();
        connection.Dispose();
        taskCompletionSource.SetResult(result);
    }, null);

    return taskCompletionSource.Task;
}

If you know a better way, please comment below.

Yes, yes, I know, but I’m not working on 4.5 yet  :(

Update: Ken Egozi suggested using Task<int>.Factory.FromAsync. Of course! I’d been doing so much TaskCompletionSource manual task creation recently that I’d forgotten about this useful shortcut. Here’s a more succinct version using FromAsync:

public Task<int> Save(string value)
{
    var connection = new SqlConnection(connectionString);
    connection.Open();
    var command = new SqlCommand("uspSaveSomeValue", connection)
    {
        CommandType = CommandType.StoredProcedure
    };
    command.Parameters.AddWithValue("@myparam", value);

    return Task<int>.Factory.FromAsync(command.BeginExecuteNonQuery(), asyncResult =>
    {
        var result = command.EndExecuteNonQuery(asyncResult);
        command.Dispose();
        connection.Dispose();
        return result;
    });
}