112 lines
2 KiB
Go
112 lines
2 KiB
Go
// Go MySQL Driver - A MySQL-Driver for Go's database/sql package
|
|
//
|
|
// Copyright 2012 The Go-MySQL-Driver Authors. All rights reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package mysql
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"io"
|
|
)
|
|
|
|
type mysqlField struct {
|
|
tableName string
|
|
name string
|
|
flags fieldFlag
|
|
fieldType byte
|
|
decimals byte
|
|
}
|
|
|
|
type mysqlRows struct {
|
|
mc *mysqlConn
|
|
columns []mysqlField
|
|
}
|
|
|
|
type binaryRows struct {
|
|
mysqlRows
|
|
}
|
|
|
|
type textRows struct {
|
|
mysqlRows
|
|
}
|
|
|
|
type emptyRows struct{}
|
|
|
|
func (rows *mysqlRows) Columns() []string {
|
|
columns := make([]string, len(rows.columns))
|
|
if rows.mc != nil && rows.mc.cfg.ColumnsWithAlias {
|
|
for i := range columns {
|
|
if tableName := rows.columns[i].tableName; len(tableName) > 0 {
|
|
columns[i] = tableName + "." + rows.columns[i].name
|
|
} else {
|
|
columns[i] = rows.columns[i].name
|
|
}
|
|
}
|
|
} else {
|
|
for i := range columns {
|
|
columns[i] = rows.columns[i].name
|
|
}
|
|
}
|
|
return columns
|
|
}
|
|
|
|
func (rows *mysqlRows) Close() error {
|
|
mc := rows.mc
|
|
if mc == nil {
|
|
return nil
|
|
}
|
|
if mc.netConn == nil {
|
|
return ErrInvalidConn
|
|
}
|
|
|
|
// Remove unread packets from stream
|
|
err := mc.readUntilEOF()
|
|
if err == nil {
|
|
if err = mc.discardResults(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
rows.mc = nil
|
|
return err
|
|
}
|
|
|
|
func (rows *binaryRows) Next(dest []driver.Value) error {
|
|
if mc := rows.mc; mc != nil {
|
|
if mc.netConn == nil {
|
|
return ErrInvalidConn
|
|
}
|
|
|
|
// Fetch next row from stream
|
|
return rows.readRow(dest)
|
|
}
|
|
return io.EOF
|
|
}
|
|
|
|
func (rows *textRows) Next(dest []driver.Value) error {
|
|
if mc := rows.mc; mc != nil {
|
|
if mc.netConn == nil {
|
|
return ErrInvalidConn
|
|
}
|
|
|
|
// Fetch next row from stream
|
|
return rows.readRow(dest)
|
|
}
|
|
return io.EOF
|
|
}
|
|
|
|
func (rows emptyRows) Columns() []string {
|
|
return nil
|
|
}
|
|
|
|
func (rows emptyRows) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (rows emptyRows) Next(dest []driver.Value) error {
|
|
return io.EOF
|
|
}
|