-
Notifications
You must be signed in to change notification settings - Fork 128
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #661 from snowflakedb/implementMFAConnectionCaching
Add connection caching for mfa and id token
- Loading branch information
Showing
194 changed files
with
18,870 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,85 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "database/sql" | ||
| "flag" | ||
| "fmt" | ||
| "log" | ||
| "os" | ||
| "strconv" | ||
|
|
||
| sf "github.com/snowflakedb/gosnowflake" | ||
| ) | ||
|
|
||
| // getDSN constructs a DSN based on the test connection parameters | ||
| func getDSN() (string, *sf.Config, error) { | ||
| env := func(k string, failOnMissing bool) string { | ||
| if value := os.Getenv(k); value != "" { | ||
| return value | ||
| } | ||
| if failOnMissing { | ||
| log.Fatalf("%v environment variable is not set.", k) | ||
| } | ||
| return "" | ||
| } | ||
|
|
||
| account := env("SNOWFLAKE_TEST_ACCOUNT", true) | ||
| user := env("SNOWFLAKE_TEST_USER", true) | ||
| password := env("SNOWFLAKE_TEST_PASSWORD", true) | ||
| host := env("SNOWFLAKE_TEST_HOST", false) | ||
| port := env("SNOWFLAKE_TEST_PORT", false) | ||
| protocol := env("SNOWFLAKE_TEST_PROTOCOL", false) | ||
|
|
||
| portStr, err := strconv.Atoi(port) | ||
| if err != nil { | ||
| return "", nil, err | ||
| } | ||
| cfg := &sf.Config{ | ||
| Account: account, | ||
| Authenticator: sf.AuthTypeUsernamePasswordMFA, | ||
| User: user, | ||
| Host: host, | ||
| Password: password, | ||
| Port: portStr, | ||
| Protocol: protocol, | ||
| } | ||
|
|
||
| dsn, err := sf.DSN(cfg) | ||
| return dsn, cfg, err | ||
| } | ||
|
|
||
| func main() { | ||
| if !flag.Parsed() { | ||
| flag.Parse() | ||
| } | ||
|
|
||
| dsn, cfg, err := getDSN() | ||
|
|
||
| if err != nil { | ||
| log.Fatalf("failed to create DSN from Config: %v, err: %v", cfg, err) | ||
| } | ||
|
|
||
| // The external browser flow should start with the call to Open | ||
| db, err := sql.Open("snowflake", dsn) | ||
| if err != nil { | ||
| log.Fatalf("failed to connect. %v, err: %v", dsn, err) | ||
| } | ||
| defer db.Close() | ||
| query := "SELECT 1" | ||
| rows, err := db.Query(query) | ||
| if err != nil { | ||
| log.Fatalf("failed to run a query. %v, err: %v", query, err) | ||
| } | ||
| defer rows.Close() | ||
| var v int | ||
| for rows.Next() { | ||
| err := rows.Scan(&v) | ||
| if err != nil { | ||
| log.Fatalf("failed to get result. err: %v", err) | ||
| } | ||
| if v != 1 { | ||
| log.Fatalf("failed to get 1. got: %v", v) | ||
| } | ||
| fmt.Printf("Congrats! You have successfully run %v with Snowflake DB!", query) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.