Java tutorial
/******************************************************************************* * Cloud Foundry * Copyright (c) [2009-2016] Pivotal Software, Inc. All Rights Reserved. * * This product is licensed to you under the Apache License, Version 2.0 (the "License"). * You may not use this product except in compliance with the License. * * This product includes a number of subcomponents with * separate copyright notices and license terms. Your use of these * subcomponents is subject to the terms and conditions of the * subcomponent's license, as noted in the LICENSE file. *******************************************************************************/ package org.cloudfoundry.identity.uaa.zone; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.cloudfoundry.identity.uaa.audit.event.SystemDeletable; import org.cloudfoundry.identity.uaa.util.JsonUtils; import org.springframework.dao.DuplicateKeyException; import org.springframework.dao.EmptyResultDataAccessException; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.core.PreparedStatementSetter; import org.springframework.jdbc.core.RowMapper; import org.springframework.util.Assert; import org.springframework.util.StringUtils; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Timestamp; import java.util.Date; import java.util.List; public class JdbcIdentityZoneProvisioning implements IdentityZoneProvisioning, SystemDeletable { public static final String ID_ZONE_FIELDS = "id,version,created,lastmodified,name,subdomain,description,config"; public static final String ID_ZONE_UPDATE_FIELDS = "version,lastmodified,name,subdomain,description,config" .replace(",", "=?,") + "=?"; public static final String CREATE_IDENTITY_ZONE_SQL = "insert into identity_zone(" + ID_ZONE_FIELDS + ") values (?,?,?,?,?,?,?,?)"; public static final String UPDATE_IDENTITY_ZONE_SQL = "update identity_zone set " + ID_ZONE_UPDATE_FIELDS + " where id=?"; public static final String DELETE_IDENTITY_ZONE_SQL = "delete from identity_zone where id=?"; public static final String IDENTITY_ZONES_QUERY = "select " + ID_ZONE_FIELDS + " from identity_zone "; public static final String IDENTITY_ZONE_BY_ID_QUERY = IDENTITY_ZONES_QUERY + "where id=?"; public static final String IDENTITY_ZONE_BY_SUBDOMAIN_QUERY = "select " + ID_ZONE_FIELDS + " from identity_zone " + "where subdomain=?"; public static final Log logger = LogFactory.getLog(JdbcIdentityZoneProvisioning.class); protected final JdbcTemplate jdbcTemplate; private final RowMapper<IdentityZone> mapper = new IdentityZoneRowMapper(); public JdbcIdentityZoneProvisioning(JdbcTemplate jdbcTemplate) { Assert.notNull(jdbcTemplate); this.jdbcTemplate = jdbcTemplate; } @Override public IdentityZone retrieve(String id) { try { IdentityZone identityZone = jdbcTemplate.queryForObject(IDENTITY_ZONE_BY_ID_QUERY, mapper, id); return identityZone; } catch (EmptyResultDataAccessException x) { throw new ZoneDoesNotExistsException("Zone[" + id + "] not found.", x); } } @Override public List<IdentityZone> retrieveAll() { return jdbcTemplate.query(IDENTITY_ZONES_QUERY, mapper); } @Override public IdentityZone retrieveBySubdomain(String subdomain) { if (subdomain == null) { throw new EmptyResultDataAccessException("Subdomain cannot be null", 1); } IdentityZone identityZone = jdbcTemplate.queryForObject(IDENTITY_ZONE_BY_SUBDOMAIN_QUERY, mapper, subdomain.toLowerCase()); return identityZone; } @Override public IdentityZone create(final IdentityZone identityZone) { try { jdbcTemplate.update(CREATE_IDENTITY_ZONE_SQL, new PreparedStatementSetter() { @Override public void setValues(PreparedStatement ps) throws SQLException { ps.setString(1, identityZone.getId().trim()); ps.setInt(2, identityZone.getVersion()); ps.setTimestamp(3, new Timestamp(new Date().getTime())); ps.setTimestamp(4, new Timestamp(new Date().getTime())); ps.setString(5, identityZone.getName()); ps.setString(6, identityZone.getSubdomain().toLowerCase()); ps.setString(7, identityZone.getDescription()); ps.setString(8, identityZone.getConfig() != null ? JsonUtils.writeValueAsString(identityZone.getConfig()) : null); } }); } catch (DuplicateKeyException e) { throw new ZoneAlreadyExistsException(e.getMostSpecificCause().getMessage(), e); } return retrieve(identityZone.getId()); } @Override public IdentityZone update(final IdentityZone identityZone) { try { jdbcTemplate.update(UPDATE_IDENTITY_ZONE_SQL, new PreparedStatementSetter() { @Override public void setValues(PreparedStatement ps) throws SQLException { ps.setInt(1, identityZone.getVersion() + 1); ps.setTimestamp(2, new Timestamp(new Date().getTime())); ps.setString(3, identityZone.getName()); ps.setString(4, identityZone.getSubdomain().toLowerCase()); ps.setString(5, identityZone.getDescription()); ps.setString(6, identityZone.getConfig() != null ? JsonUtils.writeValueAsString(identityZone.getConfig()) : null); ps.setString(7, identityZone.getId().trim()); } }); } catch (DuplicateKeyException e) { //duplicate subdomain throw new ZoneAlreadyExistsException(e.getMostSpecificCause().getMessage(), e); } return retrieve(identityZone.getId()); } @Override public int deleteByIdentityZone(String zoneId) { return jdbcTemplate.update(DELETE_IDENTITY_ZONE_SQL, zoneId); } @Override public int deleteByOrigin(String origin, String zoneId) { return 0; } @Override public Log getLogger() { return logger; } public static final class IdentityZoneRowMapper implements RowMapper<IdentityZone> { @Override public IdentityZone mapRow(ResultSet rs, int rowNum) throws SQLException { IdentityZone identityZone = new IdentityZone(); identityZone.setId(rs.getString(1).trim()); identityZone.setVersion(rs.getInt(2)); identityZone.setCreated(rs.getTimestamp(3)); identityZone.setLastModified(rs.getTimestamp(4)); identityZone.setName(rs.getString(5)); identityZone.setSubdomain(rs.getString(6)); identityZone.setDescription(rs.getString(7)); String config = rs.getString(8); if (StringUtils.hasText(config)) { try { identityZone.setConfig(JsonUtils.readValue(config, IdentityZoneConfiguration.class)); } catch (JsonUtils.JsonUtilException e) { logger.error("Invalid zone configuration found for zone id:" + identityZone.getId(), e); identityZone.setConfig(new IdentityZoneConfiguration()); } } return identityZone; } } }