|
|
@@ -17,8 +17,6 @@ func main() {
|
|
|
reader := csv.NewReader(bufio.NewReader(csvFile))
|
|
|
reader.Comma = '\t'
|
|
|
reader.LazyQuotes = true
|
|
|
- conn, _ := redis.Dial("tcp", "localhost:6379")
|
|
|
- graph := rg.Graph{}.New("Reddit", conn)
|
|
|
for {
|
|
|
line, error := reader.Read()
|
|
|
if error == io.EOF {
|
|
|
@@ -26,10 +24,18 @@ func main() {
|
|
|
} else if error != nil {
|
|
|
log.Print(error)
|
|
|
} else {
|
|
|
+ conn, _ := redis.Dial("tcp", "localhost:6379")
|
|
|
+ graph := rg.Graph{}.New("Reddit", conn)
|
|
|
query := fmt.Sprintf("MATCH (u:User {RedditId:'u%s'}), (v:User {RedditId:'u%s'}) CREATE (u)-[d:CONTACTED]->(v)", line[0], line[1])
|
|
|
- graph.Query(query)
|
|
|
+ _, jerr := graph.Query(query)
|
|
|
+ if jerr != nil {
|
|
|
+ fmt.Println(jerr)
|
|
|
+ fmt.Println(line)
|
|
|
+ fmt.Println(query)
|
|
|
+ os.Exit(1)
|
|
|
+ }
|
|
|
graph.Flush()
|
|
|
+ conn.Close()
|
|
|
}
|
|
|
}
|
|
|
- conn.Close()
|
|
|
}
|
Chris Mague