优化api

const.go

@@ -4,18 +4,19 @@ const (
 	_preparedStatementHeader = "X-Presto-Prepared-Statement"
 	_preparedStatementName   = "_trino_go"
 
-	XTrinoUserHeader     = "X-Trino-User"
+	_xTrinoUserHeader    = "X-Trino-User"
 	_xTrinoSourceHeader  = "X-Trino-Source"
 	_xTrinoCatalogHeader = "X-Trino-Catalog"
 	_xTrinoSchemaHeader  = "X-Trino-Schema"
 	_xTrinoSessionHeader = "X-Trino-Session"
 
-	XPrestoUserHeader     = "X-Presto-User"
+	_xPrestoUserHeader    = "X-Presto-User"
 	_xPrestoSourceHeader  = "X-Presto-Source"
 	_xPrestoCatalogHeader = "X-Presto-Catalog"
 	_xPrestoSchemaHeader  = "X-Presto-Schema"
 	_xPrestoSessionHeader = "X-Presto-Session"
 
+	UserHeader     = "User"
 	CallbackHeader = "Callback"
 
 	KerberosEnabledConfig     = "KerberosEnabled"
@@ -29,14 +30,14 @@ const (
 var (
 	vhs = map[version]map[string]string{
 		_trinoVersion: {
-			"user":    XTrinoUserHeader,
+			"user":    _xTrinoUserHeader,
 			"source":  _xTrinoSourceHeader,
 			"catalog": _xTrinoCatalogHeader,
 			"schema":  _xTrinoSchemaHeader,
 			"session": _xTrinoSessionHeader,
 		},
 		_prestoVersion: {
-			"user":    XPrestoUserHeader,
+			"user":    _xPrestoUserHeader,
 			"source":  _xPrestoSourceHeader,
 			"catalog": _xPrestoCatalogHeader,
 			"schema":  _xPrestoSchemaHeader,

statement.go

@@ -36,6 +36,7 @@ func (st *driverStmt) NumInput() int {
 }
 
 func (st *driverStmt) Exec(args []driver.Value) (driver.Result, error) {
+
 	return nil, ErrOperationNotSupported
 }
 
@@ -134,7 +135,7 @@ func (st *driverStmt) QueryContext(ctx context.Context, args []driver.NamedValue
 		var ss []string
 		for _, arg := range args {
 			switch arg.Name {
-			case vhs[v]["user"]:
+			case UserHeader:
 				st.user = arg.Value.(string)
 				hs.Add(vhs[v]["user"], st.user)
 			case CallbackHeader:

version.go

@@ -2,19 +2,21 @@ package trino
 
 type version string
 
-var (
-	v version = "Trino"
-)
-
 const (
 	_trinoVersion  version = "Trino"
 	_prestoVersion version = "Presto"
 )
 
+var (
+	v = _trinoVersion
+)
+
+// VersionTrion 设置当前 header 头为 trino 所需的
 func VersionTrino() {
 	v = _trinoVersion
 }
 
+// VersionPresto 设置当前 header 头为 persto 所需的
 func VersionPresto() {
 	v = _prestoVersion
 }