Code Monkey home page Code Monkey logo

Comments (2)

Abhijeetsingh4 avatar Abhijeetsingh4 commented on June 8, 2024

Code used to connect to presto->

package hive

import (
"database/sql"
"log"

// presto import
_ "github.com/experimentation-platform/presto"

)

var dsn = "http://[email protected]@localhost:8889?catalog=hive&schema=test"

// MakeQueryOnHive provides connection to hive
func MakeQueryOnHive() []map[string]interface{} {

db, err := open()
if err != nil {
	log.Fatal(err)
}
defer close(db)
sqltxt := "select * from test.test limit 10"
tables, err := query(db, sqltxt)
if err != nil {
	panic(err)
}
return tables

}

func query(db *sql.DB, sqltxt string) ([]map[string]interface{}, error) {
var tables []map[string]interface{}

rows, err := db.Query(sqltxt, sql.Named("X-Presto-User", string("[email protected]")))
if err != nil {
	return tables, err
}
defer rows.Close()
tables, err = sqlrows2Maps(rows)
if err != nil {
	return tables, err
}
return tables, nil

}

// open the connection
func open() (*sql.DB, error) {
db, err := sql.Open("presto", dsn)
if err != nil {
return db, err
}
err = db.Ping()
if err != nil {
return db, err
}
return db, nil
}

// Close the connection
func close(db *sql.DB) {
db.Close()
}

// Sqlrows2Maps sql query result rows converted to maps
func sqlrows2Maps(rws *sql.Rows) ([]map[string]interface{}, error) {
rowMaps := make([]map[string]interface{}, 0)
var columns []string
columns, err := rws.Columns()
if err != nil {
return rowMaps, err
}
values := make([]sql.RawBytes, len(columns))
scans := make([]interface{}, len(columns))
for i := range values {
scans[i] = &values[i]
}
for rws.Next() {
_ = rws.Scan(scans...)
each := map[string]interface{}{}
for i, col := range values {
each[columns[i]] = string(col)
}

	rowMaps = append(rowMaps, each)
}
return rowMaps, nil

}

from presto-go-client.

micywin avatar micywin commented on June 8, 2024

rows, err := db.Query(sqltxt, sql.Named("X-Presto-User", string("[email protected]")))
you can try to Delete the sql.Named params

from presto-go-client.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.