com.netflix.astyanax.thrift.ThriftFamilyFactory Java Examples

The following examples show how to use com.netflix.astyanax.thrift.ThriftFamilyFactory. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source File: CassandraMutagenImplTest.java    From mutagen-cassandra with Apache License 2.0 6 votes vote down vote up
private static void defineKeyspace() {
		context=new AstyanaxContext.Builder()
			.forKeyspace("mutagen_test")
			.withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
//					.setDiscoveryType(NodeDiscoveryType.RING_DESCRIBE)
//					.setCqlVersion("3.0.0")
//					.setTargetCassandraVersion("1.2")
				.setDefaultReadConsistencyLevel(ConsistencyLevel.CL_QUORUM)
				.setDefaultWriteConsistencyLevel(ConsistencyLevel.CL_QUORUM)
			)
			.withConnectionPoolConfiguration(
				new ConnectionPoolConfigurationImpl("testPool")
				.setPort(9160)
				.setMaxConnsPerHost(1)
				.setSeeds("localhost")
			)
			.withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
			.buildKeyspace(ThriftFamilyFactory.getInstance());

		context.start();
		keyspace=context.getClient();
	}
 
Example #2
Source File: AstyanaxSupport.java    From brooklyn-library with Apache License 2.0 6 votes vote down vote up
public AstyanaxContext<Keyspace> newAstyanaxContextForKeyspace(String keyspace) {
    AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
            .forCluster(clusterName)
            .forKeyspace(keyspace)
            .withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
                    .setDiscoveryType(NodeDiscoveryType.NONE))
            .withConnectionPoolConfiguration(new ConnectionPoolConfigurationImpl("BrooklynPool")
                    .setPort(thriftPort)
                    .setMaxConnsPerHost(1)
                    .setConnectTimeout(5000) // 10s
                    .setSeeds(String.format("%s:%d", hostname, thriftPort)))
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());

    context.start();
    return context;
}
 
Example #3
Source File: AstyanaxSupport.java    From brooklyn-library with Apache License 2.0 6 votes vote down vote up
public AstyanaxContext<Cluster> newAstyanaxContextForCluster() {
    AstyanaxContext<Cluster> context = new AstyanaxContext.Builder()
            .forCluster(clusterName)
            .withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
                    .setDiscoveryType(NodeDiscoveryType.NONE))
            .withConnectionPoolConfiguration(new ConnectionPoolConfigurationImpl("BrooklynPool")
                    .setPort(thriftPort)
                    .setMaxConnsPerHost(1)
                    .setConnectTimeout(5000) // 10s
                    .setSeeds(String.format("%s:%d", hostname, thriftPort)))
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildCluster(ThriftFamilyFactory.getInstance());

    context.start();
    return context;
}
 
Example #4
Source File: InstanceDataDAOCassandra.java    From Raigad with Apache License 2.0 6 votes vote down vote up
private AstyanaxContext<Keyspace> initWithThriftDriverWithEurekaHostsSupplier() {
    logger.info("Boot cluster (BOOT_CLUSTER) is {}, keyspace name (KS_NAME) is {}", BOOT_CLUSTER, KS_NAME);

    return new AstyanaxContext.Builder()
            .forCluster(BOOT_CLUSTER)
            .forKeyspace(KS_NAME)
            .withAstyanaxConfiguration(
                    new AstyanaxConfigurationImpl()
                            .setDiscoveryType(
                                    NodeDiscoveryType.DISCOVERY_SERVICE))
            .withConnectionPoolConfiguration(
                    new ConnectionPoolConfigurationImpl(
                            "MyConnectionPool")
                            .setMaxConnsPerHost(3)
                            .setPort(thriftPortForAstyanax))
            .withHostSupplier(eurekaHostsSupplier.getSupplier(BOOT_CLUSTER))
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());

}
 
