Added some comments and changed operation type to PartitionAware to ensure migration safety.