Sample usage of mysql parser - could be useful to create ERD diagrams from database schema

mysql-parser.go

package main

import (
	"fmt"
	"log"

	"github.com/pingcap/parser"
	"github.com/pingcap/parser/ast"
	_ "github.com/pingcap/tidb/types/parser_driver"
)

func main() {
	p := parser.New()

	stmtNodes, _, err := p.Parse(`
		create table if not exists users (
			id int,
			name varchar(255),
			role_id int,

			PRIMARY KEY (id),
			FOREIGN KEY (role_id) REFERENCES roles(id)
		)
	`, "", "")
	if err != nil {
		log.Fatal(err)
	}

	node := stmtNodes[0]
	fmt.Printf("%#v\n\n", node)
	stmt := node.(*ast.CreateTableStmt)
	fmt.Printf("%#v\n\n", stmt.Table)
	fmt.Println(stmt.Table)
	for _, col := range stmt.Cols {
		fmt.Println(col.Name, col.Tp, col.Options)
	}
}