package com.linkedin.restli.tools.snapshot.check;
import com.linkedin.data.DataList;
import com.linkedin.data.DataMap;
import com.linkedin.data.codec.JacksonDataCodec;
import com.linkedin.data.schema.DataSchemaResolver;
import com.linkedin.data.schema.NamedDataSchema;
import com.linkedin.data.schema.resolver.DefaultDataSchemaResolver;
import com.linkedin.restli.restspec.ResourceSchema;
import com.linkedin.restli.restspec.RestSpecCodec;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
/**
* @author Keren Jin
*/
public abstract class AbstractSnapshot
{
public static final String MODELS_KEY = "models";
public static final String SCHEMA_KEY = "schema";
protected static final JacksonDataCodec _dataCodec = new JacksonDataCodec();
protected Map<String, NamedDataSchema> _models;
protected ResourceSchema _resourceSchema;
private final DataSchemaResolver _dataSchemaResolver = new DefaultDataSchemaResolver(); // each Snapshot should have its own DataSchemaResolver.
protected Map<String, NamedDataSchema> parseModels(DataList models) throws IOException
{
final Map<String, NamedDataSchema> parsedModels = new HashMap<String, NamedDataSchema>();
for (Object modelObj : models)
{
NamedDataSchema dataSchema;
if (modelObj instanceof DataMap)
{
DataMap model = (DataMap)modelObj;
dataSchema = (NamedDataSchema) RestSpecCodec.textToSchema(_dataCodec.mapToString(model), _dataSchemaResolver);
}
else if (modelObj instanceof String)
{
String str = (String)modelObj;
dataSchema = (NamedDataSchema) RestSpecCodec.textToSchema(str, _dataSchemaResolver);
}
else
{
throw new IOException("Found " + modelObj.getClass() + " in models list; Models must be strings or DataMaps.");
}
parsedModels.put(dataSchema.getFullName(), dataSchema);
}
return parsedModels;
}
protected ResourceSchema parseSchema(DataMap data)
{
return new ResourceSchema(data);
}
/**
* @return a map containing all of the NamedDataSchemas in this Snapshot, keyed by fully qualified schema name.
*/
public Map<String, NamedDataSchema> getModels()
{
return _models;
}
/**
* @return the {@link ResourceSchema} of this snapshot
*/
public ResourceSchema getResourceSchema()
{
return _resourceSchema;
}
}