Example #5
Source File: InstanceDataDAOCassandra.java    From Raigad with Apache License 2.0 6 votes vote down vote up
private AstyanaxContext<Keyspace> initWithThriftDriverWithExternalHostsSupplier() {
    logger.info("Boot cluster (BOOT_CLUSTER) is {}, keyspace name (KS_NAME) is {}", BOOT_CLUSTER, KS_NAME);

    return new AstyanaxContext.Builder()
            .forCluster(BOOT_CLUSTER)
            .forKeyspace(KS_NAME)
            .withAstyanaxConfiguration(
                    new AstyanaxConfigurationImpl()
                            .setDiscoveryType(
                                    NodeDiscoveryType.DISCOVERY_SERVICE)
                            .setConnectionPoolType(
                                    ConnectionPoolType.ROUND_ROBIN))
            .withConnectionPoolConfiguration(
                    new ConnectionPoolConfigurationImpl(
                            "MyConnectionPool")
                            .setMaxConnsPerHost(3)
                            .setPort(thriftPortForAstyanax))
            .withHostSupplier(getSupplier())
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());

}
 
Example #6
Source File: TestChunking.java    From staash with Apache License 2.0 6 votes vote down vote up
@Before
public void setup() {
	AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
			.forCluster("Test Cluster")
			.forKeyspace(KS)
			.withAstyanaxConfiguration(
					new AstyanaxConfigurationImpl()
							.setDiscoveryType(NodeDiscoveryType.RING_DESCRIBE))
			.withConnectionPoolConfiguration(
					new ConnectionPoolConfigurationImpl("MyConnectionPool")
							.setPort(9160).setMaxConnsPerHost(1)
							.setSeeds("127.0.0.1:9160"))
			.withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
			.buildKeyspace(ThriftFamilyFactory.getInstance());

	context.start();
	keyspace = context.getClient();
}
 
Example #7
Source File: Migration.java    From blueflood with Apache License 2.0 6 votes vote down vote up
private static AstyanaxContext<Keyspace> connect(String host, int port, String keyspace, int threads, NodeDiscoveryType discovery) {
    AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
                    .forKeyspace(keyspace)
            .withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
                .setDiscoveryType(discovery)
                .setRetryPolicy(new RetryNTimes(10)))
                
            .withConnectionPoolConfiguration(new ConnectionPoolConfigurationImpl(host + ":" + keyspace)
                    .setMaxConns(threads * 2)
                    .setSeeds(host)
                    .setPort(port)
                    .setRetryBackoffStrategy(new FixedRetryBackoffStrategy(1000, 1000)))
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());
    context.start();
    return context;
}
 
Example #8
Source File: DefaultAstyanaxClusterClientProvider.java    From staash with Apache License 2.0 6 votes vote down vote up
@Override
public synchronized Cluster acquireCluster(ClusterKey clusterKey) {
    String clusterName = clusterKey.getClusterName().toLowerCase();
    Preconditions.checkNotNull(clusterName, "Invalid cluster name 'null'");
    
    ClusterContextHolder holder = contextMap.get(clusterName);
    if (holder == null) {
        HostSupplierProvider hostSupplierProvider = hostSupplierProviders.get(clusterKey.getDiscoveryType());
        Preconditions.checkNotNull(hostSupplierProvider, String.format("Unknown host supplier provider '%s' for cluster '%s'", clusterKey.getDiscoveryType(), clusterName));
        
        AstyanaxContext<Cluster> context = new AstyanaxContext.Builder()
            .forCluster(clusterName)
            .withAstyanaxConfiguration(configurationProvider.get(clusterName))
            .withConnectionPoolConfiguration(cpProvider.get(clusterName))
            .withConnectionPoolMonitor(monitorProvider.get(clusterName))
            .withHostSupplier(hostSupplierProvider.getSupplier(clusterName))
            .buildCluster(ThriftFamilyFactory.getInstance());
        
        holder = new ClusterContextHolder(context);
        holder.start();
    }
    holder.addRef();
    
    return holder.getKeyspace();
}
 
