func main() {
flag.Parse()
servenv.Init()
var schema *planbuilder.Schema
log.Info(*cell, *schemaFile)
if *schemaFile != "" {
var err error
if schema, err = planbuilder.LoadSchemaJSON(*schemaFile); err != nil {
log.Fatal(err)
}
}
ts := topo.GetServer()
defer topo.CloseServers()
resilientSrvTopoServer = vtgate.NewResilientSrvTopoServer(ts, "ResilientSrvTopoServer")
// For the initial phase vtgate is exposing
// topoReader api. This will be subsumed by
// vtgate once vtgate's client functions become active.
topoReader = NewTopoReader(resilientSrvTopoServer)
servenv.Register("toporeader", topoReader)
vtgate.Init(resilientSrvTopoServer, schema, *cell, *retryDelay, *retryCount, *timeout, *maxInFlight)
servenv.RunDefault()
}
func TestSelectEqual(t *testing.T) {
schema, err := planbuilder.LoadSchemaJSON(locateFile("router_test.json"))
if err != nil {
t.Fatal(err)
}
s := createSandbox("TestRouter")
sbc1 := &sandboxConn{}
sbc2 := &sandboxConn{}
s.MapTestConn("-20", sbc1)
s.MapTestConn("40-60", sbc2)
serv := new(sandboxTopo)
scatterConn := NewScatterConn(serv, "", "aa", 1*time.Second, 10, 1*time.Millisecond)
router := NewRouter(serv, "aa", schema, "", scatterConn)
q := proto.Query{
Sql: "select * from user where id = 1",
TabletType: topo.TYPE_MASTER,
}
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
if sbc1.ExecCount != 1 {
t.Errorf("sbc1.ExecCount: %v, want 1\n", sbc1.ExecCount)
}
wantBind := map[string]interface{}{}
if !reflect.DeepEqual(sbc1.BindVars, wantBind) {
t.Errorf("sbc1.BindVars = %#v, want %#v", sbc1.BindVars, wantBind)
}
wantQuery := "select * from user where id = 1"
if sbc1.Query != wantQuery {
t.Errorf("sbc1.Query: %q, want %q\n", sbc1.Query, wantQuery)
}
if sbc2.ExecCount != 0 {
t.Errorf("sbc2.ExecCount: %v, want 0\n", sbc2.ExecCount)
}
q.Sql = "select * from user where id = 3"
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
if sbc1.ExecCount != 1 {
t.Errorf("sbc1.ExecCount: %v, want 1\n", sbc1.ExecCount)
}
if sbc2.ExecCount != 1 {
t.Errorf("sbc2.ExecCount: %v, want 1\n", sbc2.ExecCount)
}
wantBind = map[string]interface{}{}
if !reflect.DeepEqual(sbc2.BindVars, wantBind) {
t.Errorf("sbc2.BindVars = %#v, want %#v", sbc2.BindVars, wantBind)
}
wantQuery = "select * from user where id = 3"
if sbc2.Query != wantQuery {
t.Errorf("sbc2.Query: %q, want %q\n", sbc2.Query, wantQuery)
}
}
func TestInsertLookupUnowned(t *testing.T) {
schema, err := planbuilder.LoadSchemaJSON(locateFile("router_test.json"))
if err != nil {
t.Fatal(err)
}
s := createSandbox("TestRouter")
sbc := &sandboxConn{}
s.MapTestConn("-20", sbc)
l := createSandbox("TestUnsharded")
sbclookup := &sandboxConn{}
l.MapTestConn("0", sbclookup)
serv := new(sandboxTopo)
scatterConn := NewScatterConn(serv, "", "aa", 1*time.Second, 10, 1*time.Millisecond)
router := NewRouter(serv, "aa", schema, "", scatterConn)
q := proto.Query{
Sql: "insert into music_extra(user_id, music_id) values (2, 3)",
TabletType: topo.TYPE_MASTER,
}
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantBind := map[string]interface{}{
"music_id": int64(3),
"user_id": uint64(2),
}
if !reflect.DeepEqual(sbclookup.BindVars, wantBind) {
t.Errorf("sbclookup.BindVars = %#v, want %#v", sbclookup.BindVars, wantBind)
}
wantQuery := "select music_id from music_user_map where music_id = :music_id and user_id = :user_id"
if sbclookup.Query != wantQuery {
t.Errorf("sbclookup.Query: %q, want %q\n", sbclookup.Query, wantQuery)
}
wantBind = map[string]interface{}{
"keyspace_id": "\x06\xe7\xea\"Βp\x8f",
"_user_id": int64(2),
"_music_id": int64(3),
}
if !reflect.DeepEqual(sbc.BindVars, wantBind) {
t.Errorf("sbc.BindVars = %#v, want %#v", sbc.BindVars, wantBind)
}
wantQuery = "insert into music_extra(user_id, music_id) values (:_user_id, :_music_id) /* _routing keyspace_id:06e7ea22ce92708f */"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
}
func TestInsertGenerator(t *testing.T) {
schema, err := planbuilder.LoadSchemaJSON(locateFile("router_test.json"))
if err != nil {
t.Fatal(err)
}
s := createSandbox("TestRouter")
sbc := &sandboxConn{}
s.MapTestConn("80-a0", sbc)
l := createSandbox("TestUnsharded")
sbclookup := &sandboxConn{}
l.MapTestConn("0", sbclookup)
serv := new(sandboxTopo)
scatterConn := NewScatterConn(serv, "", "aa", 1*time.Second, 10, 1*time.Millisecond)
router := NewRouter(serv, "aa", schema, "", scatterConn)
q := proto.Query{
Sql: "insert into user(v) values (2)",
TabletType: topo.TYPE_MASTER,
}
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantBind := map[string]interface{}{
"id": nil,
}
if !reflect.DeepEqual(sbclookup.BindVars, wantBind) {
t.Errorf("sbclookup.BindVars = %#v, want %#v", sbclookup.BindVars, wantBind)
}
wantQuery := "insert into user_idx(id) values(:id)"
if sbclookup.Query != wantQuery {
t.Errorf("sbclookup.Query: %q, want %q\n", sbclookup.Query, wantQuery)
}
wantBind = map[string]interface{}{
"keyspace_id": "\x8c\xa6M\xe9\xc1\xb1#\xa7",
"_id": uint64(0),
}
if !reflect.DeepEqual(sbc.BindVars, wantBind) {
t.Errorf("sbc.BindVars = %#v, want %#v", sbc.BindVars, wantBind)
}
wantQuery = "insert into user(v, id) values (2, :_id) /* _routing keyspace_id:8ca64de9c1b123a7 */"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
}
func TestUnsharded(t *testing.T) {
schema, err := planbuilder.LoadSchemaJSON(locateFile("router_test.json"))
if err != nil {
t.Fatal(err)
}
createSandbox("TestRouter")
s := createSandbox(TEST_UNSHARDED)
sbc := &sandboxConn{}
s.MapTestConn("0", sbc)
serv := new(sandboxTopo)
scatterConn := NewScatterConn(serv, "", "aa", 1*time.Second, 10, 1*time.Millisecond)
router := NewRouter(serv, "aa", schema, "", scatterConn)
q := proto.Query{
Sql: "select * from music_user_map where id = 1",
TabletType: topo.TYPE_MASTER,
}
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
if sbc.ExecCount != 1 {
t.Errorf("sbc.ExecCount: %v, want 1\n", sbc.ExecCount)
}
wantBind := map[string]interface{}{}
if !reflect.DeepEqual(sbc.BindVars, wantBind) {
t.Errorf("sbc.BindVars = %#v, want %#v", sbc.BindVars, wantBind)
}
wantQuery := "select * from music_user_map where id = 1"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
q.Sql = "update music_user_map set id = 1"
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantQuery = "update music_user_map set id = 1"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
q.Sql = "delete from music_user_map"
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantQuery = "delete from music_user_map"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
q.Sql = "insert into music_user_map values(1)"
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantQuery = "insert into music_user_map values(1)"
if sbc.Query != wantQuery {
t.Errorf("sbc.Query: %q, want %q\n", sbc.Query, wantQuery)
}
}
func TestInsertSharded(t *testing.T) {
schema, err := planbuilder.LoadSchemaJSON(locateFile("router_test.json"))
if err != nil {
t.Fatal(err)
}
s := createSandbox("TestRouter")
sbc1 := &sandboxConn{}
sbc2 := &sandboxConn{}
s.MapTestConn("-20", sbc1)
s.MapTestConn("40-60", sbc2)
l := createSandbox("TestUnsharded")
sbclookup := &sandboxConn{}
l.MapTestConn("0", sbclookup)
serv := new(sandboxTopo)
scatterConn := NewScatterConn(serv, "", "aa", 1*time.Second, 10, 1*time.Millisecond)
router := NewRouter(serv, "aa", schema, "", scatterConn)
q := proto.Query{
Sql: "insert into user(id, v) values (1, 2)",
TabletType: topo.TYPE_MASTER,
}
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
wantBind := map[string]interface{}{
"id": int64(1),
}
if !reflect.DeepEqual(sbclookup.BindVars, wantBind) {
t.Errorf("sbclookup.BindVars = %#v, want %#v", sbclookup.BindVars, wantBind)
}
wantQuery := "insert into user_idx(id) values(:id)"
if sbclookup.Query != wantQuery {
t.Errorf("sbclookup.Query: %q, want %q\n", sbclookup.Query, wantQuery)
}
wantBind = map[string]interface{}{
"keyspace_id": "\x16k@\xb4J\xbaK\xd6",
"_id": int64(1),
}
if !reflect.DeepEqual(sbc1.BindVars, wantBind) {
t.Errorf("sbc1.BindVars = %#v, want %#v", sbc1.BindVars, wantBind)
}
wantQuery = "insert into user(id, v) values (:_id, 2) /* _routing keyspace_id:166b40b44aba4bd6 */"
if sbc1.Query != wantQuery {
t.Errorf("sbc1.Query: %q, want %q\n", sbc1.Query, wantQuery)
}
if sbc2.ExecCount != 0 {
t.Errorf("sbc2.ExecCount: %v, want 0\n", sbc2.ExecCount)
}
q.Sql = "insert into user(id, v) values (3, 2)"
_, err = router.Execute(&context.DummyContext{}, &q)
if err != nil {
t.Error(err)
}
if sbc1.ExecCount != 1 {
t.Errorf("sbc1.ExecCount: %v, want 1\n", sbc1.ExecCount)
}
wantBind = map[string]interface{}{
"id": int64(3),
}
if !reflect.DeepEqual(sbclookup.BindVars, wantBind) {
t.Errorf("sbclookup.BindVars = %#v, want %#v", sbclookup.BindVars, wantBind)
}
wantQuery = "insert into user_idx(id) values(:id)"
if sbclookup.Query != wantQuery {
t.Errorf("sbclookup.Query: %q, want %q\n", sbclookup.Query, wantQuery)
}
wantBind = map[string]interface{}{
"keyspace_id": "N\xb1\x90ɢ\xfa\x16\x9c",
"_id": int64(3),
}
if !reflect.DeepEqual(sbc2.BindVars, wantBind) {
t.Errorf("sbc2.BindVars = %#v, want %#v", sbc2.BindVars, wantBind)
}
wantQuery = "insert into user(id, v) values (:_id, 2) /* _routing keyspace_id:4eb190c9a2fa169c */"
if sbc2.Query != wantQuery {
t.Errorf("sbc2.Query: %q, want %q\n", sbc2.Query, wantQuery)
}
}