forked from mirror/go-sqlite3
104 lines
3.0 KiB
Go
104 lines
3.0 KiB
Go
// Copyright (C) 2019 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.
|
|
|
|
package sqlite3
|
|
|
|
/*
|
|
|
|
#ifndef USE_LIBSQLITE3
|
|
#include "amalgamation/sqlite3-binding.h"
|
|
#else
|
|
#include <sqlite3.h>
|
|
#endif
|
|
#include <stdlib.h>
|
|
// These wrappers are necessary because SQLITE_TRANSIENT
|
|
// is a pointer constant, and cgo doesn't translate them correctly.
|
|
|
|
static inline void my_result_text(sqlite3_context *ctx, char *p, int np) {
|
|
sqlite3_result_text(ctx, p, np, SQLITE_TRANSIENT);
|
|
}
|
|
|
|
static inline void my_result_blob(sqlite3_context *ctx, void *p, int np) {
|
|
sqlite3_result_blob(ctx, p, np, SQLITE_TRANSIENT);
|
|
}
|
|
*/
|
|
import "C"
|
|
|
|
import (
|
|
"math"
|
|
"reflect"
|
|
"unsafe"
|
|
)
|
|
|
|
const i64 = unsafe.Sizeof(int(0)) > 4
|
|
|
|
// SQLiteContext behave sqlite3_context
|
|
type SQLiteContext C.sqlite3_context
|
|
|
|
// ResultBool sets the result of an SQL function.
|
|
func (c *SQLiteContext) ResultBool(b bool) {
|
|
if b {
|
|
c.ResultInt(1)
|
|
} else {
|
|
c.ResultInt(0)
|
|
}
|
|
}
|
|
|
|
// ResultBlob sets the result of an SQL function.
|
|
// See: sqlite3_result_blob, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultBlob(b []byte) {
|
|
if i64 && len(b) > math.MaxInt32 {
|
|
C.sqlite3_result_error_toobig((*C.sqlite3_context)(c))
|
|
return
|
|
}
|
|
var p *byte
|
|
if len(b) > 0 {
|
|
p = &b[0]
|
|
}
|
|
C.my_result_blob((*C.sqlite3_context)(c), unsafe.Pointer(p), C.int(len(b)))
|
|
}
|
|
|
|
// ResultDouble sets the result of an SQL function.
|
|
// See: sqlite3_result_double, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultDouble(d float64) {
|
|
C.sqlite3_result_double((*C.sqlite3_context)(c), C.double(d))
|
|
}
|
|
|
|
// ResultInt sets the result of an SQL function.
|
|
// See: sqlite3_result_int, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultInt(i int) {
|
|
if i64 && (i > math.MaxInt32 || i < math.MinInt32) {
|
|
C.sqlite3_result_int64((*C.sqlite3_context)(c), C.sqlite3_int64(i))
|
|
} else {
|
|
C.sqlite3_result_int((*C.sqlite3_context)(c), C.int(i))
|
|
}
|
|
}
|
|
|
|
// ResultInt64 sets the result of an SQL function.
|
|
// See: sqlite3_result_int64, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultInt64(i int64) {
|
|
C.sqlite3_result_int64((*C.sqlite3_context)(c), C.sqlite3_int64(i))
|
|
}
|
|
|
|
// ResultNull sets the result of an SQL function.
|
|
// See: sqlite3_result_null, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultNull() {
|
|
C.sqlite3_result_null((*C.sqlite3_context)(c))
|
|
}
|
|
|
|
// ResultText sets the result of an SQL function.
|
|
// See: sqlite3_result_text, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultText(s string) {
|
|
h := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
|
cs, l := (*C.char)(unsafe.Pointer(h.Data)), C.int(h.Len)
|
|
C.my_result_text((*C.sqlite3_context)(c), cs, l)
|
|
}
|
|
|
|
// ResultZeroblob sets the result of an SQL function.
|
|
// See: sqlite3_result_zeroblob, http://sqlite.org/c3ref/result_blob.html
|
|
func (c *SQLiteContext) ResultZeroblob(n int) {
|
|
C.sqlite3_result_zeroblob((*C.sqlite3_context)(c), C.int(n))
|
|
}
|