From 1818e42cc28255bff7ac7e33aa7e82357c5bcfc9 Mon Sep 17 00:00:00 2001
From: RudraNirvan <rudra.nirvan@gmail.com>
Date: Mon, 23 May 2022 04:25:30 +0530
Subject: [PATCH] Create connector in mssql

---
 .../service_principal_authtoken.go            | 66 +++++++++++++++++++
 mssql.go                                      | 14 ++++
 2 files changed, 80 insertions(+)
 create mode 100644 examples/azuread-service-principal-authtoken/service_principal_authtoken.go

diff --git a/examples/azuread-service-principal-authtoken/service_principal_authtoken.go b/examples/azuread-service-principal-authtoken/service_principal_authtoken.go
new file mode 100644
index 00000000..9e9d52c6
--- /dev/null
+++ b/examples/azuread-service-principal-authtoken/service_principal_authtoken.go
@@ -0,0 +1,66 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"flag"
+	"fmt"
+	"log"
+
+	mssql "github.com/denisenkom/go-mssqldb"
+)
+
+var (
+	debug         = flag.Bool("debug", true, "enable debugging")
+	password      = flag.String("password", "", "the client secret for the app/client ID")
+	port     *int = flag.Int("port", 1433, "the database port")
+	server        = flag.String("server", "", "the database server")
+	database      = flag.String("database", "", "the database name")
+)
+
+func main() {
+	flag.Parse()
+
+	if *debug {
+		fmt.Printf(" password:%s\n", *password)
+		fmt.Printf(" port:%d\n", *port)
+		fmt.Printf(" server:%s\n", *server)
+		fmt.Printf(" database:%s\n", *database)
+	}
+
+	connString := fmt.Sprintf("server=%s;password=%s;port=%d;database=%s;fedauth=ActiveDirectoryServicePrincipalAccessToken;", *server, *password, *port, *database)
+	if *debug {
+		fmt.Printf(" connString:%s\n", connString)
+	}
+
+	tokenProviderWithCtx := func(ctx context.Context) (string, error) {
+		return "access_token", nil
+	}
+
+	connector, err := mssql.NewConnectorWithAccessTokenProvider(connString, tokenProviderWithCtx)
+	conn := sql.OpenDB(connector)
+
+	//conn, err := sql.Open(azuread.DriverName, connString)
+	if err != nil {
+		log.Fatal("Open connection failed:", err.Error())
+	}
+	defer conn.Close()
+
+	stmt, err := conn.Prepare("select 1, 'abc'")
+	if err != nil {
+		log.Fatal("Prepare failed:", err.Error())
+	}
+	defer stmt.Close()
+
+	row := stmt.QueryRow()
+	var somenumber int64
+	var somechars string
+	err = row.Scan(&somenumber, &somechars)
+	if err != nil {
+		log.Fatal("Scan failed:", err.Error())
+	}
+	fmt.Printf("somenumber:%d\n", somenumber)
+	fmt.Printf("somechars:%s\n", somechars)
+
+	fmt.Printf("bye\n")
+}
diff --git a/mssql.go b/mssql.go
index fbd44d1f..274192ff 100644
--- a/mssql.go
+++ b/mssql.go
@@ -125,6 +125,20 @@ func NewConnector(dsn string) (*Connector, error) {
 	return c, nil
 }
 
+// NewConnectorWithAccessTokenProvider creates a new connector from a DSN using the given
+// access token provider. The returned connector may be used with sql.OpenDB.
+func NewConnectorWithAccessTokenProvider(dsn string, tokenProvider func(ctx context.Context) (string, error)) (*Connector, error) {
+	params, _, err := msdsn.Parse(dsn)
+	if err != nil {
+		return nil, err
+	}
+
+	return NewSecurityTokenConnector(
+		params,
+		tokenProvider,
+	)
+}
+
 // NewConnectorConfig creates a new Connector for a DSN Config struct.
 // The returned connector may be used with sql.OpenDB.
 func NewConnectorConfig(config msdsn.Config) *Connector {