func initVitess(server, keyspace, shard string) {
fmt.Println("Connecting to vitess on ", server, "...")
timeout := 10 * time.Second
var err error
// Connect to vtgate
vitessdb, err = vitessdriver.OpenShard(
server, keyspace, shard, "master", timeout)
if err != nil {
log.Fatal("vitess client error: ", err)
}
fmt.Println("Done\n\n")
//defer vitessdb.Close()
}
func main() {
flag.Parse()
keyspace := "test_keyspace"
shard := "0"
timeout := 10 * time.Second
// Connect to vtgate.
db, err := vitessdriver.OpenShard(*server, keyspace, shard, "master", timeout)
if err != nil {
fmt.Printf("client error: %v\n", err)
os.Exit(1)
}
defer db.Close()
// Insert something.
fmt.Println("Inserting into master...")
tx, err := db.Begin()
if err != nil {
fmt.Printf("begin failed: %v\n", err)
os.Exit(1)
}
if _, err := tx.Exec("INSERT INTO test_table (msg) VALUES (?)", "V is for speed"); err != nil {
fmt.Printf("exec failed: %v\n", err)
os.Exit(1)
}
if err := tx.Commit(); err != nil {
fmt.Printf("commit failed: %v\n", err)
os.Exit(1)
}
// Read it back from the master.
fmt.Println("Reading from master...")
rows, err := db.Query("SELECT id, msg FROM test_table")
if err != nil {
fmt.Printf("query failed: %v\n", err)
os.Exit(1)
}
for rows.Next() {
var id int
var msg string
if err := rows.Scan(&id, &msg); err != nil {
fmt.Printf("scan failed: %v\n", err)
os.Exit(1)
}
fmt.Printf("(%v, %q)\n", id, msg)
}
if err := rows.Err(); err != nil {
fmt.Printf("row iteration failed: %v\n", err)
os.Exit(1)
}
// Read from a replica.
// Note that this may be behind master due to replication lag.
fmt.Println("Reading from replica...")
dbr, err := vitessdriver.OpenShard(*server, keyspace, shard, "replica", timeout)
if err != nil {
fmt.Printf("client error: %v\n", err)
os.Exit(1)
}
defer dbr.Close()
rows, err = dbr.Query("SELECT id, msg FROM test_table")
if err != nil {
fmt.Printf("query failed: %v\n", err)
os.Exit(1)
}
for rows.Next() {
var id int
var msg string
if err := rows.Scan(&id, &msg); err != nil {
fmt.Printf("scan failed: %v\n", err)
os.Exit(1)
}
fmt.Printf("(%v, %q)\n", id, msg)
}
if err := rows.Err(); err != nil {
fmt.Printf("row iteration failed: %v\n", err)
os.Exit(1)
}
}