use sampleByKey for per user sampling