Drain code

This is the drain code copied from k8s.io/kubernetes, after the extraction in https://github.com/kubernetes/kubernetes/pull/80045/files

Once we are on that version of k/k (1.16), we can replace with the upstream version.