/*
Copyright 2014 Groupon, Inc.
Licensed 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 com.groupon.odo.proxylib;
import com.groupon.odo.proxylib.models.Client;
import com.groupon.odo.proxylib.models.Profile;
import com.groupon.odo.proxylib.models.ServerGroup;
import com.groupon.odo.proxylib.models.ServerRedirect;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class ServerRedirectService {
private static ServerRedirectService serviceInstance = null;
private static final Logger logger = LoggerFactory.getLogger(ServerRedirectService.class);
private static SQLService sqlService = null;
public ServerRedirectService() {
}
/**
* Obtain instance of ServerRedirectService
*
* @return instance of ServerRedirectService
*/
public static ServerRedirectService getInstance() {
if (serviceInstance == null) {
serviceInstance = new ServerRedirectService();
try {
sqlService = SQLService.getInstance();
} catch (Exception e) {
e.printStackTrace();
}
}
return serviceInstance;
}
/**
* Get the server redirects for a given clientId from the database
*
* @param clientId client ID
* @return collection of ServerRedirects
*/
public List<ServerRedirect> tableServers(int clientId) {
List<ServerRedirect> servers = new ArrayList<>();
try {
Client client = ClientService.getInstance().getClient(clientId);
servers = tableServers(client.getProfile().getId(), client.getActiveServerGroup());
} catch (SQLException e) {
e.printStackTrace();
} catch (Exception e) {
e.printStackTrace();
}
return servers;
}
/**
* Get the server redirects belonging to a server group
*
* @param profileId ID of profile
* @param serverGroupId ID of server group
* @return Collection of ServerRedirect for a server group
*/
public List<ServerRedirect> tableServers(int profileId, int serverGroupId) {
ArrayList<ServerRedirect> servers = new ArrayList<>();
PreparedStatement queryStatement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
queryStatement = sqlConnection.prepareStatement(
"SELECT * FROM " + Constants.DB_TABLE_SERVERS +
" WHERE " + Constants.GENERIC_PROFILE_ID + " = ?" +
" AND " + Constants.SERVER_REDIRECT_GROUP_ID + " = ?"
);
queryStatement.setInt(1, profileId);
queryStatement.setInt(2, serverGroupId);
results = queryStatement.executeQuery();
while (results.next()) {
ServerRedirect curServer = new ServerRedirect(results.getInt(Constants.GENERIC_ID),
results.getString(Constants.SERVER_REDIRECT_REGION),
results.getString(Constants.SERVER_REDIRECT_SRC_URL),
results.getString(Constants.SERVER_REDIRECT_DEST_URL),
results.getString(Constants.SERVER_REDIRECT_HOST_HEADER));
curServer.setProfileId(profileId);
servers.add(curServer);
}
} catch (SQLException e) {
e.printStackTrace();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (queryStatement != null) {
queryStatement.close();
}
} catch (Exception e) {
}
}
return servers;
}
/**
* Return all server groups for a profile
*
* @param profileId ID of profile
* @return collection of ServerGroups for a profile
*/
public List<ServerGroup> tableServerGroups(int profileId) {
ArrayList<ServerGroup> serverGroups = new ArrayList<>();
PreparedStatement queryStatement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
queryStatement = sqlConnection.prepareStatement(
"SELECT * FROM " + Constants.DB_TABLE_SERVER_GROUPS +
" WHERE " + Constants.GENERIC_PROFILE_ID + " = ? " +
"ORDER BY " + Constants.GENERIC_NAME
);
queryStatement.setInt(1, profileId);
results = queryStatement.executeQuery();
while (results.next()) {
ServerGroup curServerGroup = new ServerGroup(results.getInt(Constants.GENERIC_ID),
results.getString(Constants.GENERIC_NAME),
results.getInt(Constants.GENERIC_PROFILE_ID));
curServerGroup.setServers(tableServers(profileId, curServerGroup.getId()));
serverGroups.add(curServerGroup);
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (queryStatement != null) {
queryStatement.close();
}
} catch (Exception e) {
}
}
return serverGroups;
}
/**
* Returns redirect information for the given ID
*
* @param id ID of redirect
* @return ServerRedirect
* @throws Exception exception
*/
public ServerRedirect getRedirect(int id) throws Exception {
PreparedStatement queryStatement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
queryStatement = sqlConnection.prepareStatement(
"SELECT * FROM " + Constants.DB_TABLE_SERVERS +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
queryStatement.setInt(1, id);
results = queryStatement.executeQuery();
if (results.next()) {
ServerRedirect curServer = new ServerRedirect(results.getInt(Constants.GENERIC_ID),
results.getString(Constants.SERVER_REDIRECT_REGION),
results.getString(Constants.SERVER_REDIRECT_SRC_URL),
results.getString(Constants.SERVER_REDIRECT_DEST_URL),
results.getString(Constants.SERVER_REDIRECT_HOST_HEADER));
curServer.setProfileId(results.getInt(Constants.GENERIC_PROFILE_ID));
return curServer;
}
logger.info("Did not find the ID: {}", id);
} catch (SQLException e) {
throw e;
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (queryStatement != null) {
queryStatement.close();
}
} catch (Exception e) {
}
}
return null;
}
/**
* Returns server group by ID
*
* @param id ID of server group
* @return ServerGroup
* @throws Exception exception
*/
public ServerGroup getServerGroup(int id, int profileId) throws Exception {
PreparedStatement queryStatement = null;
ResultSet results = null;
if (id == 0) {
return new ServerGroup(0, "Default", profileId);
}
try (Connection sqlConnection = sqlService.getConnection()) {
queryStatement = sqlConnection.prepareStatement(
"SELECT * FROM " + Constants.DB_TABLE_SERVER_GROUPS +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
queryStatement.setInt(1, id);
results = queryStatement.executeQuery();
if (results.next()) {
ServerGroup curGroup = new ServerGroup(results.getInt(Constants.GENERIC_ID),
results.getString(Constants.GENERIC_NAME),
results.getInt(Constants.GENERIC_PROFILE_ID));
return curGroup;
}
logger.info("Did not find the ID: {}", id);
} catch (SQLException e) {
throw e;
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (queryStatement != null) {
queryStatement.close();
}
} catch (Exception e) {
}
}
return null;
}
/**
* Add server redirect to a profile, using current active ServerGroup
*
* @param region region
* @param srcUrl source URL
* @param destUrl destination URL
* @param hostHeader host header
* @param profileId profile ID
* @return ID of added ServerRedirect
* @throws Exception exception
*/
public int addServerRedirectToProfile(String region, String srcUrl, String destUrl, String hostHeader,
int profileId, int clientId) throws Exception {
int serverId = -1;
try {
Client client = ClientService.getInstance().getClient(clientId);
serverId = addServerRedirect(region, srcUrl, destUrl, hostHeader, profileId, client.getActiveServerGroup());
} catch (Exception e) {
e.printStackTrace();
}
return serverId;
}
/**
* Add server redirect to a profile
*
* @param region region
* @param srcUrl source URL
* @param destUrl destination URL
* @param hostHeader host header
* @param profileId profile ID
* @param groupId group ID
* @return ID of added ServerRedirect
* @throws Exception exception
*/
public int addServerRedirect(String region, String srcUrl, String destUrl, String hostHeader, int profileId, int groupId) throws Exception {
int serverId = -1;
PreparedStatement statement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement("INSERT INTO " + Constants.DB_TABLE_SERVERS
+ "(" + Constants.SERVER_REDIRECT_REGION + "," +
Constants.SERVER_REDIRECT_SRC_URL + "," +
Constants.SERVER_REDIRECT_DEST_URL + "," +
Constants.SERVER_REDIRECT_HOST_HEADER + "," +
Constants.SERVER_REDIRECT_PROFILE_ID + "," +
Constants.SERVER_REDIRECT_GROUP_ID + ")"
+ " VALUES (?, ?, ?, ?, ?, ?);", PreparedStatement.RETURN_GENERATED_KEYS);
statement.setString(1, region);
statement.setString(2, srcUrl);
statement.setString(3, destUrl);
statement.setString(4, hostHeader);
statement.setInt(5, profileId);
statement.setInt(6, groupId);
statement.executeUpdate();
results = statement.getGeneratedKeys();
if (results.next()) {
serverId = results.getInt(1);
} else {
// something went wrong
throw new Exception("Could not add path");
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
return serverId;
}
/**
* Add a new server group
*
* @param groupName name of the group
* @param profileId ID of associated profile
* @return id of server group
* @throws Exception
*/
public int addServerGroup(String groupName, int profileId) throws Exception {
int groupId = -1;
PreparedStatement statement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement("INSERT INTO " + Constants.DB_TABLE_SERVER_GROUPS
+ "(" + Constants.GENERIC_NAME + "," +
Constants.GENERIC_PROFILE_ID + ")"
+ " VALUES (?, ?);", PreparedStatement.RETURN_GENERATED_KEYS);
statement.setString(1, groupName);
statement.setInt(2, profileId);
statement.executeUpdate();
results = statement.getGeneratedKeys();
if (results.next()) {
groupId = results.getInt(1);
} else {
// something went wrong
throw new Exception("Could not add group");
}
} catch (SQLException e) {
e.printStackTrace();
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
return groupId;
}
/**
* Set the group name
*
* @param name new name of server group
* @param id ID of group
*/
public void setGroupName(String name, int id) {
PreparedStatement statement = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement(
"UPDATE " + Constants.DB_TABLE_SERVER_GROUPS +
" SET " + Constants.GENERIC_NAME + " = ?" +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
statement.setString(1, name);
statement.setInt(2, id);
statement.executeUpdate();
statement.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
}
/**
* Activate a server group
*
* @param groupId ID of group
* @param clientId client ID
*/
public void activateServerGroup(int groupId, int clientId) {
PreparedStatement statement = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement(
"UPDATE " + Constants.DB_TABLE_CLIENT +
" SET " + Constants.CLIENT_ACTIVESERVERGROUP + " = ? " +
" WHERE " + Constants.GENERIC_ID + " = ? "
);
statement.setInt(1, groupId);
statement.setInt(2, clientId);
statement.executeUpdate();
statement.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
}
/**
* Set source url for a server
*
* @param newUrl new URL
* @param id Server ID
*/
public void setSourceUrl(String newUrl, int id) {
PreparedStatement statement = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement(
"UPDATE " + Constants.DB_TABLE_SERVERS +
" SET " + Constants.SERVER_REDIRECT_SRC_URL + " = ?" +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
statement.setString(1, newUrl);
statement.setInt(2, id);
statement.executeUpdate();
statement.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
}
/**
* Set destination url for a server
*
* @param newUrl new URL
* @param id server ID
*/
public void setDestinationUrl(String newUrl, int id) {
PreparedStatement statement = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement(
"UPDATE " + Constants.DB_TABLE_SERVERS +
" SET " + Constants.SERVER_REDIRECT_DEST_URL + " = ?" +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
statement.setString(1, newUrl);
statement.setInt(2, id);
statement.executeUpdate();
statement.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
}
/**
* Set (optional) host header for a server
*
* @param newHost host header
* @param id server ID
*/
public void setHostHeader(String newHost, int id) {
PreparedStatement statement = null;
try (Connection sqlConnection = sqlService.getConnection()) {
statement = sqlConnection.prepareStatement(
"UPDATE " + Constants.DB_TABLE_SERVERS +
" SET " + Constants.SERVER_REDIRECT_HOST_HEADER + " = ?" +
" WHERE " + Constants.GENERIC_ID + " = ?"
);
statement.setString(1, newHost);
statement.setInt(2, id);
statement.executeUpdate();
statement.close();
} catch (Exception e) {
e.printStackTrace();
} finally {
try {
if (statement != null) {
statement.close();
}
} catch (Exception e) {
}
}
}
/**
* Deletes a redirect by id
*
* @param id redirect ID
*/
public void deleteRedirect(int id) {
try {
sqlService.executeUpdate("DELETE FROM " + Constants.DB_TABLE_SERVERS +
" WHERE " + Constants.GENERIC_ID + " = " + id + ";");
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* Delete a server group by id
*
* @param id server group ID
*/
public void deleteServerGroup(int id) {
try {
sqlService.executeUpdate("DELETE FROM " + Constants.DB_TABLE_SERVER_GROUPS +
" WHERE " + Constants.GENERIC_ID + " = " + id + ";");
sqlService.executeUpdate("DELETE FROM " + Constants.DB_TABLE_SERVERS +
" WHERE " + Constants.SERVER_REDIRECT_GROUP_ID + " = " + id);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* This returns all profiles associated with a server name
*
* @param serverName server Name
* @return profile UUID
* @throws Exception exception
*/
public Profile[] getProfilesForServerName(String serverName) throws Exception {
int profileId = -1;
ArrayList<Profile> returnProfiles = new ArrayList<Profile>();
PreparedStatement queryStatement = null;
ResultSet results = null;
try (Connection sqlConnection = sqlService.getConnection()) {
queryStatement = sqlConnection.prepareStatement(
"SELECT " + Constants.GENERIC_PROFILE_ID + " FROM " + Constants.DB_TABLE_SERVERS +
" WHERE " + Constants.SERVER_REDIRECT_SRC_URL + " = ? GROUP BY " +
Constants.GENERIC_PROFILE_ID
);
queryStatement.setString(1, serverName);
results = queryStatement.executeQuery();
while (results.next()) {
profileId = results.getInt(Constants.GENERIC_PROFILE_ID);
Profile profile = ProfileService.getInstance().findProfile(profileId);
returnProfiles.add(profile);
}
} catch (SQLException e) {
throw e;
} finally {
try {
if (results != null) {
results.close();
}
} catch (Exception e) {
}
try {
if (queryStatement != null) {
queryStatement.close();
}
} catch (Exception e) {
}
}
if (returnProfiles.size() == 0) {
return null;
}
return returnProfiles.toArray(new Profile[0]);
}
/**
* Returns true or false depending on whether or not Odo can handle the request for this server/clientUUID pair
*
* @param serverName server Name
* @return true if odo can handle, false otherwise
* @throws Exception exception
*/
public Boolean canHandleRequest(String serverName) throws Exception {
// TODO: Future optimizations
try {
Profile[] profiles = this.getProfilesForServerName(serverName);
if (profiles == null) {
logger.info("No matching profiles found for path");
return false;
}
for (Profile profile : profiles) {
List<Client> clients = ClientService.getInstance().findAllClients(profile.getId());
for (Client client : clients) {
if (client.getIsActive()) {
return true;
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
return false;
}
}