package util

import (
	"ReverseEngine/entity"
	"database/sql"
	"strings"
)

type DbUtil struct {
}

func FindColumnMessage(dbName string, tableName string, db *sql.DB) []entity.ColumnMessage {
	var cms []entity.ColumnMessage

	row, err := db.Query(strings.Join([]string{"show full columns from ", dbName, ".", tableName}, ""))
	if err != nil {
		panic("数据库查询出错：" + err.Error())
	}

	for row.Next() {
		var cm entity.ColumnMessage
		err := row.Scan(&cm.Field, &cm.Type, &cm.Collation, &cm.Null, &cm.Key, &cm.Default, &cm.Extra, &cm.Privileges, &cm.Comment)
		if err != nil {
			panic("字段赋值错误了" + err.Error())
			return nil
		}

		cms = append(cms, cm)
	}

	return cms
}