code: private void addWorkUnit(WorkUnit workUnit, ParallelRunner stateSerDeRunner,
        Map<String, TaskConfig> taskConfigMap) throws IOException {
      String workUnitFilePath = persistWorkUnit(
          new Path(this.inputWorkUnitDir, this.jobContext.getJobId()), workUnit, stateSerDeRunner);
  
      Map<String, String> rawConfigMap = Maps.newHashMap();
      rawConfigMap.put(GobblinClusterConfigurationKeys.WORK_UNIT_FILE_PATH, workUnitFilePath);
      rawConfigMap.put(ConfigurationKeys.JOB_NAME_KEY, this.jobContext.getJobName());
      rawConfigMap.put(ConfigurationKeys.JOB_ID_KEY, this.jobContext.getJobId());
      rawConfigMap.put(ConfigurationKeys.TASK_ID_KEY, workUnit.getId());
      rawConfigMap.put(GobblinClusterConfigurationKeys.TASK_SUCCESS_OPTIONAL_KEY, "true");
  
      taskConfigMap.put(workUnit.getId(), TaskConfig.Builder.from(rawConfigMap));
    }
docs: Add a single {@link WorkUnit} (flattened).