com.geotab.model.serialization.serdes.DutyStatusInfoDriverSerializer Maven / Gradle / Ivy

Go to download
package com.geotab.model.serialization.serdes;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import com.geotab.model.Id;
import com.geotab.model.entity.Entity;
import com.geotab.model.entity.group.Group;
import com.geotab.model.entity.user.Driver;
import com.geotab.util.Util;
import java.io.IOException;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class DutyStatusInfoDriverSerializer extends JsonSerializer {

  @Override
  public void serialize(Driver driver, JsonGenerator jsonGenerator,
      SerializerProvider serializerProvider) throws IOException {
    if (driver.isSystemEntity()) {
      jsonGenerator.writeString(driver.getId().getId());
    } else {
      Map driverFields = Util.mapBuilder()
          .put("id", driver.getId() != null ? driver.getId().getId() : null)
          .put("isDriver", driver.getIsDriver())
          .put("driverGroups", convertDriverGroups(driver.getDriverGroups()))
          .build();
      jsonGenerator.writeObject(driverFields);
    }
  }

  private Collection convertDriverGroups(List driverGroups) {
    if (Util.isEmpty(driverGroups)) {
      return Collections.emptyList();
    }

    Collection driverGroupsAsEntities = driverGroups.stream()
        .map(group -> (Entity) group)
        .collect(Collectors.toList());

    EntityCollectionAsIdCollectionSerializer entityCollectionAsIdCollectionSerializer =
        new EntityCollectionAsIdCollectionSerializer();

    return entityCollectionAsIdCollectionSerializer.convert(driverGroupsAsEntities);
  }
}