Home

Contribute

Source code


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

/*
 * Copyright (c) 2004 - 2012 Eike Stepper (Berlin, Germany) 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
 * 
 * Contributors:
 *    Eike Stepper - initial API and implementation
 */
package org.eclipse.net4j.db.db2;

import org.eclipse.net4j.db.DBUtil;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;

/**
 * @author Eike Stepper
 */
public final class DB2SchemaDataSource extends com.ibm.db2.jcc.DB2SimpleDataSource
{
  private static final long serialVersionUID = 1L;

  private String schemaName;

  public DB2SchemaDataSource(String schemaName)
  {
    this.schemaName = schemaName;
  }

  public String getSchemaName()
  {
    return schemaName;
  }

  @Override
  public Connection getConnection() throws SQLException
  {
    Connection connection = super.getConnection();
    setSchema(connection);
    return connection;
  }

  @Override
  public Connection getConnection(Object arg0) throws SQLException
  {
    Connection connection = super.getConnection(arg0);
    setSchema(connection);
    return connection;
  }

  @Override
  public Connection getConnection(String arg0, String arg1) throws SQLException
  {
    Connection connection = super.getConnection(arg0, arg1);
    setSchema(connection);
    return connection;
  }

  private void setSchema(Connection connection) throws SQLException
  {
    Statement stmt = null;

    try
    {
      stmt = connection.createStatement();
      stmt.execute("SET CURRENT SCHEMA " + schemaName); //$NON-NLS-1$
    }
    finally
    {
      DBUtil.close(stmt);
    }
  }
}