|
122
|
1 package jp.ac.u_ryukyu.ie.cr.tatsuki.query;
|
|
|
2
|
|
|
3 import java.nio.ByteBuffer;
|
|
|
4 import java.security.KeyStore.Entry.Attribute;
|
|
|
5 import java.util.Iterator;
|
|
|
6
|
|
|
7 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungle;
|
|
|
8 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungleTree;
|
|
|
9 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.Jungle;
|
|
|
10 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
|
|
|
11 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
|
|
|
12 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
|
|
|
13 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
|
|
|
14 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultTreeEditor;
|
|
|
15 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.TreeNode;
|
|
|
16 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.transaction.DefaultTreeNode;
|
|
|
17 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.DefaultTraverser;
|
|
|
18 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.InterfaceTraverser;
|
|
|
19 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Either;
|
|
|
20 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Error;
|
|
|
21 import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.util.Pair;
|
|
124
|
22 import jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.query.PathNodeIndexIterator;
|
|
122
|
23
|
|
|
24 import org.junit.Assert;
|
|
|
25 import org.junit.Test;
|
|
|
26
|
|
125
|
27 import fj.data.List;
|
|
|
28 import fj.data.Option;
|
|
|
29 import fj.data.TreeMap;
|
|
|
30
|
|
122
|
31 public class SearchQueryTest {
|
|
|
32
|
|
|
33 @Test
|
|
|
34 public void SearchQueryTest() {
|
|
|
35 Jungle jungle = new DefaultJungle(null, "hogehoge",new DefaultTreeEditor(new DefaultTraverser()));
|
|
|
36 jungle.createNewTree("tree");
|
|
|
37 JungleTree tree = jungle.getTreeByName("tree");
|
|
|
38 createTree(tree);
|
|
127
|
39 tree.getRootNode();
|
|
122
|
40 InterfaceTraverser ifTraverser = tree.getTraverser();
|
|
126
|
41 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> cheackIndex = ifTraverser.getIndex();
|
|
|
42 Assert.assertTrue(cheackIndex.isEmpty());
|
|
|
43
|
|
124
|
44 Iterator<Pair<TreeNode, NodePath>> searchNode = ifTraverser.find(
|
|
122
|
45 (TreeNode node) -> {
|
|
|
46 ByteBuffer attribute = node.getAttributes().get(key);
|
|
|
47 if(attribute != null){
|
|
|
48 byte[] byteAttribute = attribute.array();
|
|
|
49 String str = new String(byteAttribute);
|
|
|
50 System.out.println("attribute = " + str);
|
|
124
|
51 return str.equals("<-1,0,1>");
|
|
122
|
52 }
|
|
|
53 return false;
|
|
|
54 }
|
|
124
|
55 ,key,"<-1,0,1>");
|
|
|
56
|
|
126
|
57
|
|
124
|
58 while (searchNode.hasNext()){
|
|
|
59 Assert.assertTrue(compare(searchNode.next().left(),"<-1,0,1>"));
|
|
|
60 }
|
|
126
|
61
|
|
125
|
62 TreeMap<String, TreeMap<String, List<Pair<TreeNode, NodePath>>>> index = ifTraverser.getIndex();
|
|
|
63 Option<TreeMap<String, List<Pair<TreeNode, NodePath>>>> opIndex = index.get(key);
|
|
|
64 Assert.assertTrue(!opIndex.isNone());
|
|
126
|
65 TreeMap<String, List<Pair<TreeNode, NodePath>>> innerIndex = opIndex.some();
|
|
|
66 Assert.assertTrue(!innerIndex.get("<-1,0,0>").isNone());
|
|
122
|
67 }
|
|
|
68
|
|
124
|
69
|
|
122
|
70 public boolean compare(TreeNode compareNode, String compareAttribute) {
|
|
|
71 String labName = compareNode.getAttributes().getString(key);
|
|
|
72 if (labName.equals(compareAttribute))
|
|
|
73 return true;
|
|
|
74 return false;
|
|
|
75 }
|
|
|
76
|
|
|
77 public static String key = "KEY";
|
|
|
78 public static DefaultTreeNode factory = new DefaultTreeNode();
|
|
|
79
|
|
|
80 public void createTree(JungleTree tree) {
|
|
|
81 NodePath root = new DefaultNodePath();
|
|
|
82 createChildren(tree, root, 0);
|
|
|
83
|
|
|
84 for (int x = 0; x < 2; x++) {
|
|
|
85 createChildren(tree, root.add(0), x);
|
|
|
86 for (int y = 0; y < 2; y++) {
|
|
|
87 createChildren(tree, root.add(0).add(x), y);
|
|
|
88 }
|
|
|
89 }
|
|
|
90
|
|
|
91 }
|
|
|
92
|
|
|
93 public void createChildren(JungleTree tree, NodePath root, int num) {
|
|
|
94 JungleTreeEditor editor = tree.getTreeEditor();// Treeのeditorを作成
|
|
|
95 Either<Error, JungleTreeEditor> either = editor.addNewChildAt(root,num); // 新しく入れるところへのパス
|
|
|
96 if (either.isA()) {
|
|
|
97 Assert.fail();
|
|
|
98 }
|
|
|
99 editor = either.b();
|
|
|
100 either = editor.success();
|
|
|
101 if (either.isA()) {
|
|
|
102 Assert.fail();
|
|
|
103 }
|
|
|
104 NodePath childPath = root.add(num);
|
|
|
105 editor = tree.getTreeEditor();
|
|
|
106 NodePath attribute = root.add(num);
|
|
|
107 System.out.println(attribute.toString());
|
|
|
108 either = editor.putAttribute(childPath, key,ByteBuffer.wrap(attribute.toString().getBytes()));
|
|
|
109 if (either.isA()) {
|
|
|
110 Assert.fail();
|
|
|
111 }
|
|
|
112 editor = either.b();
|
|
|
113 either = editor.success();
|
|
|
114 }
|
|
|
115 }
|
