snapshot of https://github.com/mattermost/mattermost.git (master, 2026-09-16; blobs > 131072B omitted)

server/public/shared/driver/objects.go

@@ -0,0 +1,114 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package driver
+
+import (
+	"context"
+	"database/sql/driver"
+
+	"github.com/mattermost/mattermost/server/public/plugin"
+)
+
+type wrapperTx struct {
+	driver.Tx
+	id  string
+	api plugin.Driver
+}
+
+func (t *wrapperTx) Commit() error {
+	return t.api.TxCommit(t.id)
+}
+
+func (t *wrapperTx) Rollback() error {
+	return t.api.TxRollback(t.id)
+}
+
+type wrapperStmt struct {
+	driver.Stmt
+	id  string
+	api plugin.Driver
+}
+
+func (s *wrapperStmt) Close() error {
+	return s.api.StmtClose(s.id)
+}
+
+func (s *wrapperStmt) NumInput() int {
+	return s.api.StmtNumInput(s.id)
+}
+
+func (s *wrapperStmt) ExecContext(_ context.Context, args []driver.NamedValue) (driver.Result, error) {
+	resultContainer, err := s.api.StmtExec(s.id, args)
+	if err != nil {
+		return nil, err
+	}
+	res := &wrapperResult{
+		res: resultContainer,
+	}
+	return res, nil
+}
+
+func (s *wrapperStmt) QueryContext(_ context.Context, args []driver.NamedValue) (driver.Rows, error) {
+	rowsID, err := s.api.StmtQuery(s.id, args)
+	if err != nil {
+		return nil, err
+	}
+	rows := &wrapperRows{
+		id:  rowsID,
+		api: s.api,
+	}
+	return rows, nil
+}
+
+// wrapperResult implements the driver.Result interface.
+// This differs from other objects because it already contains the
+// information for its methods. This does two things:
+//
+// 1. Simplifies server-side code by avoiding to track result ids
+// in a map.
+// 2. Avoids round-trip to compute result methods.
+type wrapperResult struct {
+	res plugin.ResultContainer
+}
+
+func (r *wrapperResult) LastInsertId() (int64, error) {
+	return r.res.LastID, r.res.LastIDError
+}
+
+func (r *wrapperResult) RowsAffected() (int64, error) {
+	return r.res.RowsAffected, r.res.RowsAffectedError
+}
+
+type wrapperRows struct {
+	id  string
+	api plugin.Driver
+}
+
+func (r *wrapperRows) Columns() []string {
+	return r.api.RowsColumns(r.id)
+}
+
+func (r *wrapperRows) Close() error {
+	return r.api.RowsClose(r.id)
+}
+
+func (r *wrapperRows) Next(dest []driver.Value) error {
+	return r.api.RowsNext(r.id, dest)
+}
+
+func (r *wrapperRows) HasNextResultSet() bool {
+	return r.api.RowsHasNextResultSet(r.id)
+}
+
+func (r *wrapperRows) NextResultSet() error {
+	return r.api.RowsNextResultSet(r.id)
+}
+
+func (r *wrapperRows) ColumnTypeDatabaseTypeName(index int) string {
+	return r.api.RowsColumnTypeDatabaseTypeName(r.id, index)
+}
+
+func (r *wrapperRows) ColumnTypePrecisionScale(index int) (precision, scale int64, ok bool) {
+	return r.api.RowsColumnTypePrecisionScale(r.id, index)
+}