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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
/*
* Copyright (c) 2014, 2016 Hewlett-Packard Development Company, L.P. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.aaa.h2.persistence;
import com.google.common.base.Preconditions;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import org.apache.commons.lang3.StringEscapeUtils;
import org.opendaylight.aaa.api.IDMStoreUtil;
import org.opendaylight.aaa.api.model.Role;
import org.opendaylight.aaa.api.model.Roles;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
*
* @author peter.mellquist@hp.com
*
*/
public class RoleStore extends AbstractStore<Role> {
private static final Logger LOG = LoggerFactory.getLogger(RoleStore.class);
protected final static String SQL_ID = "roleid";
protected final static String SQL_DOMAIN_ID = "domainid";
protected final static String SQL_NAME = "name";
protected final static String SQL_DESCR = "description";
private static final String TABLE_NAME = "ROLES";
protected RoleStore() {
super(TABLE_NAME);
}
@Override
protected String getTableCreationStatement() {
return "CREATE TABLE ROLES "
+ "(roleid VARCHAR(128) PRIMARY KEY,"
+ "name VARCHAR(128) NOT NULL, "
+ "domainid VARCHAR(128) NOT NULL, "
+ "description VARCHAR(128) NOT NULL)";
}
protected Role fromResultSet(ResultSet rs) throws SQLException {
Role role = new Role();
try {
role.setRoleid(rs.getString(SQL_ID));
role.setDomainid(rs.getString(SQL_DOMAIN_ID));
role.setName(rs.getString(SQL_NAME));
role.setDescription(rs.getString(SQL_DESCR));
} catch (SQLException sqle) {
LOG.error("SQL Exception: ", sqle);
throw sqle;
}
return role;
}
protected Roles getRoles() throws StoreException {
Roles roles = new Roles();
roles.setRoles(listAll());
return roles;
}
protected Role getRole(String id) throws StoreException {
try (Connection conn = dbConnect();
PreparedStatement pstmt = conn
.prepareStatement("SELECT * FROM ROLES WHERE roleid = ? ")) {
pstmt.setString(1, id);
LOG.debug("query string: {}", pstmt.toString());
return firstFromStatement(pstmt);
} catch (SQLException s) {
throw new StoreException("SQL Exception: " + s);
}
}
protected Role createRole(Role role) throws StoreException {
Preconditions.checkNotNull(role);
Preconditions.checkNotNull(role.getName());
Preconditions.checkNotNull(role.getDomainid());
String query = "insert into roles (roleid,domainid,name,description) values(?,?,?,?)";
try (Connection conn = dbConnect();
PreparedStatement statement = conn.prepareStatement(query)) {
role.setRoleid(IDMStoreUtil.createRoleid(role.getName(), role.getDomainid()));
statement.setString(1, role.getRoleid());
statement.setString(2, role.getDomainid());
statement.setString(3, role.getName());
statement.setString(4, role.getDescription());
int affectedRows = statement.executeUpdate();
if (affectedRows == 0) {
throw new StoreException("Creating role failed, no rows affected.");
}
return role;
} catch (SQLException s) {
throw new StoreException("SQL Exception : " + s);
}
}
protected Role putRole(Role role) throws StoreException {
Role savedRole = this.getRole(role.getRoleid());
if (savedRole == null) {
return null;
}
if (role.getDescription() != null) {
savedRole.setDescription(role.getDescription());
}
if (role.getName() != null) {
savedRole.setName(role.getName());
}
String query = "UPDATE roles SET description = ? WHERE roleid = ?";
try (Connection conn = dbConnect();
PreparedStatement statement = conn.prepareStatement(query)) {
statement.setString(1, savedRole.getDescription());
statement.setString(2, savedRole.getRoleid());
statement.executeUpdate();
} catch (SQLException s) {
throw new StoreException("SQL Exception : " + s);
}
return savedRole;
}
protected Role deleteRole(String roleid) throws StoreException {
roleid = StringEscapeUtils.escapeHtml4(roleid);
Role savedRole = this.getRole(roleid);
if (savedRole == null) {
return null;
}
String query = String.format("DELETE FROM ROLES WHERE roleid = '%s'", roleid);
try (Connection conn = dbConnect();
Statement statement = conn.createStatement()) {
int deleteCount = statement.executeUpdate(query);
LOG.debug("deleted {} records", deleteCount);
return savedRole;
} catch (SQLException s) {
throw new StoreException("SQL Exception : " + s);
}
}
}
|
