package mydb

import (
	"database/sql"
	"sync/atomic"
	"time"
)

type MyStmt struct {
	*sql.Stmt
	db *XDb
}

func (me *MyStmt) Query(args ...interface{}) (*MyRows, error) {
	r, err := me.Stmt.Query(args...)
	if err != nil {
		return nil, err
	}
	myrows := &MyRows{Rows: r, values: make(map[string]interface{})}

	myrows.rowId = atomic.AddUint64(&me.db.rowId, 1)
	myrows.createTime = time.Now().Unix()
	myrows.db = me.db
	myrows.SaveFields()
	me.db.rowMutex.Lock()
	me.db.rowMap[myrows.rowId] = myrows
	me.db.rowMutex.Unlock()

	return myrows, nil
}
func (me *MyStmt) QueryRow(args ...interface{}) (*MyRow, error) {
	//这里调用了query来实现
	r, err := me.Query(args...)
	if err != nil {
		return nil, err
	}
	myrow := &MyRow{rows: r}
	myrow.readData()
	return myrow, nil
}