blob: 76ee14ea920a67d631a008850b74cdf8c66f4a91 (
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
|
/**
* Copyright (c) 2009 Red Hat, Inc.
*
* This software is licensed to you under the GNU General Public License,
* version 2 (GPLv2). There is NO WARRANTY for this software, express or
* implied, including the implied warranties of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2
* along with this software; if not, see
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
*
* Red Hat trademarks are not licensed under GPLv2. No permission is
* granted to use or replicate Red Hat trademarks that are incorporated
* in this software or its documentation.
*/
package org.fedoraproject.candlepin.model;
import java.util.HashSet;
import java.util.Set;
import org.hibernate.criterion.Restrictions;
import com.google.inject.Inject;
import com.wideplay.warp.persist.Transactional;
public class ConsumerCurator extends AbstractHibernateCurator<Consumer> {
@Inject private ConsumerFactCurator consumerInfoCurator;
@Inject private ProductCurator productCurator;
@Inject private EntitlementCurator entitlementCurator;
protected ConsumerCurator() {
super(Consumer.class);
}
public Consumer lookupByName(String name) {
return (Consumer) currentSession().createCriteria(Consumer.class)
.add(Restrictions.like("name", name))
.uniqueResult();
}
public Consumer lookupByUuid(String uuid) {
return (Consumer) currentSession().createCriteria(Consumer.class)
.add(Restrictions.like("uuid", uuid))
.uniqueResult();
}
@Transactional
public Consumer update(Consumer updatedConsumer) {
Consumer existingConsumer = find(updatedConsumer.getId());
if (existingConsumer == null) {
return create(updatedConsumer);
}
// TODO: Are any of these read-only?
existingConsumer.setChildConsumers(bulkUpdate(updatedConsumer.getChildConsumers()));
existingConsumer.setConsumedProducts(productCurator.bulkUpdate(updatedConsumer.getConsumedProducts()));
existingConsumer.setEntitlements(entitlementCurator.bulkUpdate(updatedConsumer.getEntitlements()));
existingConsumer.setInfo(consumerInfoCurator.update(updatedConsumer.getInfo()));
existingConsumer.setName(updatedConsumer.getName());
existingConsumer.setOwner(updatedConsumer.getOwner());
existingConsumer.setParent(updatedConsumer.getParent());
existingConsumer.setType(updatedConsumer.getType());
existingConsumer.setUuid(updatedConsumer.getUuid());
save(existingConsumer);
return existingConsumer;
}
@Transactional
public Set<Consumer> bulkUpdate(Set<Consumer> consumers) {
Set<Consumer> toReturn = new HashSet<Consumer>();
for(Consumer toUpdate: consumers) {
toReturn.add(update(toUpdate));
}
return toReturn;
}
}
|