[jiankang] refactor make the unit a single class