Example #9
Source File: CassandraConfiguration.java    From emodb with Apache License 2.0 5 votes vote down vote up
public AstyanaxCluster cluster() {
    checkNotNull(_cluster, "cluster");
    String metricName;
    ConnectionPoolConfiguration poolConfig;

    if (_keyspace == null) {
        // Use the shared pool configuration
        metricName = Objects.firstNonNull(_clusterMetric, _cluster);
        poolConfig = CassandraConfiguration.this;
    } else {
        // Use the configuration specifically for this keyspace
        KeyspaceConfiguration keyspaceConfig = checkNotNull(_keyspaces.get(_keyspace), "keyspaceConfig");
        metricName = Objects.firstNonNull(keyspaceConfig.getKeyspaceMetric(), _keyspace);
        poolConfig = keyspaceConfig;
    }

    AstyanaxContext.Builder builder = newAstyanaxBuilder(_cluster, poolConfig, _metricRegistry)
            .forCluster(_cluster);

    if (!_disableClusterMetrics) {
            builder = builder
                    .withTracerFactory(new InstrumentedTracerFactory(metricName, _metricRegistry))
                    .withConnectionPoolMonitor(new MetricConnectionPoolMonitor(metricName, _metricRegistry));
    }

    AstyanaxContext<Cluster> astyanaxContext = builder.buildCluster(ThriftFamilyFactory.getInstance());

    return new AstyanaxCluster(astyanaxContext, _cluster, _dataCenter);
}
 
Example #10
Source File: CassandraStoreImpl.java    From recipes-rss with Apache License 2.0 5 votes vote down vote up
/**
 * Connect to Cassandra
 */
private static Keyspace getKeyspace() throws Exception{
    if (ks == null) {
        try {
            AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
                .forKeyspace(DynamicPropertyFactory.getInstance().getStringProperty(RSSConstants.CASSANDRA_KEYSPACE, null).get())
                .withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
                    .setDiscoveryType(NodeDiscoveryType.RING_DESCRIBE)
                )
                .withConnectionPoolConfiguration(new ConnectionPoolConfigurationImpl("MyConnectionPool")
                    .setPort(DynamicPropertyFactory.getInstance().getIntProperty(RSSConstants.CASSANDRA_PORT, 0).get())
                    .setMaxConnsPerHost(DynamicPropertyFactory.getInstance().getIntProperty(RSSConstants.CASSANDRA_MAXCONNSPERHOST, 1).get())
                    .setSeeds(DynamicPropertyFactory.getInstance().getStringProperty(RSSConstants.CASSANDRA_HOST, "").get() + ":" +
                              DynamicPropertyFactory.getInstance().getIntProperty(RSSConstants.CASSANDRA_PORT, 0).get()
                    )
                )
                .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
                .buildKeyspace(ThriftFamilyFactory.getInstance());

            context.start();
            ks = context.getEntity();
        } catch (Exception e) {
            logger.error("Exception occurred when initializing Cassandra keyspace: " + e);
            throw e;
        }
    }

    return ks;
}
 
Example #11
Source File: CassandraClusterImpl.java    From usergrid with Apache License 2.0 5 votes vote down vote up
private AstyanaxContext<Cluster> getCluster ( final String clusterName,
                                              final AstyanaxConfiguration astyanaxConfiguration,
                                              final ConnectionPoolConfiguration poolConfig ) {

    return new AstyanaxContext.Builder().forCluster( clusterName )
        .withAstyanaxConfiguration( astyanaxConfiguration )
        .withConnectionPoolConfiguration( poolConfig )
        .withConnectionPoolMonitor( new Slf4jConnectionPoolMonitorImpl())
        .buildCluster( ThriftFamilyFactory.getInstance() );

}
 
Example #12
Source File: AstyanaxIO.java    From blueflood with Apache License 2.0 5 votes vote down vote up
private static AstyanaxContext<Keyspace> createCustomHostContext(AstyanaxConfigurationImpl configuration,
        ConnectionPoolConfigurationImpl connectionPoolConfiguration) {
    return new AstyanaxContext.Builder()
            .forCluster(CassandraModel.CLUSTER)
            .forKeyspace(CassandraModel.KEYSPACE)
            .withAstyanaxConfiguration(configuration)
            .withConnectionPoolConfiguration(connectionPoolConfiguration)
            .withConnectionPoolMonitor(new InstrumentedConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());
}
 
