/*
  * (non-Javadoc)
  *
  * @see java.lang.Object#toString()
  */
 @Override
 public String toString() {
   final StringBuilder sb = new StringBuilder("LensConnection{");
   sb.append("sessionHandle=").append(sessionHandle.getPublicId());
   sb.append('}');
   return sb.toString();
 }
Пример #2
0
  /**
   * Test session expiry.
   *
   * @throws Exception the exception
   */
  public void testSessionExpiry() throws Exception {
    HiveConf conf = LensServerConf.createHiveConf();
    conf.setVar(HiveConf.ConfVars.HIVE_SESSION_IMPL_CLASSNAME, LensSessionImpl.class.getName());
    conf.setLong(LensConfConstants.SESSION_TIMEOUT_SECONDS, 1L);
    CLIService cliService = new CLIService();
    cliService.init(conf);
    HiveSessionService lensService = new HiveSessionService(cliService);
    lensService.init(conf);
    lensService.start();
    MetricsService metricSvc = LensServices.get().getService(MetricsService.NAME);

    try {
      LensSessionHandle sessionHandle =
          lensService.openSession("foo", "bar", new HashMap<String, String>());
      LensSessionImpl session = lensService.getSession(sessionHandle);
      assertTrue(session.isActive());
      session.setLastAccessTime(
          session.getLastAccessTime()
              - 2000
                  * conf.getLong(
                      LensConfConstants.SESSION_TIMEOUT_SECONDS,
                      LensConfConstants.SESSION_TIMEOUT_SECONDS_DEFAULT));
      assertFalse(session.isActive());
      // run the expiry thread
      lensService.getSessionExpiryRunnable().run();
      log.info(
          "Keeping a sleep of 3 seconds to make sure SessionExpiryService gets enough time to close"
              + " inactive sessions");
      Thread.sleep(3000);
      assertTrue(metricSvc.getTotalExpiredSessions() >= 1);
      assertTrue(metricSvc.getTotalClosedSessions() >= 1);

      try {
        lensService.getSession(sessionHandle);
        // should throw exception since session should be expired by now
        fail("Expected get session to fail for session " + sessionHandle.getPublicId());
      } catch (Exception e) {
        // pass
      }
    } finally {
      lensService.stop();
    }
  }
Пример #3
0
 /**
  * Create a new session with Lens server.
  *
  * @param username User name of the Lens server user
  * @param password Password of the Lens server user
  * @param database (Optional) Set current database to supplied value
  * @param sessionconf Key-value properties which will be used to configure this session
  * @return A Session handle unique to this session
  */
 @POST
 @Consumes({MediaType.MULTIPART_FORM_DATA})
 @Produces({MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON, MediaType.TEXT_PLAIN})
 public LensSessionHandle openSession(
     @FormDataParam("username") String username,
     @FormDataParam("password") String password,
     @FormDataParam("database") @DefaultValue("") String database,
     @FormDataParam("sessionconf") LensConf sessionconf) {
   try {
     Map<String, String> conf;
     if (sessionconf != null) {
       conf = sessionconf.getProperties();
     } else {
       conf = new HashMap<String, String>();
     }
     LensSessionHandle handle = sessionService.openSession(username, password, database, conf);
     openSessions.put(handle.getPublicId(), handle);
     return handle;
   } catch (LensException e) {
     throw new WebApplicationException(e);
   }
 }
  /**
   * Open.
   *
   * @param password the password
   * @return the lens session handle
   */
  public LensSessionHandle open(String password) {

    WebTarget target = getSessionWebTarget();
    FormDataMultiPart mp = new FormDataMultiPart();
    mp.bodyPart(
        new FormDataBodyPart(
            FormDataContentDisposition.name("username").build(), params.getUser()));
    mp.bodyPart(
        new FormDataBodyPart(FormDataContentDisposition.name("password").build(), password));

    String database = params.getDbName();
    mp.bodyPart(
        new FormDataBodyPart(FormDataContentDisposition.name("database").build(), database));

    mp.bodyPart(
        new FormDataBodyPart(
            FormDataContentDisposition.name("sessionconf").fileName("sessionconf").build(),
            params.getSessionConf(),
            MediaType.APPLICATION_XML_TYPE));
    try {
      Response response =
          target.request().post(Entity.entity(mp, MediaType.MULTIPART_FORM_DATA_TYPE));
      if (response.getStatus() != 200) {
        throw new LensClientServerConnectionException(response.getStatus());
      }
      final LensSessionHandle handle = response.readEntity(LensSessionHandle.class);
      if (handle != null) {
        sessionHandle = handle;
        LOG.debug("Created a new session " + sessionHandle.getPublicId());
      } else {
        throw new IllegalStateException(
            "Unable to connect to lens " + "server with following paramters" + params);
      }
    } catch (ProcessingException e) {
      if (e.getCause() != null && e.getCause() instanceof ConnectException) {
        throw new LensClientServerConnectionException(e.getCause().getMessage(), e);
      }
    }

    LOG.debug("Successfully switched to database " + params.getDbName());
    open.set(true);

    return sessionHandle;
  }