func Connect(info map[string]interface{}) (conn *Connection, err error) {
defer handleError(&err)
host := mapCString(info, "host")
defer cfree(host)
port := mapint(info, "port")
uname := mapCString(info, "uname")
defer cfree(uname)
pass := mapCString(info, "pass")
defer cfree(pass)
dbname := mapCString(info, "dbname")
defer cfree(dbname)
unix_socket := mapCString(info, "unix_socket")
defer cfree(unix_socket)
charset := mapCString(info, "charset")
defer cfree(charset)
conn = &Connection{}
conn.handle = C.mysql_init(nil)
if C.mysql_real_connect(conn.handle, host, uname, pass, dbname, C.uint(port), unix_socket, 0) == nil {
defer conn.Close()
return nil, conn.lastError(nil)
}
if C.mysql_set_character_set(conn.handle, charset) != 0 {
defer conn.Close()
return nil, conn.lastError(nil)
}
return conn, nil
}
//建立一个到数据库服务器的连接,
//成功返回0, 否则返回errnum
func (mysql *MySQL) Connect(host, usr, pwd, db string, port uint) (errnum int) {
mysql.my = C.mysql_init(nil)
if mysql.my == nil {
mysql.errno = 1
return 1
}
h := C.CString(host)
u := C.CString(usr)
p := C.CString(pwd)
d := C.CString(db)
defer C.free(unsafe.Pointer(h))
defer C.free(unsafe.Pointer(u))
defer C.free(unsafe.Pointer(p))
defer C.free(unsafe.Pointer(d))
mysql.my = C.mysql_real_connect(mysql.my, h, u, p, d, (C.unsigned)(port), nil, 0)
if mysql.my == nil {
mysql.errno = 2
return 2
}
mysql.errno = 0
return 0
}
// The URL passed into this function should be of the form:
//
// mysql://user:pass@host:port/database_name
// mysql://user:pass@socket./database_name?socket=/tmp/mysql.sock
//
// Every option except for the database is optional, and missing items will be
// taken from the system-wide configuration.
//
// The scheme may be omitted:
//
// //user:pass@host:port/database_name
//
func open(uri string) (conn db.Connection, err os.Error) {
var host, uname, passwd, dbname, socket *C.char;
var port C.uint;
url, urlError := http.ParseURL(uri);
if urlError != nil {
err = MysqlError(fmt.Sprintf("Couldn't parse URL: %s", urlError));
return;
}
if len(url.Scheme) > 0 && url.Scheme != "mysql" {
err = MysqlError(fmt.Sprintf("Invalid scheme: %s", url.Scheme));
return;
}
if len(url.Host) > 0 {
if url.Host == "socket." {
sock := strings.Split(url.RawQuery, "=", 2);
if len(sock) != 2 || sock[0] != "socket" {
err = MysqlError(
fmt.Sprintf("Invalid socket specified: %s", url.RawQuery));
return;
}
sock = strings.Split(sock[1], "&", 2);
socket = C.CString(sock[0]);
} else {
hostport := strings.Split(url.Host, ":", 2);
if len(hostport) == 2 && len(hostport[1]) > 0 {
p, _ := strconv.Atoi(hostport[1]);
port = C.uint(p);
}
if len(hostport[0]) > 0 {
host = C.CString(hostport[0])
}
}
}
if len(url.Userinfo) > 0 {
userpass := strings.Split(url.Userinfo, ":", 2);
if len(userpass) == 2 && len(userpass[1]) > 0 {
passwd = C.CString(userpass[1])
}
if len(userpass[0]) > 0 {
uname = C.CString(userpass[0])
}
}
if len(url.Path) > 0 {
path := strings.Split(url.Path, "/", 2);
if len(path) == 2 && len(path[0]) == 0 && len(path[1]) > 0 {
dbname = C.CString(path[1])
}
}
c := Connection{};
c.lock = new(sync.Mutex);
c.handle = C.mysql_init(nil);
if c.handle == nil {
err = MysqlError("Couldn't init handle (likely out of memory?)");
goto cleanup;
}
rc := C.mysql_real_connect(
c.handle,
host,
uname,
passwd,
dbname,
port,
socket,
0); // client flags
// If an error was set, or if the handle returned is not the same as the
// one we allocated, there was a problem.
err = c.lastError();
if err != nil || rc != c.handle {
C.mysql_close(c.handle)
} else {
// Everything's ok, set the returned the connection
conn = c
}
cleanup:
if host != nil {
C.free(unsafe.Pointer(host))
}
if uname != nil {
C.free(unsafe.Pointer(uname))
}
if passwd != nil {
C.free(unsafe.Pointer(passwd))
}
if dbname != nil {
C.free(unsafe.Pointer(dbname))
}
if socket != nil {
C.free(unsafe.Pointer(socket))
}
return;
}