mirrors/Nim
1
0
Fork 0
mirror of https://github.com/nim-lang/Nim.git synced 2026-04-20 14:25:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add character set options when opening DB connection

Browse Source
This commit is contained in:
KeMeGe
2015-03-14 15:21:38 +08:00
parent 2f4472963f
commit ca8102b96d
3 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/impure/db_mysql.nim
View File

@@ -212,8 +212,8 @@ proc close*(db: TDbConn) {.tags: [FDb].} =
## closes the database connection.
if db != nil: mysql.close(db)
proc open*(connection, user, password, database: string): TDbConn {.
tags: [FDb].} =
proc open*(connection, user, password, database: string,
charset: string = "utf8"): TDbConn {.tags: [FDb].} =
## opens a database connection. Raises `EDb` if the connection could not
## be established.
result = mysql.init(nil)
@@ -229,3 +229,7 @@ proc open*(connection, user, password, database: string): TDbConn {.
var errmsg = $mysql.error(result)
db_mysql.close(result)
dbError(errmsg)
if mysql.set_character_set(result, charset) == 0:
var errmsg = $mysql.error(result)
db_mysql.close(result)
dbError(errmsg)

5
lib/impure/db_postgres.nim
View File

@@ -239,8 +239,8 @@ proc close*(db: TDbConn) {.tags: [FDb].} =
## closes the database connection.
if db != nil: pqfinish(db)
proc open*(connection, user, password, database: string): TDbConn {.
tags: [FDb].} =
proc open*(connection, user, password, database: string,
charset: string = "UTF-8"): TDbConn {.tags: [FDb].} =
## opens a database connection. Raises `EDb` if the connection could not
## be established.
##
@@ -260,3 +260,4 @@ proc open*(connection, user, password, database: string): TDbConn {.
## the nim db api.
result = pqsetdbLogin(nil, nil, nil, nil, database, user, password)
if pqStatus(result) != CONNECTION_OK: dbError(result) # result = nil
if pqsetClientEncoding(result, charset) != 0: dbError(result)

5
lib/impure/db_sqlite.nim
View File

@@ -183,13 +183,14 @@ proc close*(db: TDbConn) {.tags: [FDb].} =
## closes the database connection.
if sqlite3.close(db) != SQLITE_OK: dbError(db)
proc open*(connection, user, password, database: string): TDbConn {.
tags: [FDb].} =
proc open*(connection, user, password, database: string,
charset: string = "UTF-8"): TDbConn {.tags: [FDb].} =
## opens a database connection. Raises `EDb` if the connection could not
## be established. Only the ``connection`` parameter is used for ``sqlite``.
var db: TDbConn
if sqlite3.open(connection, db) == SQLITE_OK:
result = db
exec(result, sql"PRAGMA encoding = ?", [charset])
else:
dbError(db)