/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.shiro.samples.spring;
import javax.sql.DataSource;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.jdbc.core.JdbcTemplate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.shiro.crypto.hash.Sha256Hash;
/**
* A data populator that creates a set of security tables and test data that can be used by the
* Shiro Spring sample application to demonstrate the use of the {@link org.apache.shiro.realm.jdbc.JdbcRealm}
* The tables created by this class follow the default table and column names that {@link org.apache.shiro.realm.jdbc.JdbcRealm} uses.
*
* @author Les Hazlewood
*/
public class BootstrapDataPopulator implements InitializingBean {
private static final String CREATE_TABLES = "create table users (\n" +
" username varchar(255) primary key,\n" +
" password varchar(255) not null\n" +
");\n" +
"\n" +
"create table roles (\n" +
" role_name varchar(255) primary key\n" +
");\n" +
"\n" +
"create table user_roles (\n" +
" username varchar(255) not null,\n" +
" role_name varchar(255) not null,\n" +
" constraint user_roles_uq unique ( username, role_name )\n" +
");\n" +
"\n" +
"create table roles_permissions (\n" +
" role_name varchar(255) not null,\n" +
" permission varchar(255) not null,\n" +
" primary key (role_name, permission)\n" +
");";
private static final Logger log = LoggerFactory.getLogger(BootstrapDataPopulator.class);
protected DataSource dataSource = null;
public void setDataSource(DataSource dataSource) {
this.dataSource = dataSource;
}
public void afterPropertiesSet() throws Exception {
//because we're using an in-memory hsqldb for the sample app, a new one will be created each time the
//app starts, so create the tables and insert the 2 sample users on bootstrap:
JdbcTemplate jdbcTemplate = new JdbcTemplate(this.dataSource);
jdbcTemplate.execute(CREATE_TABLES);
//password is 'user1' SHA hashed and base64 encoded:
//The first argument to the hash constructor is the actual value to be hased. The 2nd is the
//salt. In this simple demo scenario, the username and the password are the same, but to clarify the
//distinction, you would see this in practice:
//new Sha256Hash( <password>, <username> )
String query = "insert into users values ('user1', '" + new Sha256Hash("user1", "user1").toBase64() + "' )";
jdbcTemplate.execute(query);
log.debug("Created user1.");
//password is 'user2' SHA hashed and base64 encoded:
query = "insert into users values ( 'user2', '" + new Sha256Hash("user2", "user2").toBase64() + "' )";
jdbcTemplate.execute(query);
log.debug("Created user2.");
query = "insert into roles values ( 'role1' )";
jdbcTemplate.execute(query);
log.debug("Created role1");
query = "insert into roles values ( 'role2' )";
jdbcTemplate.execute(query);
log.debug("Created role2");
query = "insert into roles_permissions values ( 'role1', 'permission1')";
jdbcTemplate.execute(query);
log.debug("Created permission 1 for role 1");
query = "insert into roles_permissions values ( 'role1', 'permission2')";
jdbcTemplate.execute(query);
log.debug("Created permission 2 for role 1");
query = "insert into roles_permissions values ( 'role2', 'permission1')";
jdbcTemplate.execute(query);
log.debug("Created permission 1 for role 2");
query = "insert into user_roles values ( 'user1', 'role1' )";
jdbcTemplate.execute(query);
query = "insert into user_roles values ( 'user1', 'role2' )";
jdbcTemplate.execute(query);
log.debug("Assigned user1 roles role1 and role2");
query = "insert into user_roles values ( 'user2', 'role2' )";
jdbcTemplate.execute(query);
log.debug("Assigned user2 role role2");
}
}