com.hubspot.rosetta.Rosetta Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of RosettaCore Show documentation
The newest version!
package com.hubspot.rosetta;

import com.fasterxml.jackson.databind.Module;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.hubspot.rosetta.databind.AutoDiscoveredModule;
import com.hubspot.rosetta.internal.RosettaModule;
import java.util.ArrayList;
import java.util.List;
import java.util.ServiceLoader;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicReference;

/**
 * Static public APIs to get/set some Rosetta globals.
 */
public enum Rosetta {
  INSTANCE;

  private static final List MODULES = new CopyOnWriteArrayList(
    defaultModules()
  );
  private static final AtomicReference MAPPER =
    new AtomicReference(cloneAndCustomize(new ObjectMapper()));

  public static ObjectMapper getMapper() {
    return INSTANCE.get();
  }

  /**
   * @deprecated To customize the ObjectMapper, use {@link com.hubspot.rosetta.jdbi.RosettaObjectMapperOverride}.
   */
  @Deprecated
  public static void addModule(Module module) {
    INSTANCE.add(module);
  }

  /**
   * @deprecated To customize the ObjectMapper, use {@link com.hubspot.rosetta.jdbi.RosettaObjectMapperOverride}.
   */
  @Deprecated
  public static void setMapper(ObjectMapper mapper) {
    INSTANCE.set(mapper);
  }

  public static ObjectMapper cloneAndCustomize(ObjectMapper mapper) {
    mapper = mapper.copy();

    // ObjectMapper#registerModules doesn't exist in 2.1.x
    for (Module module : MODULES) {
      mapper.registerModule(module);
    }

    return mapper;
  }

  private ObjectMapper get() {
    return MAPPER.get();
  }

  private void set(ObjectMapper mapper) {
    MAPPER.set(cloneAndCustomize(mapper));
  }

  private void add(Module module) {
    MODULES.add(module);
    MAPPER.get().registerModule(module);
  }

  private static List defaultModules() {
    List defaultModules = new ArrayList();
    defaultModules.add(new RosettaModule());

    for (Module module : ServiceLoader.load(AutoDiscoveredModule.class)) {
      defaultModules.add(module);
    }

    for (Module module : ServiceLoader.load(Module.class)) {
      defaultModules.add(module);
    }

    return defaultModules;
  }
}