如果在neo4j db中不存在,则插入对象

时间:2017-11-28 12:20:26

标签: neo4j spring-data spring-data-neo4j

我有一个看起来像这样的对象

import lombok.Data;
import org.neo4j.ogm.annotation.GraphId;
import org.neo4j.ogm.annotation.NodeEntity;
import org.neo4j.ogm.annotation.Relationship;

import java.util.HashSet;
import java.util.Set;

@Data
@NodeEntity
public class GPlayer {

    @GraphId
    private Long id;
    @Relationship(type = "comrade", direction = Relationship.UNDIRECTED)
    private Set<GPlayer> comrades;
    // @Indexed(unique = true) doesn't work in v4
    private String name;

    /**
     * Adds new comrade.
     *
     * @param comrade comrade
     */
    public void acquainted(GPlayer comrade) {
        if (null == comrades) {
            comrades = new HashSet<>();
        } else {
            if (comrades.contains(comrade)) {
                return;
            }
        }
        comrades.add(comrade);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        if (!super.equals(o)) return false;

        GPlayer gPlayer = (GPlayer) o;

        if (!id.equals(gPlayer.id)) return false;
        return name.equals(gPlayer.name);
    }

    @Override
    public int hashCode() {
        int result = super.hashCode();
        result = 31 * result + id.hashCode();
        result = 31 * result + name.hashCode();
        return result;
    }

    @Override
    public String toString() {
        return "GPlayer{" +
                "id=" + id +
                ", comrades=" + comrades +
                ", name='" + name + '\'' +
                '}';
    }
}

我想将其存储在Neo4j中,并带有唯一的name。因此,当我插入GPlayer时 - 如果具有此类名称的对象不存在,我不想插入它。

我的存储库看起来像这样

public interface GraphPlayerRepository extends GraphRepository<GPlayer> {
    List<GPlayer> findAll();
    GPlayer findByName(String name);
}

要插入我喜欢这个

private Long createPlayer(String playerName, String comrade, GraphPlayerRepository gRepo) {
    GPlayer gPlayer = gRepo.findByName(playerName);
    if (null == gPlayer) {
        gPlayer = new GPlayer();
        gPlayer.setName(playerName);
        if (null != comrade) {
            gPlayer.acquainted(gRepo.findByName(comrade));
        }
        gRepo.save(gPlayer);
        LOGGER.info("Created new GRAPH player: {}", gPlayer);
    } else {
        gPlayer.acquainted(gRepo.findByName(comrade));
        gRepo.save(gPlayer);
        LOGGER.info("Updated player: {}", gPlayer);
    }
    return gPlayer.getId();
}

但它看起来相当冗长。有没有办法让它变得更简单?

1 个答案:

答案 0 :(得分:2)

您可以使用该名称作为实体的ID

@Index(unique=true, primary=true)
private String name;

然后无需声明findByName方法,只需使用Neo4jRepository

public interface GraphPlayerRepository extends Neo4jRepository<GPlayer, String> {
...
}

并使用repository.findOne(name)从名称中获取玩家。

那就是说, hashCode和Equals的实现不正确:强烈建议不要在这些中使用Long id。请参阅here

否则,与SDN无关,但改进可能包括:

  • 使用lombok生成hashcode并等于
  • 在字段级别初始化hashSet

您也可以迁移到SDN 5并使用Optional返回类型来避免if / else nullity检查块。