Example #13
Source File: DefaultKeyspaceClientProvider.java    From staash with Apache License 2.0 5 votes vote down vote up
@Override
public Keyspace acquireKeyspace(KeyspaceKey key) {
    String schemaName = key.getSchemaName().toLowerCase();
    
    Preconditions.checkNotNull(schemaName, "Invalid schema name 'null'");
    
    KeyspaceContextHolder holder = contextMap.get(schemaName);
    if (holder == null) {
        Preconditions.checkNotNull(key.getClusterName(),   "Missing cluster name for schema " + schemaName);
        Preconditions.checkNotNull(key.getKeyspaceName(),  "Missing cluster name for schema " + schemaName);
        Preconditions.checkNotNull(key.getDiscoveryType(), "Missing cluster name for schema " + schemaName);
        
        HostSupplierProvider hostSupplierProvider = hostSupplierProviders.get(key.getDiscoveryType());
        Preconditions.checkNotNull(hostSupplierProvider, "Unknown host supplier provider " + key.getDiscoveryType());
        
        AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
            .forCluster(key.getClusterName())
            .forKeyspace(key.getKeyspaceName())
            .withAstyanaxConfiguration(configurationProvider.get(schemaName))
            .withConnectionPoolConfiguration(cpProvider.get(schemaName))
            .withConnectionPoolMonitor(monitorProvider.get(schemaName))
            .withHostSupplier(hostSupplierProvider.getSupplier(key.getClusterName()))
            .buildKeyspace(ThriftFamilyFactory.getInstance());
        
        holder = new KeyspaceContextHolder(context);
        contextMap.put(schemaName, holder);
        holder.start();
    }
    holder.addRef();
    
    return holder.getKeyspace();
}
 
Example #14
Source File: CassandraClusterHolder.java    From staash with Apache License 2.0 5 votes vote down vote up
/**
 * Register a keyspace such that a client is created for it and it is now accessible to 
 * this instance
 * 
 * @param keyspaceName
 * @throws AlreadyExistsException
 */
public synchronized void registerKeyspace(String keyspaceName) throws AlreadyExistsException {
    Preconditions.checkNotNull(keyspaceName);
    
    if (keyspaces.containsKey(keyspaceName)) {
        throw new AlreadyExistsException("keyspace", keyspaceName);
    }
    
    CassandraKeyspaceHolder keyspace = new CassandraKeyspaceHolder(new AstyanaxContext.Builder()
            .forCluster(clusterName)
            .forKeyspace(keyspaceName)
            .withAstyanaxConfiguration(
                    new AstyanaxConfigurationImpl()
                            .setDiscoveryType(NodeDiscoveryType.RING_DESCRIBE)
                            .setConnectionPoolType(ConnectionPoolType.ROUND_ROBIN)
                            .setDiscoveryDelayInSeconds(60000))
            .withConnectionPoolConfiguration(
                    new ConnectionPoolConfigurationImpl(
                            clusterName + "_" + keyspaceName)
                            .setSeeds("localhost:9160"))
            .withConnectionPoolMonitor(new Slf4jConnectionPoolMonitorImpl())
            .buildKeyspace(ThriftFamilyFactory.getInstance()));
    
    try {
        keyspace.initialize();
    }
    finally {
        keyspaces.put(keyspaceName, keyspace);
    }
}
 
Example #15
Source File: PaasPropertiesModule.java    From staash with Apache License 2.0 5 votes vote down vote up
@Provides
@Named("astmetaks")
Keyspace provideKeyspace(@Named("staash.metacluster") String clustername,EurekaAstyanaxHostSupplier hs) {
    String clusterNameOnly = "";
    String[] clusterinfo = clustername.split(":");
    if (clusterinfo != null && clusterinfo.length == 2) {
        clusterNameOnly = clusterinfo[0];
    } else {
        clusterNameOnly = clustername;
    }
    AstyanaxContext<Keyspace> keyspaceContext = new AstyanaxContext.Builder()
    .forCluster(clusterNameOnly)
    .forKeyspace(MetaConstants.META_KEY_SPACE)
    .withAstyanaxConfiguration(
            new AstyanaxConfigurationImpl()
                    .setDiscoveryType(
                            NodeDiscoveryType.RING_DESCRIBE)
                    .setConnectionPoolType(
                            ConnectionPoolType.TOKEN_AWARE)
                    .setDiscoveryDelayInSeconds(60)
                    .setTargetCassandraVersion("1.2")
                    .setCqlVersion("3.0.0"))
                    .withHostSupplier(hs.getSupplier(clustername))
    .withConnectionPoolConfiguration(
            new ConnectionPoolConfigurationImpl(clusterNameOnly
                    + "_" + MetaConstants.META_KEY_SPACE)
                    .setSocketTimeout(11000)
                    .setConnectTimeout(2000)
                    .setMaxConnsPerHost(10).setInitConnsPerHost(3))
    .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
    .buildKeyspace(ThriftFamilyFactory.getInstance());
    keyspaceContext.start();
    Keyspace keyspace;
    keyspace = keyspaceContext.getClient();
    return keyspace;
}
 
