Newer
Older
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
package it.inaf.ia2.gms.persistence;
import it.inaf.ia2.gms.persistence.model.MembershipEntity;
import it.inaf.ia2.gms.persistence.model.PermissionEntity;
import java.sql.PreparedStatement;
import java.sql.Types;
import java.util.Collections;
import java.util.Set;
import javax.sql.DataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
@Component
public class JoinDAO {
private static final Logger LOG = LoggerFactory.getLogger(JoinDAO.class);
private final JdbcTemplate jdbcTemplate;
@Autowired
public JoinDAO(DataSource dataSource) {
jdbcTemplate = new JdbcTemplate(dataSource);
}
@Transactional
public void join(Set<MembershipEntity> membershipsToAdd, Set<PermissionEntity> permissionsToAdd, String userToDelete) {
if (!membershipsToAdd.isEmpty()) {
addMemberships(membershipsToAdd);
}
if (!permissionsToAdd.isEmpty()) {
addPermissions(permissionsToAdd);
}
deleteUserMemberships(userToDelete);
deleteUserPermissions(userToDelete);
}
private void addMemberships(Set<MembershipEntity> membershipsToAdd) {
String sql = "INSERT INTO gms_membership (group_id, user_id) VALUES "
+ String.join(", ", Collections.nCopies(membershipsToAdd.size(), "(?, ?)"));
LOG.trace("Executing {}", sql);
jdbcTemplate.update(conn -> {
PreparedStatement ps = conn.prepareStatement(sql);
int i = 0;
for (MembershipEntity membership : membershipsToAdd) {
ps.setString(++i, membership.getGroupId());
ps.setString(++i, membership.getUserId());
}
return ps;
});
}
private void addPermissions(Set<PermissionEntity> permissionsToAdd) {
Sonia Zorba
committed
String sql = "INSERT INTO gms_permission (group_id, user_id, permission, group_path, set_by) VALUES "
+ String.join(", ", Collections.nCopies(permissionsToAdd.size(), "(?, ?, ?, ?, ?)")) + "\n"
+ "ON CONFLICT (group_id, user_id) DO UPDATE\n"
Sonia Zorba
committed
+ "SET permission = EXCLUDED.permission, set_by = EXCLUDED.set_by, update_time = NOW()";
LOG.trace("Executing {}", sql);
jdbcTemplate.update(conn -> {
PreparedStatement ps = conn.prepareStatement(sql);
int i = 0;
for (PermissionEntity permission : permissionsToAdd) {
ps.setString(++i, permission.getGroupId());
ps.setString(++i, permission.getUserId());
ps.setObject(++i, permission.getPermission().toString(), Types.OTHER);
ps.setObject(++i, permission.getGroupPath(), Types.OTHER);
Sonia Zorba
committed
ps.setString(++i, permission.getSetBy());
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
108
}
return ps;
});
}
private void deleteUserMemberships(String userId) {
String sql = "DELETE FROM gms_membership WHERE user_id = ?";
LOG.trace("Executing {}", sql);
jdbcTemplate.update(conn -> {
PreparedStatement ps = conn.prepareStatement(sql);
ps.setString(1, userId);
return ps;
});
}
private void deleteUserPermissions(String userId) {
String sql = "DELETE FROM gms_permission WHERE user_id = ?";
LOG.trace("Executing {}", sql);
jdbcTemplate.update(conn -> {
PreparedStatement ps = conn.prepareStatement(sql);
ps.setString(1, userId);
return ps;
});
}
}