surrogate key generator stage in spark