190 lines
4.3 KiB
Go
190 lines
4.3 KiB
Go
package dbUtil
|
||
|
||
import (
|
||
"database/sql"
|
||
"errors"
|
||
)
|
||
|
||
// 数据表结构
|
||
type DataTable struct {
|
||
// 行对象集合
|
||
rowData []*DataRow
|
||
|
||
// 列名称集合
|
||
columnNames map[string]int
|
||
}
|
||
|
||
// 数据表初始化
|
||
// rows:原始的数据行信息
|
||
// 返回值：
|
||
// error:初始化的错误信息
|
||
func (this *DataTable) init(rows *sql.Rows) error {
|
||
defer func() {
|
||
rows.Close()
|
||
}()
|
||
|
||
// 读取列信息和保存列名称
|
||
tmpColumns, errMsg := rows.Columns()
|
||
if errMsg != nil {
|
||
return errMsg
|
||
}
|
||
this.columnNames = make(map[string]int)
|
||
for index, val := range tmpColumns {
|
||
this.columnNames[val] = index
|
||
}
|
||
|
||
// 读取行数据
|
||
this.rowData = make([]*DataRow, 0)
|
||
columnCount := len(this.columnNames)
|
||
|
||
args := make([]interface{}, columnCount)
|
||
for rows.Next() {
|
||
values := make([]interface{}, columnCount)
|
||
for i := 0; i < columnCount; i++ {
|
||
args[i] = &values[i]
|
||
}
|
||
rows.Scan(args...)
|
||
|
||
this.rowData = append(this.rowData, newDataRow(this, values))
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// 获取原始单元格值（一般为:string或[]byte）
|
||
// rowIndex:行序号
|
||
// cellIndex:单元格序号
|
||
// 返回值:
|
||
// interface{}:原始单元格值（一般为:string或[]byte）
|
||
// error:获取错误信息
|
||
func (this *DataTable) OriginCellValueByIndex(rowIndex int, cellIndex int) (interface{}, error) {
|
||
if len(this.rowData) <= rowIndex {
|
||
return nil, errors.New("row out of range")
|
||
}
|
||
|
||
rowItem := this.rowData[rowIndex]
|
||
if len(rowItem.cells) <= cellIndex {
|
||
return nil, errors.New("column out of range")
|
||
}
|
||
|
||
return rowItem.OriginCellValue(cellIndex)
|
||
}
|
||
|
||
// 获取原始单元格值（一般为:string或[]byte）
|
||
// rowIndex:行序号
|
||
// cellIndex:单元格序号
|
||
// 返回值:
|
||
// interface{}:原始单元格值（一般为:string或[]byte）
|
||
// error:获取错误信息
|
||
func (this *DataTable) OriginCellValueByCellName(rowIndex int, cellName string) (interface{}, error) {
|
||
if len(this.rowData) <= rowIndex {
|
||
return nil, errors.New("row out of range")
|
||
}
|
||
|
||
rowItem := this.rowData[rowIndex]
|
||
|
||
return rowItem.OriginCellValueByName(cellName)
|
||
}
|
||
|
||
// 获取字符串的单元格值（有可能为nil）
|
||
// rowIndex:行序号
|
||
// cellIndex:单元格序号
|
||
// 返回值:
|
||
// interface{}:字符串的单元格值（有可能为nil）
|
||
// error:获取错误信息
|
||
func (this *DataTable) CellValueByIndex(rowIndex int, cellIndex int) (interface{}, error) {
|
||
if len(this.rowData) <= rowIndex {
|
||
return nil, errors.New("row out of range")
|
||
}
|
||
|
||
rowItem := this.rowData[rowIndex]
|
||
if len(rowItem.cells) <= cellIndex {
|
||
return nil, errors.New("column out of range")
|
||
}
|
||
|
||
return rowItem.CellValue(cellIndex)
|
||
}
|
||
|
||
// 获取字符串的单元格值（有可能为nil）
|
||
// rowIndex:行序号
|
||
// cellIndex:单元格序号
|
||
// 返回值:
|
||
// interface{}:字符串的单元格值（有可能为nil）
|
||
// error:获取错误信息
|
||
func (this *DataTable) CellValueByName(rowIndex int, cellName string) (interface{}, error) {
|
||
if len(this.rowData) <= rowIndex {
|
||
return nil, errors.New("row out of range")
|
||
}
|
||
|
||
rowItem := this.rowData[rowIndex]
|
||
|
||
return rowItem.CellValueByName(cellName)
|
||
}
|
||
|
||
// 获取行对象
|
||
// rowIndex:行序号
|
||
// 返回值:
|
||
// *DataRow:行对象
|
||
// error:错误信息
|
||
func (this *DataTable) Row(rowIndex int) (*DataRow, error) {
|
||
if len(this.rowData) <= rowIndex {
|
||
return nil, errors.New("row out of range")
|
||
}
|
||
|
||
return this.rowData[rowIndex], nil
|
||
}
|
||
|
||
// 根据列名获取列序号
|
||
// cellName:列名
|
||
// 返回值:
|
||
// int:列序号
|
||
func (this *DataTable) cellIndex(cellName string) int {
|
||
cellIndex, isExist := this.columnNames[cellName]
|
||
if isExist == false {
|
||
return -1
|
||
}
|
||
|
||
return cellIndex
|
||
}
|
||
|
||
// 获取所有列的名字
|
||
// 返回值:
|
||
// []string:列字段名集合
|
||
func (this *DataTable) Columns() []string {
|
||
result := make([]string, len(this.columnNames))
|
||
for key, val := range this.columnNames {
|
||
result[val] = key
|
||
}
|
||
|
||
return result
|
||
}
|
||
|
||
// 获取列数量
|
||
// 返回值:
|
||
// int:列数量
|
||
func (this *DataTable) ColumnCount() int {
|
||
return len(this.columnNames)
|
||
}
|
||
|
||
// 获取数据行数
|
||
// 返回值:
|
||
// int:行数
|
||
func (this *DataTable) RowCount() int {
|
||
return len(this.rowData)
|
||
}
|
||
|
||
// 新建数据表对象
|
||
// rows:数据行对象
|
||
// 返回值：
|
||
// *DataTable:数据表对象
|
||
// error:错误信息
|
||
func NewDataTable(rows *sql.Rows) (*DataTable, error) {
|
||
table := &DataTable{}
|
||
errMsg := table.init(rows)
|
||
if errMsg != nil {
|
||
return nil, errMsg
|
||
}
|
||
|
||
return table, nil
|
||
}
|