package com.xd.pre.modules.security.social; import lombok.extern.slf4j.Slf4j; import org.springframework.dao.DataAccessException; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.core.ResultSetExtractor; import org.springframework.jdbc.core.namedparam.MapSqlParameterSource; import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate; import org.springframework.security.crypto.encrypt.TextEncryptor; import org.springframework.social.connect.*; import javax.sql.DataSource; import java.sql.ResultSet; import java.sql.SQLException; import java.util.Arrays; import java.util.HashSet; import java.util.List; import java.util.Set; /** * @Classname PreJdbcUsersConnectionRepository * @Description jdbc连接 * @Author Created by Lihaodong (alias:小东啊) [email protected] * @Date 2019-07-19 13:09 * @Version 1.0 */ public class PreJdbcUsersConnectionRepository implements UsersConnectionRepository { private final JdbcTemplate jdbcTemplate; private final ConnectionFactoryLocator connectionFactoryLocator; private final TextEncryptor textEncryptor; private ConnectionSignUp connectionSignUp; private String tablePrefix = ""; public PreJdbcUsersConnectionRepository(DataSource dataSource, ConnectionFactoryLocator connectionFactoryLocator, TextEncryptor textEncryptor) { this.jdbcTemplate = new JdbcTemplate(dataSource); this.connectionFactoryLocator = connectionFactoryLocator; this.textEncryptor = textEncryptor; } public void setConnectionSignUp(ConnectionSignUp connectionSignUp) { this.connectionSignUp = connectionSignUp; } public void setTablePrefix(String tablePrefix) { this.tablePrefix = tablePrefix; } /** * 1.获取到第三用户信息后调用这个api向数据库查询 看有没有绑定 需要再此加上租户id * * @param connection * @return */ @Override public List<String> findUserIdsWithConnection(Connection<?> connection) { ConnectionKey key = connection.getKey(); List<String> localUserIds = jdbcTemplate.queryForList("select userId from " + tablePrefix + "UserConnection where providerId = ? and providerUserId = ?", String.class, key.getProviderId(), key.getProviderUserId()); if (localUserIds.size() == 0 && connectionSignUp != null) { String newUserId = connectionSignUp.execute(connection); if (newUserId != null) { createConnectionRepository(newUserId).addConnection(connection); return Arrays.asList(newUserId); } } return localUserIds; } @Override public Set<String> findUserIdsConnectedTo(String providerId, Set<String> providerUserIds) { MapSqlParameterSource parameters = new MapSqlParameterSource(); parameters.addValue("providerId", providerId); parameters.addValue("providerUserIds", providerUserIds); final Set<String> localUserIds = new HashSet<String>(); return new NamedParameterJdbcTemplate(jdbcTemplate).query("select userId from " + tablePrefix + "UserConnection where providerId = :providerId and providerUserId in (:providerUserIds)", parameters, new ResultSetExtractor<Set<String>>() { @Override public Set<String> extractData(ResultSet rs) throws SQLException, DataAccessException { while (rs.next()) { localUserIds.add(rs.getString("userId")); } return localUserIds; } }); } @Override public ConnectionRepository createConnectionRepository(String userId) { if (userId == null) { throw new IllegalArgumentException("userId cannot be null"); } return new PreJdbcConnectionRepository(userId, jdbcTemplate, connectionFactoryLocator, textEncryptor, tablePrefix); } }