Db4oClientServer.java :  » Database-DBMS » db4o-6.4 » db4ounit » extensions » fixtures » Java Open Source

/* Copyright (C) 2004 - 2007  db4objects Inc.  http://www.db4o.com

This file is part of the db4o open source object database.

db4o is free software; you can redistribute it and/or modify it under
the terms of version 2 of the GNU General Public License as published
by the Free Software Foundation and as clarified by db4objects' GPL 
interpretation policy, available at
http://www.db4o.com/about/company/legalpolicies/gplinterpretation/
Alternatively you can write to db4objects, Inc., 1900 S Norfolk Street,
Suite 350, San Mateo, CA 94403, USA.

db4o is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
59 Temple Place - Suite 330, Boston, MA  02111-1307, USA. */
package db4ounit.extensions.fixtures;

import java.io.*;

import com.db4o.*;
import com.db4o.config.*;
import com.db4o.ext.*;
import com.db4o.internal.*;

import db4ounit.extensions.*;
import db4ounit.extensions.util.*;


public class Db4oClientServer extends
    AbstractDb4oFixture implements Db4oClientServerFixture {
    
  protected static final String FILE = "Db4oClientServer.yap";
    
    public static final String HOST = "localhost";

    public static final String USERNAME = "db4o";

    public static final String PASSWORD = USERNAME;

    private ObjectServer _server;

    private final File _yap;

  private boolean _embeddedClient;

  private ExtObjectContainer _objectContainer;
  
  private String _label;
    
    private int _port;

  private Configuration _serverConfig;
    
    
    public Db4oClientServer(ConfigurationSource configSource,String fileName, boolean embeddedClient, String label) {
      super(configSource);
        _yap = new File(fileName);
        _embeddedClient = embeddedClient;
        _label = label;
    }
    
    public Db4oClientServer(ConfigurationSource configSource, boolean embeddedClient, String label){
        this(configSource,filePath(), embeddedClient, label);
    }
    
    public void open() throws Exception {
    openServer();
    _objectContainer = _embeddedClient ? openEmbeddedClient().ext() : Db4o
        .openClient(config(), HOST, _port, USERNAME, PASSWORD).ext();
  }

  public ExtObjectContainer openNewClient() {
    return _embeddedClient ? openEmbeddedClient().ext() : Db4o.openClient(
        cloneDb4oConfiguration((Config4Impl) config()), HOST, _port,
        USERNAME, PASSWORD).ext();
  }

    private void openServer() throws Exception {
        _serverConfig = cloneDb4oConfiguration(config());
    _server = Db4o.openServer(_serverConfig,_yap.getAbsolutePath(), -1);
        _port = _server.ext().port();
        _server.grantAccess(USERNAME, PASSWORD);
    }
    
    public void close() throws Exception {
    if (null != _objectContainer) {
      _objectContainer.close();
      _objectContainer = null;
    }
    closeServer();
  }

    private void closeServer() throws Exception {
      if (null != _server) {
          _server.close();
          _server = null;
      }
    }  

  public ExtObjectContainer db() {
    return _objectContainer;
  }
    
    protected void doClean() {
        _yap.delete();
    }
    
    public ObjectServer server() {
      return _server;
    }
    
    public boolean embeddedClients() {
      return _embeddedClient;
    }
    
    /**
   * Does not accept a clazz which is assignable from OptOutCS, or not
   * assignable from Db4oTestCase.
   * 
   * @return returns false if the clazz is assignable from OptOutCS, or not
   *         assignable from Db4oTestCase. Otherwise, returns true.
   */
  public boolean accept(Class clazz) {
    if (!AbstractDb4oTestCase.class.isAssignableFrom(clazz)) {
      return false;
    }
    if (OptOutCS.class.isAssignableFrom(clazz)) {
      return false;
    }
    if (_embeddedClient && (OptOutAllButNetworkingCS.class.isAssignableFrom(clazz))) {
      return false;
    }
    return true;
  }
    
  public LocalObjectContainer fileSession() {
    return (LocalObjectContainer)_server.ext().objectContainer();
  }
  
  public void defragment() throws Exception {
    defragment(filePath());
  }
  
  private ObjectContainer openEmbeddedClient() {
    return _server.openClient(config());
  }
  
  private Config4Impl cloneDb4oConfiguration(Configuration config) {
    return (Config4Impl) ((Config4Impl)config).deepClone(this);
  }

  public String getLabel() {
    return _label;
  }

  public int serverPort() {
    return _port;
  }

  private static String filePath() {
    return CrossPlatformServices.databasePath(FILE);
  }

  public void configureAtRuntime(RuntimeConfigureAction action) {
    action.apply(config());
    action.apply(_serverConfig);
  }
}