Example #16
Source File: TestStaashModule.java    From staash with Apache License 2.0 5 votes vote down vote up
@Provides
    @Named("astmetaks")
    @Singleton
    Keyspace provideKeyspace() {
         AstyanaxContext<Keyspace> keyspaceContext = new AstyanaxContext.Builder()
        .forCluster("test cluster")
        .forKeyspace(MetaConstants.META_KEY_SPACE)
        .withAstyanaxConfiguration(
                new AstyanaxConfigurationImpl()
                        .setDiscoveryType(
                                NodeDiscoveryType.NONE)
                        .setConnectionPoolType(
                                ConnectionPoolType.ROUND_ROBIN)
                        .setTargetCassandraVersion("1.2")
                        .setCqlVersion("3.0.0"))
//                        .withHostSupplier(hs.getSupplier(clustername))
        .withConnectionPoolConfiguration(
                new ConnectionPoolConfigurationImpl("localpool"
                        + "_" + MetaConstants.META_KEY_SPACE)
                        .setSocketTimeout(30000)
                        .setMaxTimeoutWhenExhausted(20000)
                        .setMaxConnsPerHost(3).setInitConnsPerHost(1)
                        .setSeeds("localhost"+":"+"9160"))  //uncomment for localhost
//        .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
        .buildKeyspace(ThriftFamilyFactory.getInstance());
        keyspaceContext.start();
        Keyspace keyspace;
        keyspace = keyspaceContext.getClient();
        return keyspace;
    }
 
Example #17
Source File: CassandraStorage.java    From greycat with Apache License 2.0 5 votes vote down vote up
public CassandraStorage(String keySpace) {
    context = new AstyanaxContext.Builder()
            .forKeyspace(keySpace)
            .withAstyanaxConfiguration(new AstyanaxConfigurationImpl()
                    .setDiscoveryType(NodeDiscoveryType.RING_DESCRIBE)
            )
            .withConnectionPoolConfiguration(new ConnectionPoolConfigurationImpl("MyConnectionPool")
                    .setMaxConnsPerHost(1)
                    .setSeeds("127.0.0.1:9160")
            )
            .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
            .buildKeyspace(ThriftFamilyFactory.getInstance());
}
 
Example #18
Source File: TestPaasPropertiesModule.java    From staash with Apache License 2.0 4 votes vote down vote up
@Provides
    @Named("astmetaks")
    @Singleton
    Keyspace provideKeyspace(@Named("paas.metacluster") String clustername) {
        String clusterNameOnly = "";
        String clusterPortOnly = "";
        String[] clusterinfo = clustername.split(":");
        if (clusterinfo != null && clusterinfo.length == 2) {
            clusterNameOnly = clusterinfo[0];
            clusterPortOnly = clusterinfo[1];
        } else {
            clusterNameOnly = clustername;
            clusterPortOnly = "9160";
        }
//        hs = new EurekaAstyanaxHostSupplier();
        AstyanaxContext<Keyspace> keyspaceContext = new AstyanaxContext.Builder()
        .forCluster(clusterNameOnly)
        .forKeyspace(MetaConstants.META_KEY_SPACE)
        .withAstyanaxConfiguration(
                new AstyanaxConfigurationImpl()
                        .setDiscoveryType(
                                NodeDiscoveryType.RING_DESCRIBE)
                        .setConnectionPoolType(
                                ConnectionPoolType.TOKEN_AWARE)
                        .setDiscoveryDelayInSeconds(60000)
                        .setTargetCassandraVersion("1.1")
                        .setCqlVersion("3.0.0"))
//                        .withHostSupplier(hs.getSupplier(clustername))
        .withConnectionPoolConfiguration(
                new ConnectionPoolConfigurationImpl(clusterNameOnly
                        + "_" + MetaConstants.META_KEY_SPACE)
                        .setSocketTimeout(3000)
                        .setMaxTimeoutWhenExhausted(2000)
                        .setMaxConnsPerHost(3).setInitConnsPerHost(1)
                        .setSeeds(clusterNameOnly+":"+clusterPortOnly))  //uncomment for localhost
        .withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
        .buildKeyspace(ThriftFamilyFactory.getInstance());
        keyspaceContext.start();
        Keyspace keyspace;
        keyspace = keyspaceContext.getClient();
        return keyspace;
    }
 
