summaryrefslogtreecommitdiff
path: root/src/test/java/com/amazon/carbonado/sample/FileInfoTest.java
blob: 0f38de72dfad82d1cf029308f04f00a66daaa529 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/*
 * Copyright 2006 Amazon Technologies, Inc. or its affiliates.
 * Amazon, Amazon.com and Carbonado are trademarks or registered trademarks
 * of Amazon Technologies, Inc. or its affiliates.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.amazon.carbonado.sample;

import java.io.File;
import java.io.InputStream;

import org.joda.time.DateTime;

import com.amazon.carbonado.Cursor;
import com.amazon.carbonado.Query;
import com.amazon.carbonado.Repository;
import com.amazon.carbonado.Storage;

import com.amazon.carbonado.lob.FileBlob;

import com.amazon.carbonado.repo.sleepycat.BDBRepositoryBuilder;

import com.amazon.carbonado.stored.FileInfo;

/**
 * Test program which implements storage for a simple indexed file system.
 *
 * @author Brian S O'Neill
 */
public class FileInfoTest {
    public static Storage<FileInfo> openStorage(String envHome) throws Exception {
        BDBRepositoryBuilder builder = new BDBRepositoryBuilder();
        builder.setName("Simple file system");
        builder.setEnvironmentHome(envHome);
        builder.setTransactionNoSync(true);
        //builder.setProduct("DB");

        Repository repository = builder.build();
        return repository.storageFor(FileInfo.class);
    }

    /**
     * @param args [0] - BDB environment home, [1] - optional directory to read
     * files from for populating the simple file system
     */
    public static void main(String[] args) throws Exception {
        Storage<FileInfo> storage = openStorage(args[0]);
        if (args.length > 1) {
            populate(storage, new File(args[1]), null);
        }
    }

    private static void populate(Storage<FileInfo> storage, File file, FileInfo parent)
        throws Exception
    {
        FileInfo info = storage.prepare();
        if (parent != null) {
            info.setParentID(parent.getID());
        }
        info.setName(file.getName());
        info.setLastModified(new DateTime(file.lastModified()));

        if (file.isFile()) {
            info.setDirectory(false);
            info.setLength(file.length());
            info.setFileData(new FileBlob(file));
            info.insert();
        } else if (file.isDirectory()) {
            info.setDirectory(true);
            info.setLength(-1);
            info.insert();
            File[] children = file.listFiles();
            if (children != null) {
                for (int i=0; i<children.length; i++) {
                    populate(storage, children[i], info);
                }
            }
        }
    }
}