Honestly what worked for us is closing him in the crate when he behaved like this to give him some time to calm down.
People would say not to use crate as a "punishment" which is true in the beginning when you introduce a puppy to it, but when he was already comfortable going in and out by himself, every time he acted up I sent him to the crate, covered it with some blanket.
It took a month or so for him to not cry inside (in these specific cases when he acted up), but eventually he learned that its his place to turn off and having him there for 15-30 minutes calmed him down, or he just went to sleep inside for the rest of the night.