com.graphaware.example.module.FriendshipStrengthApi Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of friendship-strength-counter-module Show documentation

Example of a GraphAware Transaction-Driven Runtime Module

There is a newer version: 4.2.0.58

Show newest version

/*
 * Copyright (c) 2013-2016 GraphAware
 *
 * This file is part of the GraphAware Framework.
 *
 * GraphAware Framework is free software: you can redistribute it and/or modify it under the terms of
 * the GNU General Public License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of
 * the GNU General Public License along with this program.  If not, see
 * .
 */

package com.graphaware.example.module;

import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.graphdb.Transaction;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 *  Simple REST API for finding the total friendship strength.
 */
@Controller
@RequestMapping("/friendship/strength")
public class FriendshipStrengthApi {

    private final GraphDatabaseService database;
    private final FriendshipStrengthCounter counter;

    @Autowired
    public FriendshipStrengthApi(GraphDatabaseService database) {
        this.database = database;
        counter = new FriendshipStrengthCounter(database);
    }

    @RequestMapping(method = RequestMethod.GET)
    @ResponseBody
    public long getTotalFriendshipStrength() {
        long totalFriendshipStrength;

        try (Transaction tx = database.beginTx()) {
            totalFriendshipStrength = counter.getTotalFriendshipStrength();
            tx.success();
        }

        return totalFriendshipStrength;
    }
}