~ubuntu-branches/ubuntu/natty/eucalyptus/natty-updates

« back to all changes in this revision

Viewing changes to clc/modules/cloud/src/main/java/com/eucalyptus/bootstrap/LocalDatabaseBootstrapper.java

  • Committer: Bazaar Package Importer
  • Author(s): Dustin Kirkland
  • Date: 2009-12-17 18:22:02 UTC
  • mto: This revision was merged to the branch mainline in revision 83.
  • Revision ID: james.westby@ubuntu.com-20091217182202-0v2v09ry3cxrvh84
Tags: upstream-1.6.2~bzr1103
Import upstream version 1.6.2~bzr1103

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*******************************************************************************
2
 
 *Copyright (c) 2009 Eucalyptus Systems, Inc.
3
 
 * 
4
 
 * This program is free software: you can redistribute it and/or modify
5
 
 * it under the terms of the GNU General Public License as published by
6
 
 * the Free Software Foundation, only version 3 of the License.
7
 
 * 
8
 
 * 
9
 
 * This file is distributed in the hope that it will be useful, but WITHOUT
10
 
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11
 
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12
 
 * for more details.
13
 
 * 
14
 
 * You should have received a copy of the GNU General Public License along
15
 
 * with this program. If not, see <http://www.gnu.org/licenses/>.
16
 
 * 
17
 
 * Please contact Eucalyptus Systems, Inc., 130 Castilian
18
 
 * Dr., Goleta, CA 93101 USA or visit <http://www.eucalyptus.com/licenses/>
19
 
 * if you need additional information or have any questions.
20
 
 * 
21
 
 * This file may incorporate work covered under the following copyright and
22
 
 * permission notice:
23
 
 * 
24
 
 * Software License Agreement (BSD License)
25
 
 * 
26
 
 * Copyright (c) 2008, Regents of the University of California
27
 
 * All rights reserved.
28
 
 * 
29
 
 * Redistribution and use of this software in source and binary forms, with
30
 
 * or without modification, are permitted provided that the following
31
 
 * conditions are met:
32
 
 * 
33
 
 * Redistributions of source code must retain the above copyright notice,
34
 
 * this list of conditions and the following disclaimer.
35
 
 * 
36
 
 * Redistributions in binary form must reproduce the above copyright
37
 
 * notice, this list of conditions and the following disclaimer in the
38
 
 * documentation and/or other materials provided with the distribution.
39
 
 * 
40
 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
41
 
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
42
 
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
43
 
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
44
 
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
45
 
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
46
 
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
47
 
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
48
 
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
49
 
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
50
 
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. USERS OF
51
 
 * THIS SOFTWARE ACKNOWLEDGE THE POSSIBLE PRESENCE OF OTHER OPEN SOURCE
52
 
 * LICENSED MATERIAL, COPYRIGHTED MATERIAL OR PATENTED MATERIAL IN THIS
53
 
 * SOFTWARE, AND IF ANY SUCH MATERIAL IS DISCOVERED THE PARTY DISCOVERING
54
 
 * IT MAY INFORM DR. RICH WOLSKI AT THE UNIVERSITY OF CALIFORNIA, SANTA
55
 
 * BARBARA WHO WILL THEN ASCERTAIN THE MOST APPROPRIATE REMEDY, WHICH IN
56
 
 * THE REGENTS’ DISCRETION MAY INCLUDE, WITHOUT LIMITATION, REPLACEMENT
57
 
 * OF THE CODE SO IDENTIFIED, LICENSING OF THE CODE SO IDENTIFIED, OR
58
 
 * WITHDRAWAL OF THE CODE CAPABILITY TO THE EXTENT NEEDED TO COMPLY WITH
59
 
 * ANY SUCH LICENSES OR RIGHTS.
60
 
 *******************************************************************************/
61
 
/*
62
 
 * Author: chris grzegorczyk <grze@eucalyptus.com>
63
 
 */
64
 
package com.eucalyptus.bootstrap;
65
 
 
66
 
import java.util.concurrent.atomic.AtomicBoolean;
67
 
 
68
 
import org.apache.log4j.Logger;
69
 
import org.hsqldb.DatabaseURL;
70
 
import org.hsqldb.Server;
71
 
import org.hsqldb.ServerConstants;
72
 
import org.hsqldb.persist.HsqlProperties;
73
 
 
74
 
import com.eucalyptus.auth.util.SslSetup;
75
 
import com.eucalyptus.event.ClockTick;
76
 
import com.eucalyptus.event.Event;
77
 
import com.eucalyptus.event.EventListener;
78
 
import com.eucalyptus.event.ListenerRegistry;
79
 
import com.eucalyptus.util.DatabaseUtil;
80
 
import com.eucalyptus.util.DebugUtil;
81
 
import com.eucalyptus.util.GroovyUtil;
82
 
import com.eucalyptus.util.LogUtil;
83
 
 
84
 
@Provides( resource = Resource.Database )
85
 
@Depends( resources = Resource.SystemCredentials, local = Component.eucalyptus )
86
 
public class LocalDatabaseBootstrapper extends Bootstrapper implements EventListener, Runnable, DatabaseBootstrapper {
87
 
  private static Logger             LOG = Logger.getLogger( LocalDatabaseBootstrapper.class );
88
 
