summaryrefslogtreecommitdiff
path: root/source/com/c2kernel/entity/transfer/TransferSet.java
blob: ee98f8dea676b773c9c56ac5e94a98352ed40714 (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
103
104
105
106
107
package com.c2kernel.entity.transfer;

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

import com.c2kernel.lookup.EntityPath;
import com.c2kernel.lookup.NextKeyManager;
import com.c2kernel.process.Gateway;
import com.c2kernel.utils.CastorXMLUtility;
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 items;
    
    public TransferSet() {
    }
    
    public TransferSet(int[] sysKeys) {
        items = new ArrayList();
        for (int i = 0; i < sysKeys.length; i++) {
            try {
                items.add(new TransferItem(sysKeys[i]));
            } catch (Exception ex) {
                Logger.error("Could not add item "+sysKeys[i]);
                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 (Iterator iter = items.iterator(); iter.hasNext();) {
            TransferItem element = (TransferItem)iter.next();
            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 = CastorXMLUtility.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 (Iterator iter = items.iterator(); iter.hasNext();) {
            TransferItem element = (TransferItem)iter.next();
            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 (Iterator iter = items.iterator(); iter.hasNext();) {
            TransferItem element = (TransferItem)iter.next();
            if (element.sysKey > packageLastKey)
                packageLastKey = element.sysKey;
        }
        
        try
        {   // find the current last key
            NextKeyManager nextKeyMan = Gateway.getLDAPLookup().getNextKeyManager();
            EntityPath 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);
        }           
    }
}