Esempio n. 1
0
func insert(session *gocql.Session) {
	for value := range ops_insert {
		query := session.Query(`INSERT INTO stress_table (key1,key2,key3,col1,col2,col3) VALUES (?,?,?,?,?,?)`, value[0], value[1], value[2], value[3], value[4], value[5])
		if err := query.Exec(); err != nil {
			log.Fatal(err)
		}
	}
	done <- true

}
Esempio n. 2
0
func (writer *queueWriter) writeBatch(db *gocql.Session, requests []*batchRequest) {
	dbBatch := gocql.NewBatch(gocql.UnloggedBatch)
	i := int64(0)

	for _, request := range requests {
		for _, item := range request.items {
			itemID := writer.nextIndex + i
			dbBatch.Query(`INSERT INTO queue_items (queue_id, item_id, item_value) VALUES (?, ?, ?)`, writer.id, itemID, item)
			i++
		}
	}

	err := db.ExecuteBatch(dbBatch)
	writer.respond(requests, err)
}
Esempio n. 3
0
func createks(session *gocql.Session, ks string, table string) {

	if err := session.Query(fmt.Sprintf(
		`CREATE KEYSPACE IF NOT EXISTS %s WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1 };`, ks)).Exec(); err != nil {
		log.Fatalf("Could not create keyspace %s : %s", ks, err)
	} else {
		log.Printf("Success::Keyspace %s was created or already exists", ks)
	}

	session.Query(fmt.Sprintf("use %s;", ks)).Exec()

	if err := session.Query(fmt.Sprintf(
		`CREATE TABLE IF NOT EXISTS %s ( key1 int, key2 int, key3 int, col1 int, col2 int, col3 int, PRIMARY KEY (key1,key2,key3));`, table)).Exec(); err != nil {
		log.Fatalf("Could not create table %s : %s", table, err)
	} else {
		log.Printf("Success::Table %s was created or already exists", table)
	}

}