schema: quote schema (go-mysql-org#99)

Author: siddontang

schema/schema.go

@@ -162,7 +162,7 @@ func NewTable(conn mysql.Executer, schema string, name string) (*Table, error) {
 }
 
 func (ta *Table) fetchColumns(conn mysql.Executer) error {
-r, err := conn.Execute(fmt.Sprintf("describe %s.%s", ta.Schema, ta.Name))
+r, err := conn.Execute(fmt.Sprintf("describe `%s`.`%s`", ta.Schema, ta.Name))
 if err != nil {
 return errors.Trace(err)
 }
@@ -179,7 +179,7 @@ func (ta *Table) fetchColumns(conn mysql.Executer) error {
 }
 
 func (ta *Table) fetchIndexes(conn mysql.Executer) error {
-r, err := conn.Execute(fmt.Sprintf("show index from %s.%s", ta.Schema, ta.Name))
+r, err := conn.Execute(fmt.Sprintf("show index from `%s`.`%s`", ta.Schema, ta.Name))
 if err != nil {
 return errors.Trace(err)
 }

schema/schema_test.go

@@ -70,3 +70,15 @@ func (s *schemaTestSuite) TestSchema(c *C) {
 c.Assert(ta.Columns[5].SetValues, DeepEquals, []string{"a", "b", "c"})
 c.Assert(ta.Columns[7].Type, Equals, TYPE_FLOAT)
 }
+
+func (s *schemaTestSuite) TestQuoteSchema(c *C) {
+str := "CREATE TABLE IF NOT EXISTS `a-b_test` (`a.b` INT) ENGINE = INNODB"
+
+_, err := s.conn.Execute(str)
+c.Assert(err, IsNil)
+
+ta, err := NewTable(s.conn, "test", "a-b_test")
+c.Assert(err, IsNil)
+
+c.Assert(ta.Columns[0].Name, Equals, "a.b")
+}