7
|
1 package models;
|
|
2
|
20
|
3 import java.util.ArrayList;
|
22
|
4 import java.util.HashMap;
|
17
|
5
|
7
|
6 import com.tinkerpop.blueprints.Vertex;
|
18
|
7 import com.tinkerpop.gremlin.java.GremlinPipeline;
|
7
|
8
|
|
9 public class UserModel extends NodeModel {
|
8
|
10
|
7
|
11
|
|
12 public UserModel(Vertex vertex) {
|
|
13 super(vertex);
|
|
14 }
|
|
15
|
18
|
16 private Object[] getEdgeInUser(String labels) {
|
|
17 GremlinPipeline<Vertex,Vertex> pipe = new GremlinPipeline<Vertex,Vertex>();
|
|
18 pipe.start(this.vertex).in(labels);
|
20
|
19 ArrayList<Object> array = new ArrayList<Object>();
|
|
20 for (Vertex v : pipe) {
|
|
21 array.add(v.getId());
|
18
|
22 }
|
20
|
23 if (array.size() == 0) return null;
|
|
24 return array.toArray();
|
9
|
25 }
|
|
26
|
18
|
27 public Object[] getUserRequests() {
|
|
28 return getEdgeInUser(L_REQUEST);
|
|
29 }
|
|
30
|
|
31 public Object[] getUserClaims() {
|
21
|
32 return getEdgeInUser(L_AUTHOR);
|
10
|
33 }
|
17
|
34
|
18
|
35 public Object[] getUserConsensus() {
|
|
36 return null;
|
10
|
37 }
|
22
|
38
|
|
39 public HashMap<Object,Object[]> getUserInfo() {
|
|
40 HashMap<Object,Object[]> hash = new HashMap<Object, Object[]>();
|
|
41 hash.put(NodeModel.REQUESTS, getUserRequests());
|
|
42 hash.put(NodeModel.CLAIMS, getUserClaims());
|
|
43 hash.put(NodeModel.CONSENSUS, getUserConsensus());
|
|
44 return hash;
|
|
45 }
|
10
|
46
|
20
|
47
|
7
|
48
|
|
49 }
|