Netflix Graph Basic - Java Big Data

Java examples for Big Data:Netflix Graph

  1. HOME
  2. Java
  3. Big Data
  4. Netflix Graph

Description

Netflix Graph Basic

Demo Code

package org.blackbananacoin.premature;

import static com.netflix.nfgraph.spec.NFPropertySpec.COMPACT;
import static com.netflix.nfgraph.spec.NFPropertySpec.HASH;
import static com.netflix.nfgraph.spec.NFPropertySpec.MULTIPLE;
import static com.netflix.nfgraph.spec.NFPropertySpec.SINGLE;

import com.netflix.nfgraph.OrdinalIterator;
import com.netflix.nfgraph.build.NFBuildGraph;
import com.netflix.nfgraph.compressed.NFCompressedGraph;
import com.netflix.nfgraph.spec.NFGraphSpec;
import com.netflix.nfgraph.spec.NFNodeSpec;
import com.netflix.nfgraph.spec.NFPropertySpec;
import com.netflix.nfgraph.util.OrdinalMap;

public class NetflixGraphHello {

    NFGraphSpec movieSchema = new NFGraphSpec(new NFNodeSpec("Actor",
            new NFPropertySpec("starredIn", "Movie", MULTIPLE | COMPACT),
            new NFPropertySpec("costarredWith", "Actor", MULTIPLE | HASH),
            new NFPropertySpec("premieredIn", "Movie", SINGLE)),
            new NFNodeSpec("Movie", new NFPropertySpec("rated", "Rating",
                    SINGLE), new NFPropertySpec("starring", "Actor",
                    MULTIPLE | HASH)), new NFNodeSpec("Rating"));
    private int matrixOrdinal;
    private int keanuOrdinal;
    private int laurenceOrdinal;
    OrdinalMap<String> movieOrdinals = new OrdinalMap<String>();
    OrdinalMap<String> actorOrdinals = new OrdinalMap<String>();
    OrdinalMap<String> ratingOrdinals = new OrdinalMap<String>();

    public NFCompressedGraph initGraph() {
        NFBuildGraph buildGraph = new NFBuildGraph(movieSchema);

        matrixOrdinal = movieOrdinals.add("The Matrix");
        keanuOrdinal = actorOrdinals.add("Keanu Reeves");
        laurenceOrdinal = actorOrdinals.add("Laurence Fishburn");

        int ratedROrdinal = ratingOrdinals.add("R");

        buildGraph.addConnection("Actor", keanuOrdinal, "starredIn",
                matrixOrdinal);/*from  w w w  . ja  va 2  s .  c  o m*/
        buildGraph.addConnection("Actor", laurenceOrdinal, "starredIn",
                matrixOrdinal);
        buildGraph.addConnection("Movie", matrixOrdinal, "starring",
                keanuOrdinal);
        buildGraph.addConnection("Movie", matrixOrdinal, "starring",
                laurenceOrdinal);
        buildGraph.addConnection("Movie", matrixOrdinal, "rated",
                ratedROrdinal);

        NFCompressedGraph compressedGraph = buildGraph.compress();
        return compressedGraph;
    }

    public void testGraph(NFCompressedGraph compressedGraph) {
        int matrixOrdinal = movieOrdinals.get("The Matrix");

        OrdinalIterator iter = compressedGraph.getConnectionIterator(
                "Movie", matrixOrdinal, "starring");

        int currentOrdinal = iter.nextOrdinal();

        while (currentOrdinal != OrdinalIterator.NO_MORE_ORDINALS) {
            System.out.println(actorOrdinals.get(currentOrdinal)
                    + " starred in The Matrix!");
            currentOrdinal = iter.nextOrdinal();
        }
    }

    public static void main(String[] args) {

        NetflixGraphHello app = new NetflixGraphHello();
        NFCompressedGraph g = app.initGraph();
        app.testGraph(g);

    }

}

Related Tutorials