mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-29 00:45:31 +00:00
4f93520e47
* rebase js-layer PR from @ice3man543 * package restructuring * working * fix duplicated event & matcher status * fix lint error * fix response field * add new functions * multiple minor improvements * fix incorrect stats in js protocol * sort output metadata in cli * remove temp files * remove dead code * add unit and integration test * fix lint error * add jsdoclint using llm * fix error in test * add js lint using llm * generate docs of libs * llm lint * remove duplicated docs * update generated docs * update prompt in doclint * update docs * temp disable version check test * fix unit test and add retry * fix panic in it * update and move jsdocs * updated jsdocs * update docs * update container platform in test * dir restructure and adding docs * add api_reference and remove markdown docs * fix imports * add javascript design and contribution docs * add js protocol documentation * update integration test and docs * update doc ext mdx->md * minor update to docs * new integration test and more * move go libs and add docs * gen new net docs and more * final docs update * add new devtool * use fastdialer * fix build fail * use fastdialer + network sandbox support * add reserved keyword 'Port' * update Port to new syntax * misc update * always enable templatectx in js protocol * move docs to 'js-proto-docs' repo * remove scrapefuncs binary --------- Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
// UnmarshalSQLRows unmarshals sql rows to json
|
|
//
|
|
// This function provides a way to unmarshal arbitrary sql rows
|
|
// to json.
|
|
func UnmarshalSQLRows(rows *sql.Rows) ([]byte, error) {
|
|
columnTypes, err := rows.ColumnTypes()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
count := len(columnTypes)
|
|
finalRows := []interface{}{}
|
|
|
|
for rows.Next() {
|
|
|
|
scanArgs := make([]interface{}, count)
|
|
|
|
for i, v := range columnTypes {
|
|
|
|
switch v.DatabaseTypeName() {
|
|
case "VARCHAR", "TEXT", "UUID", "TIMESTAMP":
|
|
scanArgs[i] = new(sql.NullString)
|
|
case "BOOL":
|
|
scanArgs[i] = new(sql.NullBool)
|
|
case "INT4":
|
|
scanArgs[i] = new(sql.NullInt64)
|
|
default:
|
|
scanArgs[i] = new(sql.NullString)
|
|
}
|
|
}
|
|
|
|
err := rows.Scan(scanArgs...)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
masterData := map[string]interface{}{}
|
|
|
|
for i, v := range columnTypes {
|
|
|
|
if z, ok := (scanArgs[i]).(*sql.NullBool); ok {
|
|
masterData[v.Name()] = z.Bool
|
|
continue
|
|
}
|
|
|
|
if z, ok := (scanArgs[i]).(*sql.NullString); ok {
|
|
masterData[v.Name()] = z.String
|
|
continue
|
|
}
|
|
|
|
if z, ok := (scanArgs[i]).(*sql.NullInt64); ok {
|
|
masterData[v.Name()] = z.Int64
|
|
continue
|
|
}
|
|
|
|
if z, ok := (scanArgs[i]).(*sql.NullFloat64); ok {
|
|
masterData[v.Name()] = z.Float64
|
|
continue
|
|
}
|
|
|
|
if z, ok := (scanArgs[i]).(*sql.NullInt32); ok {
|
|
masterData[v.Name()] = z.Int32
|
|
continue
|
|
}
|
|
|
|
masterData[v.Name()] = scanArgs[i]
|
|
}
|
|
|
|
finalRows = append(finalRows, masterData)
|
|
}
|
|
return jsoniter.Marshal(finalRows)
|
|
}
|