result.go

// Copyright 2022 lesismal. All rights reserved.
// Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.

package sqlw

import (
	"database/sql"
	"fmt"
)

type sqlResult struct {
	sql.Result
	db       *DB
	query    string
	args     []interface{}
	notFound bool
}

type Result = *sqlResult

func (r *sqlResult) LastInsertId() (int64, error) {
	if r.Result != nil {
		return r.Result.LastInsertId()
	}
	return 0, nil
}

func (r *sqlResult) RowsAffected() (int64, error) {
	if r.Result != nil {
		return r.Result.RowsAffected()
	}
	return 0, nil
}

func (r *sqlResult) Sql() string {
	return fmt.Sprintf(`[%s], %v`, r.query, r.args)
}

func (r *sqlResult) Query() string {
	return r.query
}

func (r *sqlResult) Args() []interface{} {
	return r.args
}

func (r *sqlResult) IsNotFound() bool {
	return r.notFound
}

func newResult(db *DB, r sql.Result, query string, args []interface{}, notFound bool) Result {
	ret := &sqlResult{r, db, query, args, notFound}
	if db != nil && db.printSql != nil {
		db.printSql(ret.Sql())
	}
	return ret
}