00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025 package org.objectweb.cjdbc.controller.backend;
00026
00027 import java.io.IOException;
00028 import java.io.Serializable;
00029 import java.io.StringReader;
00030 import java.net.ConnectException;
00031 import java.sql.Connection;
00032 import java.sql.SQLException;
00033 import java.sql.Statement;
00034 import java.util.ArrayList;
00035 import java.util.HashMap;
00036 import java.util.Hashtable;
00037 import java.util.Iterator;
00038 import java.util.Map;
00039 import java.util.Vector;
00040
00041 import javax.management.NotCompliantMBeanException;
00042
00043 import org.dom4j.Document;
00044 import org.dom4j.Element;
00045 import org.dom4j.io.SAXReader;
00046 import org.objectweb.cjdbc.common.exceptions.NoTransactionStartWhenDisablingException;
00047 import org.objectweb.cjdbc.common.exceptions.UnreachableBackendException;
00048 import org.objectweb.cjdbc.common.i18n.Translate;
00049 import org.objectweb.cjdbc.common.jmx.mbeans.DatabaseBackendMBean;
00050 import org.objectweb.cjdbc.common.jmx.notifications.CjdbcNotificationList;
00051 import org.objectweb.cjdbc.common.log.Trace;
00052 import org.objectweb.cjdbc.common.shared.BackendInfo;
00053 import org.objectweb.cjdbc.common.shared.BackendState;
00054 import org.objectweb.cjdbc.common.sql.metadata.MetadataContainer;
00055 import org.objectweb.cjdbc.common.sql.schema.DatabaseSchema;
00056 import org.objectweb.cjdbc.common.xml.DatabasesXmlTags;
00057 import org.objectweb.cjdbc.common.xml.XmlComponent;
00058 import org.objectweb.cjdbc.controller.backend.rewriting.AbstractRewritingRule;
00059 import org.objectweb.cjdbc.controller.connection.AbstractConnectionManager;
00060 import org.objectweb.cjdbc.controller.connection.FailFastPoolConnectionManager;
00061 import org.objectweb.cjdbc.controller.connection.RandomWaitPoolConnectionManager;
00062 import org.objectweb.cjdbc.controller.connection.SimpleConnectionManager;
00063 import org.objectweb.cjdbc.controller.connection.VariablePoolConnectionManager;
00064 import org.objectweb.cjdbc.controller.jmx.AbstractStandardMBean;
00065 import org.objectweb.cjdbc.controller.jmx.MBeanServerManager;
00066 import org.objectweb.cjdbc.controller.jmx.RmiConnector;
00067 import org.objectweb.cjdbc.controller.loadbalancer.AbstractLoadBalancer;
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080 public final class DatabaseBackend extends AbstractStandardMBean
00081 implements
00082 Serializable,
00083 XmlComponent,
00084 DatabaseBackendMBean
00085 {
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098 private String name;
00099
00100
00101 private String driverPath;
00102
00103
00104 private String driverClassName;
00105
00106
00107 private transient DriverCompliance driverCompliance;
00108
00109
00110 private String url;
00111
00112
00113 private String virtualDatabaseName;
00114
00115
00116 private boolean writeCanBeEnabled;
00117
00118
00119 private String connectionTestStatement;
00120
00121
00122 private transient DatabaseSchema schema;
00123
00124
00125 private boolean schemaIsStatic = false;
00126
00127
00128 private transient HashMap connectionManagers;
00129
00130
00131 protected transient Trace logger;
00132
00133
00134 private transient ArrayList activeTransactions = new ArrayList();
00135
00136
00137 private transient Vector pendingRequests = new Vector();
00138
00139
00140 private int totalRequest;
00141 private int totalWriteRequest;
00142 private int totalReadRequest;
00143 private int totalTransactions;
00144
00145
00146 private ArrayList rewritingRules;
00147
00148
00149 private int dynamicPrecision;
00150 private boolean gatherSystemTables = false;
00151 private String schemaName = null;
00152
00153
00154 private int sqlShortFormLength = 40;
00155
00156 private String lastKnownCheckpoint;
00157
00158
00159
00160
00161
00162
00163 private int state = BackendState.DISABLED;
00164
00165 private transient BackendStateListener stateListener;
00166
00167
00168
00169
00170
00171
00172
00173
00174
00175
00176
00177
00178
00179
00180
00181
00182 public DatabaseBackend(String name, String driverPath,
00183 String driverClassName, String url, String vdbName,
00184 boolean writeCanBeEnabled, String connectionTestStatement)
00185 throws NotCompliantMBeanException
00186 {
00187 super(DatabaseBackendMBean.class);
00188 if (name == null)
00189 throw new IllegalArgumentException(Translate
00190 .get("backend.null.backend.name"));
00191
00192 if (driverClassName == null)
00193 throw new IllegalArgumentException(Translate.get("backend.null.driver"));
00194
00195 if (url == null)
00196 throw new IllegalArgumentException(Translate.get("backend.null.url"));
00197
00198 if (vdbName == null)
00199 throw new IllegalArgumentException(Translate
00200 .get("backend.null.virtualdatabase.name"));
00201
00202 if (connectionTestStatement == null)
00203 throw new IllegalArgumentException(Translate
00204 .get("backend.null.connection.test"));
00205
00206 this.name = name;
00207 this.writeCanBeEnabled = writeCanBeEnabled;
00208 this.driverPath = driverPath;
00209 this.driverClassName = driverClassName;
00210 this.url = url;
00211 this.virtualDatabaseName = vdbName;
00212 this.connectionTestStatement = connectionTestStatement;
00213 this.connectionManagers = new HashMap();
00214 logger = Trace
00215 .getLogger("org.objectweb.cjdbc.controller.backend.DatabaseBackend."
00216 + name);
00217 this.driverCompliance = new DriverCompliance(logger);
00218 totalRequest = 0;
00219 dynamicPrecision = DatabaseBackendSchemaConstants.DynamicPrecisionAll;
00220 }
00221
00222
00223
00224
00225
00226
00227
00228
00229 public DatabaseBackend(BackendInfo info) throws NotCompliantMBeanException
00230 {
00231 this(info.getName(), info.getDriverPath(), info.getDriverClassName(), info
00232 .getUrl(), info.getVirtualDatabaseName(), true, info
00233 .getConnectionTestStatement());
00234 try
00235 {
00236 String xml = info.getXml();
00237 StringReader sreader = new StringReader(xml);
00238 SAXReader reader = new SAXReader();
00239 Document document = reader.read(sreader);
00240 Element root = document.getRootElement();
00241 Iterator iter1 = root.elementIterator();
00242 while (iter1.hasNext())
00243 {
00244 Element elem = (Element) iter1.next();
00245 if (elem.getName().equals(DatabasesXmlTags.ELT_ConnectionManager))
00246 {
00247 String vuser = elem.valueOf("@" + DatabasesXmlTags.ATT_vLogin);
00248 String rlogin = elem.valueOf("@" + DatabasesXmlTags.ATT_rLogin);
00249 String rpassword = elem.valueOf("@" + DatabasesXmlTags.ATT_rPassword);
00250 Iterator iter2 = elem.elementIterator();
00251 while (iter2.hasNext())
00252 {
00253 Element connectionManager = (Element) iter2.next();
00254 String cname = connectionManager.getName();
00255 if (cname
00256 .equals(DatabasesXmlTags.ELT_VariablePoolConnectionManager))
00257 {
00258 int minPoolSize = Integer.parseInt(connectionManager.valueOf("@"
00259 + DatabasesXmlTags.ATT_minPoolSize));
00260 int maxPoolSize = Integer.parseInt(connectionManager.valueOf("@"
00261 + DatabasesXmlTags.ATT_maxPoolSize));
00262 int idleTimeout = Integer.parseInt(connectionManager.valueOf("@"
00263 + DatabasesXmlTags.ATT_idleTimeout));
00264 int waitTimeout = Integer.parseInt(connectionManager.valueOf("@"
00265 + DatabasesXmlTags.ATT_waitTimeout));
00266 this.addConnectionManager(vuser,
00267 new VariablePoolConnectionManager(url, name, rlogin,
00268 rpassword, driverPath, driverClassName, minPoolSize,
00269 maxPoolSize, idleTimeout, waitTimeout));
00270 }
00271 else if (cname.equals(DatabasesXmlTags.ELT_SimpleConnectionManager))
00272 {
00273 this.addConnectionManager(vuser, new SimpleConnectionManager(url,
00274 name, rlogin, rpassword, driverPath, driverClassName));
00275 }
00276 else if (cname
00277 .equals(DatabasesXmlTags.ELT_RandomWaitPoolConnectionManager))
00278 {
00279 int poolSize = Integer.parseInt(connectionManager.valueOf("@"
00280 + DatabasesXmlTags.ATT_poolSize));
00281 int timeout = Integer.parseInt(connectionManager.valueOf("@"
00282 + DatabasesXmlTags.ATT_timeout));
00283 this
00284 .addConnectionManager(vuser,
00285 new RandomWaitPoolConnectionManager(url, name, rlogin,
00286 rpassword, driverPath, driverClassName, poolSize,
00287 timeout));
00288 }
00289 else if (cname
00290 .equals(DatabasesXmlTags.ELT_FailFastPoolConnectionManager))
00291 {
00292 int poolSize = Integer.parseInt(connectionManager.valueOf("@"
00293 + DatabasesXmlTags.ATT_poolSize));
00294 this.addConnectionManager(vuser,
00295 new FailFastPoolConnectionManager(url, name, rlogin,
00296 rpassword, driverPath, driverClassName, poolSize));
00297 }
00298 }
00299 }
00300 }
00301
00302 }
00303 catch (Exception e)
00304 {
00305 logger
00306 .error(Translate.get("backend.add.connection.manager.failed", e), e);
00307 }
00308 }
00309
00310
00311
00312
00313
00314
00315
00316
00317
00318
00319
00320
00321
00322
00323
00324
00325
00326
00327 public DatabaseBackend(String name, String driverPath,
00328 String driverClassName, String url, String vdbName,
00329 String connectionTestStatement, String dynamicSchemaLevel)
00330 throws NotCompliantMBeanException
00331 {
00332 this(name, driverPath, driverClassName, url, vdbName, true,
00333 connectionTestStatement);
00334 this.dynamicPrecision = DatabaseBackendSchemaConstants
00335 .getDynamicSchemaLevel(dynamicSchemaLevel);
00336 }
00337
00338
00339
00340
00341
00342
00343 public void setSqlShortFormLength(int sqlShortFormLength)
00344 {
00345 this.sqlShortFormLength = sqlShortFormLength;
00346 }
00347
00348
00349
00350
00351
00352
00353
00354 public int getSQLShortFormLength()
00355 {
00356 return sqlShortFormLength;
00357 }
00358
00359
00360
00361
00362
00363
00364
00365
00366
00367
00368 public boolean equals(Object other)
00369 {
00370 if ((other == null) || (!(other instanceof DatabaseBackend)))
00371 return false;
00372 else
00373 {
00374 DatabaseBackend b = (DatabaseBackend) other;
00375 return name.equals(b.getName())
00376 && driverClassName.equals(b.getDriverClassName())
00377 && url.equals(b.getURL());
00378 }
00379 }
00380
00381
00382
00383
00384
00385
00386
00387
00388
00389
00390
00391 public boolean hasTables(ArrayList tables)
00392 {
00393 if (schema == null)
00394 throw new NullPointerException(Translate.get("backend.schema.not.set"));
00395
00396 if (tables == null)
00397 throw new IllegalArgumentException(Translate.get("backend.null.tables"));
00398
00399 int size = tables.size();
00400 for (int i = 0; i < size; i++)
00401 {
00402 if (!schema.hasTable((String) tables.get(i)))
00403 return false;
00404 }
00405 return true;
00406 }
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417 public boolean hasTable(String table)
00418 {
00419 if (schema == null)
00420 throw new NullPointerException(Translate.get("backend.schema.not.set"));
00421
00422 return schema.hasTable(table);
00423 }
00424
00425
00426
00427
00428
00429
00430 public ArrayList getTables()
00431 {
00432 if (schema == null)
00433 throw new NullPointerException(Translate.get("backend.schema.not.set"));
00434 return schema.getTables();
00435 }
00436
00437
00438
00439
00440
00441
00442
00443
00444
00445
00446 public boolean hasStoredProcedure(String procedureName)
00447 {
00448 if (schema == null)
00449 throw new NullPointerException(Translate.get("backend.schema.not.set"));
00450
00451 return schema.hasProcedure(procedureName);
00452 }
00453
00454
00455
00456
00457
00458
00459
00460
00461
00462 public synchronized void initializeConnections() throws SQLException
00463 {
00464 if (connectionManagers.isEmpty())
00465 throw new SQLException(Translate.get("backend.not.defined", new String[]{
00466 name, url}));
00467
00468 AbstractConnectionManager connectionManager;
00469 Iterator iter = connectionManagers.values().iterator();
00470 while (iter.hasNext())
00471 {
00472 connectionManager = (AbstractConnectionManager) iter.next();
00473 if (!connectionManager.isInitialized())
00474 connectionManager.initializeConnections();
00475 }
00476 }
00477
00478
00479
00480
00481
00482
00483
00484 public synchronized void finalizeConnections() throws SQLException
00485 {
00486 if (connectionManagers.isEmpty())
00487 throw new SQLException(Translate.get("backend.not.defined", new String[]{
00488 name, url}));
00489
00490 AbstractConnectionManager connectionManager;
00491 Iterator iter = connectionManagers.values().iterator();
00492 while (iter.hasNext())
00493 {
00494 connectionManager = (AbstractConnectionManager) iter.next();
00495 if (connectionManager.isInitialized())
00496 connectionManager.finalizeConnections();
00497 }
00498 }
00499
00500
00501
00502
00503
00504
00505
00506
00507
00508
00509 public boolean isValidConnection(Connection connection)
00510 {
00511 try
00512 {
00513 Statement s = connection.createStatement();
00514 s.executeQuery(connectionTestStatement);
00515 }
00516 catch (SQLException e)
00517 {
00518 if ("25P02".equals(e.getSQLState())
00519 || (e.getMessage() != null && e
00520 .getMessage()
00521 .indexOf(
00522 "current transaction is aborted, queries ignored until end of transaction block") > 0))
00523 {
00524
00525
00526
00527
00528
00529
00530
00531
00532 return true;
00533 }
00534 return false;
00535 }
00536 return true;
00537 }
00538
00539
00540
00541
00542
00543
00544
00545
00546 public void addConnectionManager(String vLogin,
00547 AbstractConnectionManager connectionManager)
00548 {
00549 if (connectionManager == null)
00550 throw new IllegalArgumentException(Translate.get(
00551 "backend.null.connection.manager", new String[]{name, url}));
00552 if (logger.isInfoEnabled())
00553 logger.info(Translate.get("backend.add.connection.manager.for.user",
00554 vLogin));
00555 connectionManager.setVLogin(vLogin);
00556 connectionManagers.put(vLogin, connectionManager);
00557 }
00558
00559
00560
00561
00562
00563
00564
00565
00566
00567
00568
00569
00570
00571
00572
00573
00574 public synchronized Connection getConnectionForTransactionAndLazyBeginIfNeeded(
00575 Long tid, AbstractConnectionManager cm)
00576 throws UnreachableBackendException,
00577 NoTransactionStartWhenDisablingException, SQLException
00578 {
00579 if (isStartedTransaction(tid))
00580 {
00581 return cm.retrieveConnection(tid.longValue());
00582 }
00583 else
00584 {
00585 if (isDisabling())
00586 throw new NoTransactionStartWhenDisablingException();
00587
00588
00589 startTransaction(tid);
00590
00591
00592 return AbstractLoadBalancer.getConnectionAndBeginTransaction(this, cm,
00593 tid.longValue());
00594 }
00595 }
00596
00597
00598
00599
00600
00601
00602
00603
00604
00605 public void startTransaction(Long tid)
00606 {
00607 synchronized (activeTransactions)
00608 {
00609 totalTransactions++;
00610 activeTransactions.add(tid);
00611 }
00612 }
00613
00614
00615
00616
00617
00618
00619
00620 public void stopTransaction(Long tid)
00621 {
00622 synchronized (activeTransactions)
00623 {
00624 if (!activeTransactions.remove(tid))
00625 throw new IllegalArgumentException(Translate.get(
00626 "backend.transaction.not.started", new String[]{"" + tid, name}));
00627
00628
00629 if (activeTransactions.isEmpty())
00630 {
00631 activeTransactions.notifyAll();
00632 }
00633 }
00634 }
00635
00636
00637
00638
00639
00640
00641 public void waitForAllTransactionsToComplete()
00642 {
00643 synchronized (activeTransactions)
00644 {
00645 if (activeTransactions.isEmpty())
00646 return;
00647 else
00648 try
00649 {
00650 activeTransactions.wait();
00651 }
00652 catch (InterruptedException ignore)
00653 {
00654 }
00655 }
00656 }
00657
00658
00659
00660
00661
00662
00663
00664
00665 public boolean isStartedTransaction(Long tid)
00666 {
00667 synchronized (activeTransactions)
00668 {
00669 return activeTransactions.contains(tid);
00670 }
00671 }
00672
00673
00674
00675
00676
00677
00678
00679 public synchronized boolean isInitialized() throws SQLException
00680 {
00681 if (connectionManagers.isEmpty())
00682 throw new SQLException(Translate.get("backend.null.connection.manager",
00683 new String[]{name, url}));
00684 Iterator iter = connectionManagers.values().iterator();
00685 while (iter.hasNext())
00686 {
00687 if (!((AbstractConnectionManager) iter.next()).isInitialized())
00688 return false;
00689 }
00690 return true;
00691 }
00692
00693
00694
00695
00696
00697
00698
00699 public synchronized boolean isJDBCConnected()
00700 {
00701 try
00702 {
00703 if (connectionManagers.isEmpty())
00704 throw new SQLException(Translate.get("backend.null.connection.manager",
00705 new String[]{name, url}));
00706
00707 AbstractConnectionManager connectionManager;
00708 Iterator iter = connectionManagers.values().iterator();
00709 connectionManager = (AbstractConnectionManager) iter.next();
00710
00711 Connection con = connectionManager.getConnectionFromDriver();
00712 con.createStatement().execute(this.connectionTestStatement);
00713 return true;
00714 }
00715 catch (Exception e)
00716 {
00717 String msg = Translate.get("loadbalancer.backend.unreacheable", name);
00718 logger.warn(msg, e);
00719 return false;
00720 }
00721 }
00722
00723
00724
00725
00726
00727
00728 public synchronized boolean isReadEnabled()
00729 {
00730 return state == BackendState.READ_ENABLED_WRITE_DISABLED
00731 || state == BackendState.READ_ENABLED_WRITE_ENABLED;
00732 }
00733
00734
00735
00736
00737
00738
00739 public synchronized boolean isWriteEnabled()
00740 {
00741 return state == BackendState.READ_ENABLED_WRITE_ENABLED
00742 || state == BackendState.READ_DISABLED_WRITE_ENABLED
00743 || state == BackendState.DISABLING;
00744 }
00745
00746
00747
00748
00749
00750
00751 public boolean isRecovering()
00752 {
00753 return state == BackendState.RECOVERING;
00754 }
00755
00756
00757
00758
00759
00760
00761 public boolean isDisabling()
00762 {
00763 return state == BackendState.DISABLING;
00764 }
00765
00766
00767
00768
00769 public boolean isDisabled()
00770 {
00771 return state == BackendState.DISABLED;
00772 }
00773
00774
00775
00776
00777
00778
00779
00780 public boolean isKilled()
00781 {
00782 return state == BackendState.UNKNOWN;
00783 }
00784
00785
00786
00787
00788
00789
00790
00791
00792
00793
00794
00795
00796 public String getState()
00797 {
00798 switch (state)
00799 {
00800 case BackendState.READ_ENABLED_WRITE_DISABLED :
00801 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_ENABLED;
00802 case BackendState.READ_ENABLED_WRITE_ENABLED :
00803 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_ENABLED_WRITE;
00804 case BackendState.READ_DISABLED_WRITE_ENABLED :
00805 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_ENABLED_WRITE;
00806 case BackendState.DISABLING :
00807 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_DISABLING;
00808 case BackendState.BACKUPING :
00809 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_BACKINGUP;
00810 case BackendState.RECOVERING :
00811 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_RECOVERING;
00812 case BackendState.REPLAYING :
00813 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_REPLAYING;
00814 case BackendState.DISABLED :
00815 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_DISABLED;
00816 case BackendState.UNKNOWN :
00817 return CjdbcNotificationList.VIRTUALDATABASE_BACKEND_UNKNOWN;
00818 default :
00819 throw new IllegalArgumentException("Unknown backend state:" + state);
00820 }
00821 }
00822
00823
00824
00825
00826
00827
00828
00829
00830 public int getStateValue()
00831 {
00832 return state;
00833 }
00834
00835
00836
00837
00838
00839 public synchronized void enableRead()
00840 {
00841 if (isWriteEnabled())
00842 setState(BackendState.READ_ENABLED_WRITE_ENABLED);
00843 else
00844 setState(BackendState.READ_ENABLED_WRITE_DISABLED);
00845 }
00846
00847
00848
00849
00850 public synchronized void disableRead()
00851 {
00852 if (isWriteEnabled())
00853 setState(BackendState.READ_DISABLED_WRITE_ENABLED);
00854 else
00855 setState(BackendState.DISABLED);
00856 }
00857
00858
00859
00860
00861 public synchronized void disableWrite()
00862 {
00863 if (isReadEnabled())
00864 setState(BackendState.READ_ENABLED_WRITE_DISABLED);
00865 else
00866 setState(BackendState.DISABLED);
00867 }
00868
00869
00870
00871
00872
00873 public synchronized void enableWrite()
00874 {
00875 if (isReadEnabled())
00876 setState(BackendState.READ_ENABLED_WRITE_ENABLED);
00877 else
00878 setState(BackendState.READ_DISABLED_WRITE_ENABLED);
00879 }
00880
00881
00882
00883
00884
00885
00886
00887
00888
00889
00890
00891
00892
00893 public void setDisabling()
00894 {
00895 setState(BackendState.DISABLING);
00896 }
00897
00898
00899
00900
00901
00902
00903
00904 public synchronized void disable()
00905 {
00906 setState(BackendState.DISABLED);
00907 }
00908
00909
00910
00911
00912
00913
00914
00915
00916
00917
00918 public AbstractConnectionManager getConnectionManager(String vLogin)
00919 {
00920 return (AbstractConnectionManager) connectionManagers.get(vLogin);
00921 }
00922
00923
00924
00925
00926
00927
00928
00929 public HashMap getConnectionManagers()
00930 {
00931 return this.connectionManagers;
00932 }
00933
00934
00935
00936
00937
00938
00939 public String getConnectionTestStatement()
00940 {
00941 return connectionTestStatement;
00942 }
00943
00944
00945
00946
00947 public String getDriverPath()
00948 {
00949 return driverPath;
00950 }
00951
00952
00953
00954
00955 public String getAssociatedString()
00956 {
00957 return "backend";
00958 }
00959
00960
00961
00962
00963
00964
00965 public String getDriverClassName()
00966 {
00967 return driverClassName;
00968 }
00969
00970
00971
00972
00973
00974
00975 public String getName()
00976 {
00977 return name;
00978 }
00979
00980
00981
00982
00983
00984
00985 public String getVirtualDatabaseName()
00986 {
00987 return virtualDatabaseName;
00988 }
00989
00990
00991
00992
00993
00994
00995
00996 public Vector getPendingRequests()
00997 {
00998 return pendingRequests;
00999 }
01000
01001
01002
01003
01004
01005 public ArrayList getPendingRequestsDescription(int count, boolean fromFirst,
01006 boolean clone)
01007 {
01008 int size = pendingRequests.size();
01009 int limit = (count == 0 || count > size) ? size : Math.min(size, count);
01010 ArrayList list = new ArrayList(limit);
01011 int start = (fromFirst) ? 0 : Math.min(limit - count, 0);
01012 if (!clone)
01013 {
01014 synchronized (pendingRequests)
01015 {
01016 for (int i = start; i < limit; i++)
01017 list.add(pendingRequests.get(i).toString());
01018 }
01019 return list;
01020 }
01021 else
01022 {
01023 Vector cloneVector = (Vector) pendingRequests.clone();
01024 for (int i = start; i < limit; i++)
01025 list.add(cloneVector.get(i).toString());
01026 return list;
01027 }
01028 }
01029
01030
01031
01032
01033
01034
01035
01036 public void addPendingReadRequest(Object request)
01037 {
01038 synchronized (this)
01039 {
01040 totalRequest++;
01041 totalReadRequest++;
01042 }
01043 pendingRequests.add(request);
01044 }
01045
01046
01047
01048
01049
01050
01051
01052 public void addPendingWriteRequest(Object request)
01053 {
01054 synchronized (this)
01055 {
01056 totalRequest++;
01057 totalWriteRequest++;
01058 }
01059 pendingRequests.add(request);
01060 }
01061
01062
01063
01064
01065
01066
01067
01068
01069 public boolean removePendingRequest(Object request)
01070 {
01071 return pendingRequests.remove(request);
01072 }
01073
01074
01075
01076
01077
01078
01079
01080
01081 public DatabaseSchema getDatabaseSchema()
01082 {
01083 return schema;
01084 }
01085
01086
01087
01088
01089
01090
01091
01092
01093 public void setDatabaseSchema(DatabaseSchema databaseSchema, boolean isStatic)
01094 {
01095 if (schema == null)
01096 {
01097 schemaIsStatic = isStatic;
01098 schema = databaseSchema;
01099 }
01100 else
01101 {
01102 if (!isStatic)
01103 schema = databaseSchema;
01104 }
01105 }
01106
01107
01108
01109
01110 public void refreshSchema()
01111 {
01112 setDatabaseSchema(null, isSchemaStatic());
01113 checkDatabaseSchema();
01114 }
01115
01116
01117
01118
01119
01120
01121
01122 public MetadataContainer getDatabaseStaticMetadata()
01123 {
01124 AbstractConnectionManager connectionMananger;
01125 Iterator iter = connectionManagers.values().iterator();
01126 if (iter.hasNext())
01127 {
01128 connectionMananger = (AbstractConnectionManager) iter.next();
01129
01130 DatabaseBackendMetaData meta = new DatabaseBackendMetaData(
01131 connectionMananger, logger, dynamicPrecision, gatherSystemTables,
01132 schemaName);
01133 try
01134 {
01135 return meta.retrieveDatabaseMetadata();
01136 }
01137 catch (SQLException e)
01138 {
01139 return null;
01140 }
01141 }
01142 else
01143 return null;
01144 }
01145
01146
01147
01148
01149 public boolean checkDatabaseSchema()
01150 {
01151 if (logger.isDebugEnabled())
01152 logger.debug(Translate.get("backend.dynamic.schema",
01153 DatabaseBackendSchemaConstants
01154 .getDynamicSchemaLevel(dynamicPrecision)));
01155 boolean checked = true;
01156 AbstractConnectionManager connectionMananger;
01157 Iterator iter = connectionManagers.values().iterator();
01158 while (iter.hasNext())
01159 {
01160 connectionMananger = (AbstractConnectionManager) iter.next();
01161
01162
01163 DatabaseBackendMetaData meta = new DatabaseBackendMetaData(
01164 connectionMananger, logger, dynamicPrecision, gatherSystemTables,
01165 schemaName);
01166
01167 DatabaseSchema metaSchema;
01168 try
01169 {
01170 if (logger.isInfoEnabled())
01171 logger.info(Translate.get("backend.gathering.database.schema"));
01172 metaSchema = meta.getDatabaseSchema();
01173 }
01174 catch (SQLException e)
01175 {
01176 if (logger.isWarnEnabled())
01177 logger.warn(Translate.get("backend.gather.schema.failed", e));
01178 return false;
01179 }
01180 if (schema == null)
01181 {
01182 if (logger.isDebugEnabled())
01183 logger.debug(Translate.get("backend.use.gathered.schema.as.new"));
01184 schema = metaSchema;
01185 }
01186 else
01187 {
01188 if (dynamicPrecision == DatabaseBackendSchemaConstants.DynamicPrecisionStatic)
01189 {
01190 if (logger.isInfoEnabled())
01191 logger.info(Translate.get("backend.schema.static.no.check", name));
01192 }
01193 else
01194 {
01195 if (logger.isInfoEnabled())
01196 logger.info(Translate.get("backend.check.schema.compatibility"));
01197 if (schema.isCompatibleSubset(metaSchema))
01198 logger.info(Translate.get("backend.schema.compatible.for.login",
01199 connectionMananger.getLogin()));
01200 else
01201 {
01202 checked = false;
01203 logger.warn(Translate.get(
01204 "backend.schema.not.compatible.for.login", connectionMananger
01205 .getLogin()));
01206 }
01207 }
01208 }
01209 }
01210 return checked;
01211 }
01212
01213
01214
01215
01216 public DriverCompliance getDriverCompliance()
01217 {
01218 return driverCompliance;
01219 }
01220
01221
01222
01223
01224
01225
01226 public void checkDriverCompliance() throws SQLException
01227 {
01228 if (connectionManagers.isEmpty())
01229 throw new SQLException(Translate.get("backend.null.connection.manager",
01230 new String[]{name, url}));
01231
01232 AbstractConnectionManager connectionManager;
01233 Iterator iter = connectionManagers.values().iterator();
01234 connectionManager = (AbstractConnectionManager) iter.next();
01235
01236 try
01237 {
01238 if (!driverCompliance.complianceTest(url, connectionManager.getLogin(),
01239 connectionManager.getPassword(), connectionManager.getDriverPath(),
01240 connectionManager.getDriverClassName(), connectionTestStatement))
01241 throw new SQLException(Translate.get("backend.driver.not.compliant",
01242 driverClassName));
01243 }
01244 catch (ConnectException e)
01245 {
01246 throw new SQLException(Translate.get("backend.cannot.connect.to", e));
01247 }
01248 }
01249
01250
01251
01252
01253
01254
01255 public String getURL()
01256 {
01257 return url;
01258 }
01259
01260
01261
01262
01263
01264
01265
01266
01267
01268
01269 public void addRewritingRule(AbstractRewritingRule rule)
01270 {
01271 if (rewritingRules == null)
01272 rewritingRules = new ArrayList();
01273 if (logger.isDebugEnabled())
01274 logger.debug(Translate.get("backend.rewriting.rule.add", new String[]{
01275 rule.getQueryPattern(), rule.getRewrite()}));
01276 rewritingRules.add(rule);
01277 }
01278
01279
01280
01281
01282
01283
01284
01285 public String rewriteQuery(String sqlQuery)
01286 {
01287 if (rewritingRules == null)
01288 return sqlQuery;
01289 int size = rewritingRules.size();
01290 for (int i = 0; i < size; i++)
01291 {
01292 AbstractRewritingRule rule = (AbstractRewritingRule) rewritingRules
01293 .get(i);
01294 sqlQuery = rule.rewrite(sqlQuery);
01295 if (rule.hasMatched())
01296 {
01297 if (logger.isDebugEnabled())
01298 logger.debug(Translate.get("backend.rewriting.query", sqlQuery));
01299 if (rule.isStopOnMatch())
01300 break;
01301 }
01302 }
01303 return sqlQuery;
01304 }
01305
01306
01307
01308
01309
01310
01311
01312
01313
01314
01315 public String getXml()
01316 {
01317 StringBuffer info = new StringBuffer();
01318 info.append("<" + DatabasesXmlTags.ELT_DatabaseBackend + " "
01319 + DatabasesXmlTags.ATT_name + "=\"" + name + "\" "
01320 + DatabasesXmlTags.ATT_driver + "=\"" + driverClassName + "\" "
01321 + DatabasesXmlTags.ATT_url + "=\"" + url + "\" "
01322 + DatabasesXmlTags.ATT_connectionTestStatement + "=\""
01323 + connectionTestStatement + "\">");
01324
01325 boolean expandSchema = this.schema != null
01326 && dynamicPrecision == DatabaseBackendSchemaConstants.DynamicPrecisionStatic;
01327
01328 info.append(getSchemaXml(expandSchema));
01329
01330 if (rewritingRules != null)
01331 {
01332 int size = rewritingRules.size();
01333 for (int i = 0; i < size; i++)
01334 info.append(((AbstractRewritingRule) rewritingRules.get(i)).getXml());
01335 }
01336 if (connectionManagers != null)
01337 {
01338 if (connectionManagers.isEmpty() == false)
01339 {
01340 AbstractConnectionManager connectionManager;
01341 Iterator iter = connectionManagers.values().iterator();
01342 while (iter.hasNext())
01343 {
01344 connectionManager = (AbstractConnectionManager) iter.next();
01345 info.append(connectionManager.getXml());
01346 }
01347 }
01348 }
01349 info.append("</" + DatabasesXmlTags.ELT_DatabaseBackend + ">");
01350 return info.toString();
01351 }
01352
01353
01354
01355
01356 public String getSchemaXml(boolean expandSchema)
01357 {
01358 StringBuffer info = new StringBuffer();
01359 info.append("<"
01360 + DatabasesXmlTags.ELT_DatabaseSchema
01361 + " "
01362 + DatabasesXmlTags.ATT_dynamicPrecision
01363 + "=\""
01364 + DatabaseBackendSchemaConstants
01365 .getDynamicSchemaLevel(dynamicPrecision) + "\" "
01366 + DatabasesXmlTags.ATT_gatherSystemTables + "=\""
01367 + (gatherSystemTables ? "true" : "false") + "\">");
01368 if (expandSchema)
01369 info.append(schema.getXml());
01370 info.append("</" + DatabasesXmlTags.ELT_DatabaseSchema + ">");
01371 return info.toString();
01372 }
01373
01374
01375
01376
01377 public ArrayList getActiveTransactions()
01378 {
01379 return activeTransactions;
01380 }
01381
01382
01383
01384
01385 public boolean isSchemaStatic()
01386 {
01387 return schemaIsStatic;
01388 }
01389
01390
01391
01392
01393
01394
01395
01396
01397
01398
01399
01400
01401
01402
01403
01404
01405
01406
01407
01408
01409
01410
01411
01412 public String[] getBackendData()
01413 {
01414 String[] data = new String[14];
01415 data[0] = this.name;
01416 data[1] = this.driverClassName;
01417 data[2] = this.url;
01418 data[3] = String.valueOf(this.activeTransactions.size());
01419 data[4] = String.valueOf(this.pendingRequests.size());
01420 data[5] = String.valueOf(this.isReadEnabled());
01421 data[6] = String.valueOf(this.isWriteEnabled());
01422 try
01423 {
01424 data[7] = String.valueOf(this.isInitialized());
01425 }
01426 catch (Exception e)
01427 {
01428 data[7] = "unknown";
01429 }
01430 data[8] = String.valueOf(this.schemaIsStatic);
01431
01432 data[9] = String.valueOf(this.connectionManagers.size());
01433 data[10] = String.valueOf(getTotalActiveConnections());
01434 data[11] = String.valueOf(totalRequest);
01435 data[12] = String.valueOf(totalTransactions);
01436 if (lastKnownCheckpoint == null || lastKnownCheckpoint.equalsIgnoreCase(""))
01437 data[13] = "<unknown>";
01438 else
01439 data[13] = lastKnownCheckpoint;
01440 return data;
01441 }
01442
01443
01444
01445
01446
01447
01448
01449 public long getTotalActiveConnections()
01450 {
01451 int activeConnections = 0;
01452 Iterator iter = connectionManagers.keySet().iterator();
01453 while (iter.hasNext())
01454 activeConnections += ((AbstractConnectionManager) connectionManagers
01455 .get(iter.next())).getCurrentNumberOfConnections();
01456 return activeConnections;
01457 }
01458
01459
01460
01461
01462 public int getDynamicPrecision()
01463 {
01464 return dynamicPrecision;
01465 }
01466
01467
01468
01469
01470
01471
01472
01473
01474
01475 public void setDynamicPrecision(int dynamicPrecision,
01476 boolean gatherSystemTables, String schemaName)
01477 {
01478 this.dynamicPrecision = dynamicPrecision;
01479 this.gatherSystemTables = gatherSystemTables;
01480 this.schemaName = schemaName;
01481 }
01482
01483
01484
01485
01486
01487
01488 public int getTotalTransactions()
01489 {
01490 return totalTransactions;
01491 }
01492
01493
01494
01495
01496
01497
01498 public int getTotalReadRequest()
01499 {
01500 return totalReadRequest;
01501 }
01502
01503
01504
01505
01506
01507
01508 public int getTotalWriteRequest()
01509 {
01510 return totalWriteRequest;
01511 }
01512
01513
01514
01515
01516
01517
01518 public int getTotalRequest()
01519 {
01520 return totalRequest;
01521 }
01522
01523
01524
01525
01526
01527
01528 public void setLastKnownCheckpoint(String checkpoint)
01529 {
01530 this.lastKnownCheckpoint = checkpoint;
01531 }
01532
01533
01534
01535
01536
01537
01538 public String getLastKnownCheckpoint()
01539 {
01540 return lastKnownCheckpoint;
01541 }
01542
01543
01544
01545
01546
01547
01548 public String getDatabaseProductName()
01549 {
01550 return driverCompliance.getDatabaseProductName();
01551 }
01552
01553
01554
01555
01556
01557
01558 public ArrayList getRewritingRules()
01559 {
01560 return rewritingRules;
01561 }
01562
01563
01564
01565
01566
01567
01568 public void setRewritingRules(ArrayList rewritingRules)
01569 {
01570 this.rewritingRules = rewritingRules;
01571 }
01572
01573
01574
01575
01576
01577
01578
01579
01580
01581
01582
01583
01584
01585
01586
01587
01588
01589
01590
01591
01592
01593 public DatabaseBackend copy(String newName, Map parameters) throws Exception
01594 {
01595
01596
01597 String driverPath = parameters.containsKey(DatabasesXmlTags.ATT_driverPath)
01598 ? (String) parameters.get(DatabasesXmlTags.ATT_driverPath)
01599 : this.getDriverPath();
01600
01601 String driverClassName = parameters
01602 .containsKey(DatabasesXmlTags.ATT_driver) ? (String) parameters
01603 .get(DatabasesXmlTags.ATT_driver) : this.getDriverClassName();
01604
01605 String url = parameters.containsKey(DatabasesXmlTags.ATT_url)
01606 ? (String) parameters.get(DatabasesXmlTags.ATT_url)
01607 : this.getURL();
01608
01609 String connectionTestStatement = parameters
01610 .containsKey(DatabasesXmlTags.ATT_connectionTestStatement)
01611 ? (String) parameters.get(DatabasesXmlTags.ATT_connectionTestStatement)
01612 : this.getConnectionTestStatement();
01613
01614
01615 DatabaseBackend newBackend = new DatabaseBackend(newName, driverPath,
01616 driverClassName, url, virtualDatabaseName, writeCanBeEnabled,
01617 connectionTestStatement);
01618
01619
01620
01621 newBackend.setRewritingRules(this.getRewritingRules());
01622
01623
01624 HashMap connectionManagers = this.getConnectionManagers();
01625 Iterator iter = connectionManagers.keySet().iterator();
01626
01627 String vlogin = null;
01628 AbstractConnectionManager connectionManager;
01629 while (iter.hasNext())
01630 {
01631 vlogin = (String) iter.next();
01632 connectionManager = (AbstractConnectionManager) connectionManagers
01633 .get(vlogin);
01634 newBackend.addConnectionManager(vlogin, connectionManager.copy(url,
01635 newName));
01636 }
01637
01638 return newBackend;
01639
01640 }
01641
01642
01643
01644
01645
01646
01647 public boolean isBackuping()
01648 {
01649 return state == BackendState.BACKUPING;
01650 }
01651
01652
01653
01654
01655
01656
01657
01658 public synchronized void setState(int state)
01659 {
01660 switch (state)
01661 {
01662 case BackendState.UNKNOWN :
01663 lastKnownCheckpoint = null;
01664 case BackendState.READ_ENABLED_WRITE_DISABLED :
01665 case BackendState.READ_ENABLED_WRITE_ENABLED :
01666 case BackendState.READ_DISABLED_WRITE_ENABLED :
01667 case BackendState.DISABLING :
01668 case BackendState.BACKUPING :
01669 case BackendState.RECOVERING :
01670 case BackendState.REPLAYING :
01671 case BackendState.DISABLED :
01672 this.state = state;
01673 if (logger.isDebugEnabled())
01674 logger.debug(Translate.get("backend.state.changed", new String[]{
01675 name, getState()}));
01676 notifyStateChange();
01677 break;
01678 default :
01679 throw new IllegalArgumentException("Unknown backend state:" + state);
01680 }
01681 }
01682
01683
01684
01685
01686
01687
01688
01689
01690
01691 public void notifyStateChange()
01692 {
01693 if (stateListener != null)
01694 stateListener.changeState(this);
01695 notifyJmx(getState());
01696 }
01697
01698
01699
01700
01701
01702
01703
01704 public void notifyJmx(String type)
01705 {
01706 notifyJmx(type, CjdbcNotificationList.NOTIFICATION_LEVEL_INFO, Translate
01707 .get(type, getName()));
01708 }
01709
01710
01711
01712
01713
01714
01715
01716
01717 public void notifyJmxError(String type, Exception e)
01718 {
01719 notifyJmx(type, CjdbcNotificationList.NOTIFICATION_LEVEL_ERROR, Translate
01720 .get(type, new String[]{getName(), e.getMessage()}));
01721
01722 }
01723
01724 private void notifyJmx(String type, String level, String message)
01725 {
01726 if (MBeanServerManager.isJmxEnabled())
01727 {
01728
01729 Hashtable data = new Hashtable();
01730 data.put(CjdbcNotificationList.DATA_DATABASE, getVirtualDatabaseName());
01731 data.put(CjdbcNotificationList.DATA_DRIVER, getDriverClassName());
01732 String checkpoint = getLastKnownCheckpoint();
01733 checkpoint = (checkpoint == null) ? "" : checkpoint;
01734 data.put(CjdbcNotificationList.DATA_CHECKPOINT, checkpoint);
01735 data.put(CjdbcNotificationList.DATA_NAME, getName());
01736 data.put(CjdbcNotificationList.DATA_URL, getURL());
01737 RmiConnector.broadcastNotification(this, type, level, message, data);
01738 }
01739 }
01740
01741
01742
01743
01744
01745
01746 public boolean isWriteCanBeEnabled()
01747 {
01748 return writeCanBeEnabled;
01749 }
01750
01751
01752
01753
01754
01755
01756 public void setStateListener(BackendStateListener stateListener)
01757 {
01758 this.stateListener = stateListener;
01759 }
01760
01761
01762
01763
01764
01765
01766 public String toString()
01767 {
01768 return "Backend: Name[" + this.name + "] State[" + this.state
01769 + "] JDBCConnected[" + isJDBCConnected() + "] ActiveTransactions["
01770 + activeTransactions.size() + "] PendingRequests["
01771 + pendingRequests.size() + "]";
01772 }
01773
01774
01775
01776
01777
01778 private void writeObject(java.io.ObjectOutputStream out) throws IOException
01779 {
01780 out.writeUTF(name);
01781 out.writeUTF(connectionTestStatement);
01782 out.writeUTF(driverClassName);
01783 out.writeUTF(url);
01784
01785 boolean b1 = driverPath != null;
01786 out.writeBoolean(b1);
01787 if (b1)
01788 out.writeUTF(driverPath);
01789
01790 boolean b2 = lastKnownCheckpoint != null;
01791 out.writeBoolean(b2);
01792 if (b2)
01793 out.writeUTF(lastKnownCheckpoint);
01794
01795 out.writeInt(state);
01796 out.writeInt(totalReadRequest);
01797 out.writeInt(totalRequest);
01798 out.writeInt(totalTransactions);
01799 out.writeInt(totalWriteRequest);
01800 out.writeInt(sqlShortFormLength);
01801
01802 out.writeBoolean(schemaIsStatic);
01803 out.writeBoolean(writeCanBeEnabled);
01804
01805 out.flush();
01806 }
01807
01808 private void readObject(java.io.ObjectInputStream in) throws IOException
01809 {
01810 this.name = in.readUTF();
01811 this.connectionTestStatement = in.readUTF();
01812 this.driverClassName = in.readUTF();
01813 this.url = in.readUTF();
01814
01815 if (in.readBoolean())
01816 this.driverPath = in.readUTF();
01817 if (in.readBoolean())
01818 this.lastKnownCheckpoint = in.readUTF();
01819
01820 this.state = in.readInt();
01821 this.totalReadRequest = in.readInt();
01822 this.totalRequest = in.readInt();
01823 this.totalTransactions = in.readInt();
01824 this.totalWriteRequest = in.readInt();
01825 this.sqlShortFormLength = in.readInt();
01826
01827 this.schemaIsStatic = in.readBoolean();
01828 this.writeCanBeEnabled = in.readBoolean();
01829 }
01830 }