Example #19
Source File: DefaultKeyspaceClientProvider.java    From staash with Apache License 2.0 4 votes vote down vote up
@Override
public synchronized Keyspace acquireKeyspace(String schemaName) {
    schemaName = schemaName.toLowerCase();
    
    Preconditions.checkNotNull(schemaName, "Invalid schema name 'null'");
    
    KeyspaceContextHolder holder = contextMap.get(schemaName);
    if (holder == null) {
        LOG.info("Creating schema for '{}'", new Object[]{schemaName});
        
        String clusterName   = configuration.getString(String.format(CLUSTER_NAME_FORMAT,   schemaName));
        String keyspaceName  = configuration.getString(String.format(KEYSPACE_NAME__FORMAT, schemaName));
        String discoveryType = configuration.getString(String.format(DISCOVERY_TYPE_FORMAT, schemaName));
        if (clusterName==null || clusterName.equals("")) clusterName   = configuration.getString(String.format(CLUSTER_NAME_FORMAT,   "configuration"));
        if (keyspaceName == null || keyspaceName.equals("")) keyspaceName = schemaName;
        if (discoveryType==null || discoveryType.equals("")) discoveryType = configuration.getString(String.format(DISCOVERY_TYPE_FORMAT, "configuration"));
        Preconditions.checkNotNull(clusterName,   "Missing cluster name for schema " + schemaName + " " + String.format(CLUSTER_NAME_FORMAT,schemaName));
        Preconditions.checkNotNull(keyspaceName,  "Missing cluster name for schema " + schemaName + " " + String.format(KEYSPACE_NAME__FORMAT,schemaName));
        Preconditions.checkNotNull(discoveryType, "Missing cluster name for schema " + schemaName + " " + String.format(DISCOVERY_TYPE_FORMAT,schemaName));
        
        HostSupplierProvider hostSupplierProvider = hostSupplierProviders.get(discoveryType);
        Preconditions.checkNotNull(hostSupplierProvider, 
                String.format("Unknown host supplier provider '%s' for schema '%s'", discoveryType, schemaName));
        
        AstyanaxContext<Keyspace> context = new AstyanaxContext.Builder()
            .forCluster(clusterName)
            .forKeyspace(keyspaceName)
            .withAstyanaxConfiguration(configurationProvider.get(schemaName))
            .withConnectionPoolConfiguration(cpProvider.get(schemaName))
            .withConnectionPoolMonitor(monitorProvider.get(schemaName))
            .withHostSupplier(hostSupplierProvider.getSupplier(clusterName))
            .buildKeyspace(ThriftFamilyFactory.getInstance());
        context.start();
        try {
            context.getClient().createKeyspace(defaultKsOptions);
        } catch (ConnectionException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        holder = new KeyspaceContextHolder(context);
        contextMap.put(schemaName, holder);
        holder.start();
    }
    holder.addRef();
    
    return holder.getKeyspace();
}
 
Example #20
Source File: AstyanaxCassandraConnection.java    From staash with Apache License 2.0 4 votes vote down vote up
private Keyspace createAstyanaxKeyspace(String clustername, String db,
		EurekaAstyanaxHostSupplier supplier) {
	String clusterNameOnly = "localhost";
	String clusterPortOnly = "9160";
	String[] clusterinfo = clustername.split(":");
	if (clusterinfo != null && clusterinfo.length == 2) {
		clusterNameOnly = clusterinfo[0];
	} else {
		clusterNameOnly = clustername;
	}
	AstyanaxContext<Keyspace> keyspaceContext;
	if (supplier!=null) {
	keyspaceContext = new AstyanaxContext.Builder()
			.forCluster("Casss_Paas")
			.forKeyspace(db)
			.withAstyanaxConfiguration(
					new AstyanaxConfigurationImpl()
							.setDiscoveryType(
									NodeDiscoveryType.DISCOVERY_SERVICE)
							.setConnectionPoolType(
									ConnectionPoolType.TOKEN_AWARE)
							.setDiscoveryDelayInSeconds(60)
							.setTargetCassandraVersion("1.2")
							.setCqlVersion("3.0.0"))
			.withHostSupplier(supplier.getSupplier(clustername))
			.withConnectionPoolConfiguration(
					new ConnectionPoolConfigurationImpl(clusterNameOnly
							+ "_" + db)
							.setSocketTimeout(10000)
							.setPort(7102)
							.setMaxConnsPerHost(10).setInitConnsPerHost(3)
							.setSeeds(null))
			.withConnectionPoolMonitor(new CountingConnectionPoolMonitor())
			.buildKeyspace(ThriftFamilyFactory.getInstance());
	} else {
		keyspaceContext = new AstyanaxContext.Builder()
        .forCluster(clusterNameOnly)
        .forKeyspace(db)
        .withAstyanaxConfiguration(
                new AstyanaxConfigurationImpl()
                        .setDiscoveryType(
                                NodeDiscoveryType.RING_DESCRIBE)
                        .setConnectionPoolType(
                                ConnectionPoolType.TOKEN_AWARE)
                        .setDiscoveryDelayInSeconds(60)
                        .setTargetCassandraVersion("1.2")
                        .setCqlVersion("3.0.0"))
                        //.withHostSupplier(hs.getSupplier(clustername))
        .withConnectionPoolConfiguration(
                new ConnectionPoolConfigurationImpl(clusterNameOnly
                        + "_" + db)
                        .setSocketTimeout(11000)
                        .setConnectTimeout(2000)
                        .setMaxConnsPerHost(10).setInitConnsPerHost(3)	 
                      .setSeeds(clusterNameOnly+":"+clusterPortOnly))
         .buildKeyspace(ThriftFamilyFactory.getInstance());
	}
	keyspaceContext.start();
	Keyspace keyspace;
	keyspace = keyspaceContext.getClient();
	return keyspace;
}
 
Example #21
Source File: AstyanaxStoreManager.java    From titan1withtp3.1 with Apache License 2.0 4 votes vote down vote up
private static AstyanaxContext<Cluster> createCluster(AstyanaxContext.Builder cb) {
    AstyanaxContext<Cluster> clusterCtx = cb.buildCluster(ThriftFamilyFactory.getInstance());
    clusterCtx.start();

    return clusterCtx;
}
 
Example #22
Source File: AstyanaxStoreManager.java    From titan1withtp3.1 with Apache License 2.0 3 votes vote down vote up
public AstyanaxStoreManager(Configuration config) throws BackendException {
    super(config);

    this.clusterName = config.get(CLUSTER_NAME);

    retryDelaySlice = config.get(RETRY_DELAY_SLICE);
    retryMaxDelaySlice = config.get(RETRY_MAX_DELAY_SLICE);
    retrySuspendWindow = config.get(RETRY_SUSPEND_WINDOW);
    retryBackoffStrategy = getRetryBackoffStrategy(config.get(RETRY_BACKOFF_STRATEGY));
    retryPolicy = getRetryPolicy(config.get(RETRY_POLICY));

    localDatacenter = config.has(LOCAL_DATACENTER) ?
            config.get(LOCAL_DATACENTER) : "";

    final int maxConnsPerHost = config.get(MAX_CONNECTIONS_PER_HOST);

    final int maxClusterConnsPerHost = config.get(MAX_CLUSTER_CONNECTIONS_PER_HOST);

    this.clusterContext = createCluster(getContextBuilder(config, maxClusterConnsPerHost, "Cluster"));

    ensureKeyspaceExists(clusterContext.getClient());

    this.keyspaceContext = getContextBuilder(config, maxConnsPerHost, "Keyspace").buildKeyspace(ThriftFamilyFactory.getInstance());
    this.keyspaceContext.start();

    openStores = new HashMap<String, AstyanaxKeyColumnValueStore>(8);
}