diff options
author | Jesse Hallam <jesse.hallam@gmail.com> | 2018-04-16 21:23:20 -0400 |
---|---|---|
committer | Christopher Speller <crspeller@gmail.com> | 2018-04-16 18:23:20 -0700 |
commit | 3a4c7603b3ce35ab247b4dddaf7a2506375ba2c9 (patch) | |
tree | b60b9b7dd321ba741946f4bd57556895c211c3ed /vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go | |
parent | ba6f9075c1fcc3fbfa5ba94e97df6a4dd9e70b50 (diff) | |
download | chat-3a4c7603b3ce35ab247b4dddaf7a2506375ba2c9.tar.gz chat-3a4c7603b3ce35ab247b4dddaf7a2506375ba2c9.tar.bz2 chat-3a4c7603b3ce35ab247b4dddaf7a2506375ba2c9.zip |
MM-10020: avoid duplicating unique indexes (#8587)
* unit test TestGet(Search)Replica
This adds partial, testing-focused support for SQLite, as well as removing
some translated log messages that required initializing i18n just for
testing.
* avoid returning master twice when no replicas are configured
* remove duplicate indexes
* unit test GetAllConns
Diffstat (limited to 'vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go')
-rw-r--r-- | vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go b/vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go new file mode 100644 index 000000000..bb7e25f53 --- /dev/null +++ b/vendor/github.com/mattn/go-sqlite3/sqlite3_load_extension.go @@ -0,0 +1,69 @@ +// Copyright (C) 2014 Yasuhiro Matsumoto <mattn.jp@gmail.com>. +// +// Use of this source code is governed by an MIT-style +// license that can be found in the LICENSE file. +// +build !sqlite_omit_load_extension + +package sqlite3 + +/* +#ifndef USE_LIBSQLITE3 +#include <sqlite3-binding.h> +#else +#include <sqlite3.h> +#endif +#include <stdlib.h> +*/ +import "C" +import ( + "errors" + "unsafe" +) + +func (c *SQLiteConn) loadExtensions(extensions []string) error { + rv := C.sqlite3_enable_load_extension(c.db, 1) + if rv != C.SQLITE_OK { + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + + for _, extension := range extensions { + cext := C.CString(extension) + defer C.free(unsafe.Pointer(cext)) + rv = C.sqlite3_load_extension(c.db, cext, nil, nil) + if rv != C.SQLITE_OK { + C.sqlite3_enable_load_extension(c.db, 0) + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + } + + rv = C.sqlite3_enable_load_extension(c.db, 0) + if rv != C.SQLITE_OK { + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + return nil +} + +// LoadExtension load the sqlite3 extension. +func (c *SQLiteConn) LoadExtension(lib string, entry string) error { + rv := C.sqlite3_enable_load_extension(c.db, 1) + if rv != C.SQLITE_OK { + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + + clib := C.CString(lib) + defer C.free(unsafe.Pointer(clib)) + centry := C.CString(entry) + defer C.free(unsafe.Pointer(centry)) + + rv = C.sqlite3_load_extension(c.db, clib, centry, nil) + if rv != C.SQLITE_OK { + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + + rv = C.sqlite3_enable_load_extension(c.db, 0) + if rv != C.SQLITE_OK { + return errors.New(C.GoString(C.sqlite3_errmsg(c.db))) + } + + return nil +} |