Dustin Johnson and Paulina Gretzky's 1-year-old son is getting lessons on how to be hockey's next biggest thing ... from grandpa Wayne!!!

Paulina and D.J. posted the adorable vid of Rivers Johnson -- their youngest child -- with The Great One ... and we gotta say, it's hard not to see the resemblance.

Wayne and the little Gretzky did some stick handling drills ... with Rivers even gettin' a few past Grandpa!!!

Of course, Dustin and Paulina's kids (they have a 4-year-old son as well) have some tough choices to make when it comes to sports ... Dad is a pretty good golfer too.