summaryrefslogtreecommitdiff
path: root/src/main/java/com/c2kernel/entity/transfer/TransferSet.java
blob: a7d81b64c0ade14b97434ecd5f78504a6855faa9 (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
93
94
95
96
97
98
99
100
101
102
package com.c2kernel.entity.transfer;

import java.io.File;
import java.util.ArrayList;

import com.c2kernel.lookup.ItemPath;
import com.c2kernel.lookup.NextKeyManager;
import com.c2kernel.process.Gateway;
import com.c2kernel.utils.FileStringUtility;
import com.c2kernel.utils.Logger;

/**************************************************************************
 *
 * $Revision: 1.5 $
 * $Date: 2005/04/26 06:48:13 $
 *
 * Copyright (C) 2003 CERN - European Organization for Nuclear Research
 * All rights reserved.
 **************************************************************************/

public class TransferSet {

    public ArrayList<TransferItem> items;

    public TransferSet() {
    }

    public TransferSet(int[] sysKeys) {
        items = new ArrayList<TransferItem>();
        for (int sysKey : sysKeys) {
            try {
                items.add(new TransferItem(sysKey));
            } catch (Exception ex) {
                Logger.error("Could not add item "+sysKey);
                Logger.error(ex);
            }
        }
    }

    public void exportPackage(File dir) throws Exception {
        if (items==null || items.size() == 0)
            throw new Exception("Nothing to dump");
        FileStringUtility.createNewDir(dir.getAbsolutePath());
        for (TransferItem element : items) {
            try {
                element.exportItem(new File(dir, String.valueOf(element.sysKey)), "/");
            } catch (Exception ex) {
                Logger.error("Error dumping item "+element.sysKey);
                Logger.error(ex);
            }
        }

        try {
            String self = Gateway.getMarshaller().marshall(this);
            FileStringUtility.string2File(new File(dir, "transferSet.xml"), self);
        } catch (Exception ex) {
            Logger.error("Error writing header file");
            Logger.error(ex);
        }
    }

    public void importPackage(File rootDir) {
        for (TransferItem element : items) {
            Logger.msg(5, "Importing "+element.sysKey);
            try {
                element.importItem(new File(rootDir, String.valueOf(element.sysKey)));
            } catch (Exception ex) {
                Logger.error("Import of item "+element.sysKey+" failed. Rolling back");
                Logger.error(ex);
                Gateway.getStorage().abort(element);
            }
        }
        checkLastKey();
    }

    private void checkLastKey()
    {
        // find highest key in out import set
        int packageLastKey = 0;
        for (TransferItem element : items) {
            if (element.sysKey > packageLastKey)
                packageLastKey = element.sysKey;
        }

        try
        {   // find the current last key
            NextKeyManager nextKeyMan = Gateway.getLDAPLookup().getNextKeyManager();
            ItemPath lastKey = nextKeyMan.getLastEntityPath();
            Logger.msg(1, "Last key imported was "+packageLastKey+". LDAP lastkey was "+lastKey.getSysKey());


            if (packageLastKey > lastKey.getSysKey()) { // set new last
                Logger.msg(1, "Updating lastKey to "+packageLastKey);
                nextKeyMan.writeLastEntityKey(packageLastKey);
            }
        }
        catch (Exception ex)
        {
            Logger.error("Exception::LoadKeys::processFile() " + ex);
        }
    }
}