  private static DatabaseBootstrapper singleton;
89
 
 
90
 
  public static DatabaseBootstrapper getInstance( ) {
91
 
    synchronized ( LocalDatabaseBootstrapper.class ) {
92
 
      if ( singleton == null ) {
93
 
        for( DatabaseUtil.PoolConfig p : DatabaseUtil.PoolConfig.values( ) ) {
94
 
          LOG.debug( String.format( "Default pool config: %s=%s", p.getKey( ), p.getValue( ) ) ); 
95
 
        }
96
 
        singleton = new LocalDatabaseBootstrapper( );
97
 
        SystemBootstrapper.setDatabaseBootstrapper( singleton );
98
 
      }
99
 
    }
100
 
    return singleton;
101
 
  }
102
 
 
103
 
  private static void setDefault( String key, Object value ) {
104
 
    if( System.getProperties( ).containsKey( key ) )  System.setProperty( key, value.toString() );
105
 
  }
106
 
 
107
 
  private Server db;
108
 
  private String fileName;
109
 
 
110
 
  private LocalDatabaseBootstrapper( ) {
111
 
    ListenerRegistry.getInstance( ).register( ClockTick.class, this );
112
 
  }
113
 
 
114
 
  @Override
115
 
  public boolean check( ) {
116
 
    return false;
117
 
  }
118
 
 
119
 
  @Override
120
 
  public boolean destroy( ) throws Exception {
121
 
    return false;
122
 
  }
123
 
 
124
 
  @Override
125
 
  public boolean load( Resource current ) throws Exception {
126
 
    LOG.debug( "Initializing SSL just in case: " + SslSetup.class );
127
 
    try {
128
 
      DatabaseConfig.initialize( );
129
 
    } catch ( Exception e ) {
130
 
      LOG.debug( e, e );
131
 
    }
132
 
    createDatabase( );
133
 
    return true;
134
 
  }
135
 
 
136
 
  public boolean isRunning( ) {
137
 
    try {
138
 
      if ( this.db != null ) this.db.checkRunning( true );
139
 
      return true;
140
 
    } catch ( RuntimeException e ) {
141
 
      return false;
142
 
    }
143
 
  }
144
 
 
145
 
  private static AtomicBoolean hup = new AtomicBoolean( false );
146
 
  public void hup( ) {
147
 
    if( hup.compareAndSet( false, true ) ) {
148
 
      DebugUtil.printDebugDetails( );
149
 
      try {
150
 
        if ( this.db != null ) { 
151
 
          this.db.checkRunning( true );
152
 
          this.db.start( );
153
 
        }
154
 
      } catch ( RuntimeException e ) {
155
 
        this.db.stop( );
156
 
        this.createDatabase( );
157
 
        this.waitForDatabase( );        
158
 
      }
159
 
    } else {
160
 
      LOG.info("Already scheduled a database restart.");
161
 
    }
162
 
  }
163
 
 
164
 
  private void createDatabase( ) {
165
 
    this.db = new Server( );
166
 
    this.db.setProperties( new HsqlProperties( DatabaseConfig.getProperties( ) ) );
167
 
    SystemBootstrapper.makeSystemThread( this ).start( );
168
 
  }
169
 
 
170
 
  @Override
171
 
  public boolean start( ) throws Exception {
172
 
    this.waitForDatabase( );
173
 
    try {
174
 
      GroovyUtil.evaluateScript( "startup.groovy" );
175
 
    } catch ( Exception e ) {
176
 
      LOG.fatal( e, e );
177
 
      System.exit( -1 );
178
 
    }
179
 
    return true;
180
 
  }
181
 
 
182
 
  private void waitForDatabase( ) {
183
 
    while ( this.db.getState( ) != 1 ) {
184
 
      Throwable t = this.db.getServerError( );
185
 
      if ( t != null ) {
186
 
        LOG.error( t, t );
187
 
        throw new RuntimeException( t );
188
 
      }
189
 
      try {
190
 
        Thread.sleep( 1000 );
191
 
      } catch ( InterruptedException e ) {
192
 
        Thread.currentThread( ).interrupt( );
193
 
      }
194
 
      LOG.info( "Waiting for database to start..." );
195
 
    } 
196
 
  }
197
 
 
198
 
  public void run( ) {
199
 
    this.db.start( );
200
 
  }
201
 
 
202
 
  @Override
203
 
  public boolean stop( ) {
204
 
    return false;
205
 
  }
206
 
 
207
 
  public String getFileName( ) {
208
 
    return fileName;
209
 
  }
210
 
 
211
 
  public void setFileName( String fileName ) {
212
 
    LOG.info( "Setting hsqldb filename=" + fileName );
213
 
    this.fileName = fileName;
214
 
  }
215
 
 
216
 
  @Override
217
 
  public void advertiseEvent( Event event ) {}
218
 
 
219
 
  @Override
220
 
  public void fireEvent( Event event ) {
221
 
    if( event instanceof ClockTick && Component.eucalyptus.isLocal( ) ) {
222
 
      try {
223
 
        LOG.trace( "-> Ping database." );
224
 
        this.isRunning( );
225
 
      } catch ( RuntimeException e ) {
226
 
        LOG.fatal( e, e );
227
 
      }
228
 
    }
229
 
  }
230
 
 
231
 
}