forked from alexbrainman/odbc
/
env.go
52 lines (47 loc) · 1.33 KB
/
env.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package odbc
import (
"github.com/mpcjanssen/odbc/api"
"unsafe"
)
type DSN struct {
Name string
Description string
}
func DataSources() ([]DSN, error) {
//Allocate environment handle
var out api.SQLHANDLE
in := api.SQLHANDLE(api.SQL_NULL_HANDLE)
ret := api.SQLAllocHandle(api.SQL_HANDLE_ENV, in, &out)
if IsError(ret) {
return nil, NewError("SQLAllocHandle", api.SQLHENV(in))
}
h := api.SQLHENV(out)
err := drv.Stats.updateHandleCount(api.SQL_HANDLE_ENV, 1)
if err != nil {
return nil, err
}
defer releaseHandle(h)
// will use ODBC v3
ret = api.SQLSetEnvUIntPtrAttr(h, api.SQL_ATTR_ODBC_VERSION, api.SQL_OV_ODBC3, 0)
if IsError(ret) {
return nil, NewError("SQLSetEnvUIntPtrAttr", drv.h)
}
// Read DSNs
var dsnName [api.SQL_MAX_DSN_LENGTH]byte
var dsnNameLength api.SQLSMALLINT
var dsnDescription [200]byte
var dsnDescriptionLength api.SQLSMALLINT
DSNs := make([]DSN, 0)
for {
ret = api.SQLDataSources(h, api.SQL_FETCH_NEXT,
(*api.SQLSCHAR)(unsafe.Pointer(&dsnName[0])),
api.SQLSMALLINT(len(dsnName)), &dsnNameLength,
(*api.SQLSCHAR)(unsafe.Pointer(&dsnDescription[0])),
api.SQLSMALLINT(len(dsnDescription)), &dsnDescriptionLength)
if ret != api.SQL_SUCCESS {
break
}
DSNs = append(DSNs, DSN{string(dsnName[:dsnNameLength]), string(dsnDescription[:dsnDescriptionLength])})
}
return DSNs, nil
}