本文整理汇总了Golang中github.com/cockroachdb/cockroach/pkg/testutils/sqlutils.PGUrl函数的典型用法代码示例。如果您正苦于以下问题:Golang PGUrl函数的具体用法?Golang PGUrl怎么用?Golang PGUrl使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了PGUrl函数的15个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Golang代码示例。
示例1: TestPGWireDBName
func TestPGWireDBName(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPGWireDBName", url.User(security.RootUser))
pgURL.Path = "foo"
defer cleanupFn()
{
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
if _, err := db.Exec(`CREATE DATABASE foo`); err != nil {
t.Fatal(err)
}
if _, err := db.Exec(`CREATE TABLE bar (i INT PRIMARY KEY)`); err != nil {
t.Fatal(err)
}
}
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
if _, err := db.Exec(`INSERT INTO bar VALUES ($1)`, 1); err != nil {
t.Fatal(err)
}
}
示例2: TestPGWireConnectionCloseReleasesLeases
// Test that abruptly closing a pgwire connection releases all leases held by
// that session.
func TestPGWireConnectionCloseReleasesLeases(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, kvDB := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
url, cleanupConn := sqlutils.PGUrl(t, s.ServingAddr(), "SetupServer", url.User(security.RootUser))
defer cleanupConn()
conn, err := pq.Open(url.String())
if err != nil {
t.Fatal(err)
}
ex := conn.(driver.Execer)
if _, err := ex.Exec("CREATE DATABASE test", nil); err != nil {
t.Fatal(err)
}
if _, err := ex.Exec("CREATE TABLE test.t (i INT PRIMARY KEY)", nil); err != nil {
t.Fatal(err)
}
// Start a txn so leases are accumulated by queries.
if _, err := ex.Exec("BEGIN", nil); err != nil {
t.Fatal(err)
}
// Get a table lease.
if _, err := ex.Exec("SELECT * FROM test.t", nil); err != nil {
t.Fatal(err)
}
// Abruptly close the connection.
if err := conn.Close(); err != nil {
t.Fatal(err)
}
// Verify that there are no leases held.
tableDesc := sqlbase.GetTableDescriptor(kvDB, "test", "t")
lm := s.LeaseManager().(*LeaseManager)
// Looking for a table state validates that there used to be a lease on the
// table.
ts := lm.findTableState(tableDesc.ID, false /* create */)
if ts == nil {
t.Fatal("table state not found")
}
ts.mu.Lock()
leases := ts.active.data
ts.mu.Unlock()
if len(leases) != 1 {
t.Fatalf("expected one lease, found: %d", len(leases))
}
// Wait for the lease to be released.
util.SucceedsSoon(t, func() error {
ts.mu.Lock()
refcount := ts.active.data[0].refcount
ts.mu.Unlock()
if refcount != 0 {
return errors.Errorf(
"expected lease to be unused, found refcount: %d", refcount)
}
return nil
})
}
示例3: TestDumpBytes
func TestDumpBytes(t *testing.T) {
defer leaktest.AfterTest(t)()
c, err := newCLITest(t, false)
if err != nil {
t.Fatal(err)
}
defer c.stop(true)
url, cleanup := sqlutils.PGUrl(t, c.ServingAddr(), "TestDumpBytes", url.User(security.RootUser))
defer cleanup()
conn := makeSQLConn(url.String())
defer conn.Close()
if err := conn.Exec(`
CREATE DATABASE d;
SET DATABASE = d;
CREATE TABLE t (b BYTES PRIMARY KEY);
`, nil); err != nil {
t.Fatal(err)
}
for i := int64(0); i < 256; i++ {
if err := conn.Exec("INSERT INTO t VALUES ($1)", []driver.Value{[]byte{byte(i)}}); err != nil {
t.Fatal(err)
}
}
var b bytes.Buffer
if err := DumpTable(&b, conn, "d", "t"); err != nil {
t.Fatal(err)
}
dump := b.String()
b.Reset()
if err := conn.Exec(`
CREATE DATABASE o;
SET DATABASE = o;
`, nil); err != nil {
t.Fatal(err)
}
if err := conn.Exec(dump, nil); err != nil {
t.Fatal(err)
}
if err := DumpTable(&b, conn, "o", "t"); err != nil {
t.Fatal(err)
}
dump2 := b.String()
if dump != dump2 {
t.Fatalf("unmatching dumps:\n%s\n%s", dump, dump2)
}
}
示例4: BenchmarkPgbenchExec_Cockroach
func BenchmarkPgbenchExec_Cockroach(b *testing.B) {
defer tracing.Disable()()
s, _, _ := serverutils.StartServer(b, base.TestServerArgs{Insecure: true})
defer s.Stopper().Stop()
pgUrl, cleanupFn := sqlutils.PGUrl(
b, s.ServingAddr(), "benchmarkCockroach", url.User(security.RootUser))
pgUrl.RawQuery = "sslmode=disable"
defer cleanupFn()
execPgbench(b, pgUrl)
}
示例5: TestRollbackInRestartWait
// TestRollbackInRestartWait ensures that a ROLLBACK while the txn is in the
// RetryWait state works.
func TestRollbackInRestartWait(t *testing.T) {
defer leaktest.AfterTest(t)()
aborter := NewTxnAborter()
defer aborter.Close(t)
params, _ := createTestServerParams()
params.Knobs.SQLExecutor = aborter.executorKnobs()
s, sqlDB, _ := serverutils.StartServer(t, params)
defer s.Stopper().Stop()
{
pgURL, cleanup := sqlutils.PGUrl(t, s.ServingAddr(), "TestRollbackInRestartWait", url.User(security.RootUser))
defer cleanup()
if err := aborter.Init(pgURL); err != nil {
t.Fatal(err)
}
}
if _, err := sqlDB.Exec(`
CREATE DATABASE t;
CREATE TABLE t.test (k INT PRIMARY KEY, v TEXT);
`); err != nil {
t.Fatal(err)
}
// Set up error injection that causes retries.
const insertStmt = "INSERT INTO t.test(k, v) VALUES (0, 'boulanger')"
if err := aborter.QueueStmtForAbortion(
insertStmt, 1 /* abortCount */, false, /* willBeRetriedIbid */
); err != nil {
t.Fatal(err)
}
tx, err := sqlDB.Begin()
if err != nil {
t.Fatal(err)
}
if _, err := tx.Exec("SAVEPOINT cockroach_restart"); err != nil {
t.Fatal(err)
}
if _, err := tx.Exec(insertStmt); err != nil {
t.Fatal(err)
}
if _, err := tx.Exec("RELEASE SAVEPOINT cockroach_restart"); !testutils.IsError(
err, "pq: restart transaction") {
t.Fatalf("unexpected error: %s", err)
}
if err := tx.Rollback(); err != nil {
t.Fatal(err)
}
}
示例6: setUser
// setUser sets the DB client to the specified user.
// It returns a cleanup function to be run when the credentials
// are no longer needed.
func (t *logicTest) setUser(user string) func() {
var outDBName string
if t.db != nil {
var inDBName string
if err := t.db.QueryRow("SHOW DATABASE").Scan(&inDBName); err != nil {
t.Fatal(err)
}
defer func() {
if inDBName != outDBName {
// Propagate the DATABASE setting to the newly-live connection.
if _, err := t.db.Exec(fmt.Sprintf("SET DATABASE = %s", inDBName)); err != nil {
t.Fatal(err)
}
}
}()
}
if t.clients == nil {
t.clients = map[string]*gosql.DB{}
}
if db, ok := t.clients[user]; ok {
t.db = db
t.user = user
if err := t.db.QueryRow("SHOW DATABASE").Scan(&outDBName); err != nil {
t.Fatal(err)
}
// No cleanup necessary, but return a no-op func to avoid nil pointer dereference.
return func() {}
}
pgURL, cleanupFunc := sqlutils.PGUrl(t.T, t.srv.ServingAddr(), "TestLogic", url.User(user))
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
t.clients[user] = db
t.db = db
t.user = user
if t.verbose {
fmt.Printf("--- new user: %s\n", user)
}
return cleanupFunc
}
示例7: TestPGPrepareNameQual
// Names should be qualified automatically during Prepare when a database name
// was given in the connection string.
func TestPGPrepareNameQual(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPGPrepareNameQual", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
if _, err := db.Exec(`CREATE DATABASE IF NOT EXISTS testing`); err != nil {
t.Fatal(err)
}
pgURL.Path = "/testing"
db2, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db2.Close()
statements := []string{
`CREATE TABLE IF NOT EXISTS f (v INT)`,
`INSERT INTO f VALUES (42)`,
`SELECT * FROM f`,
`DELETE FROM f WHERE v = 42`,
`DROP TABLE IF EXISTS f`,
}
for _, stmtString := range statements {
if _, err = db2.Exec(stmtString); err != nil {
t.Fatal(err)
}
stmt, err := db2.Prepare(stmtString)
if err != nil {
t.Fatal(err)
}
if _, err = stmt.Exec(); err != nil {
t.Fatal(err)
}
}
}
示例8: TestPGPrepareFail
func TestPGPrepareFail(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPGPrepareFail", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
testFailures := map[string]string{
"SELECT $1 = $1": "pq: could not determine data type of placeholder $1",
"SELECT $1": "pq: could not determine data type of placeholder $1",
"SELECT $1 + $1": "pq: could not determine data type of placeholder $1",
"SELECT CASE WHEN TRUE THEN $1 END": "pq: could not determine data type of placeholder $1",
"SELECT CASE WHEN TRUE THEN $1 ELSE $2 END": "pq: could not determine data type of placeholder $1",
"SELECT $1 > 0 AND NOT $1": "pq: incompatible NOT argument type: int",
"CREATE TABLE $1 (id INT)": "pq: syntax error at or near \"1\"\nCREATE TABLE $1 (id INT)\n ^\n",
"UPDATE d.t SET s = i + $1": "pq: unsupported binary operator: <int> + <placeholder{1}> (desired <string>)",
"SELECT $0 > 0": "pq: invalid placeholder name: $0",
"SELECT $2 > 0": "pq: could not determine data type of placeholder $1",
"SELECT 3 + CASE (4) WHEN 4 THEN $1 END": "pq: could not determine data type of placeholder $1",
"SELECT ($1 + $1) + CURRENT_DATE()": "pq: could not determine data type of placeholder $1",
"SELECT $1 + $2, $2::FLOAT": "pq: could not determine data type of placeholder $1",
"SELECT $1[2]": "pq: could not determine data type of placeholder $1",
"SELECT ($1 + 2) + ($1 + 2.5::FLOAT)": "pq: unsupported binary operator: <int> + <float>",
}
if _, err := db.Exec(`CREATE DATABASE d; CREATE TABLE d.t (i INT, s STRING, d INT)`); err != nil {
t.Fatal(err)
}
for query, reason := range testFailures {
if stmt, err := db.Prepare(query); err == nil {
t.Errorf("expected error: %s", query)
if err := stmt.Close(); err != nil {
t.Fatal(err)
}
} else if err.Error() != reason {
t.Errorf(`%s: got: %q, expected: %q`, query, err, reason)
}
}
}
示例9: TestAbortedTxnOnlyRetriedOnce
// Test that aborted txn are only retried once.
// Prevents regressions of #8456.
func TestAbortedTxnOnlyRetriedOnce(t *testing.T) {
defer leaktest.AfterTest(t)()
aborter := NewTxnAborter()
defer aborter.Close(t)
params, _ := createTestServerParams()
params.Knobs.SQLExecutor = aborter.executorKnobs()
// Disable one phase commits because they cannot be restarted.
params.Knobs.Store.(*storage.StoreTestingKnobs).DisableOnePhaseCommits = true
s, sqlDB, _ := serverutils.StartServer(t, params)
defer s.Stopper().Stop()
{
pgURL, cleanup := sqlutils.PGUrl(t, s.ServingAddr(), "TestAbortedTxnOnlyRetriedOnce", url.User(security.RootUser))
defer cleanup()
if err := aborter.Init(pgURL); err != nil {
t.Fatal(err)
}
}
const insertStmt = "INSERT INTO t.test(k, v) VALUES (1, 'boulanger')"
if err := aborter.QueueStmtForAbortion(
insertStmt, 1 /* abortCount */, true, /* willBeRetriedIbid */
); err != nil {
t.Fatal(err)
}
if _, err := sqlDB.Exec(`
CREATE DATABASE t;
CREATE TABLE t.test (k INT PRIMARY KEY, v TEXT);
`); err != nil {
t.Fatal(err)
}
if _, err := sqlDB.Exec(insertStmt); err != nil {
t.Fatalf("unexpected error: %s", err)
}
execCount, ok := aborter.GetExecCount(insertStmt)
if !ok {
t.Fatalf("aborter has no state on %q", insertStmt)
}
if execCount != 2 {
t.Fatalf("expected %q to be executed 2 times, but got %d", insertStmt, execCount)
}
}
示例10: TestCmdCompleteVsEmptyStatements
// A DDL should return "CommandComplete", not "EmptyQuery" Response.
func TestCmdCompleteVsEmptyStatements(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(
t, s.ServingAddr(), "TestCmdCompleteVsEmptyStatements", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
// lib/pq handles the empty query response by returning a nil driver.Result.
// Unfortunately gosql.Exec wraps that, nil or not, in a gosql.Result which doesn't
// expose the underlying driver.Result.
// gosql.Result does however have methods which attempt to dereference the underlying
// driver.Result and can thus be used to determine if it is nil.
// TODO(dt): This would be prettier and generate better failures with testify/assert's helpers.
// Result of a DDL (command complete) yields a non-nil underlying driver result.
nonempty, err := db.Exec(`CREATE DATABASE IF NOT EXISTS testing`)
if err != nil {
t.Fatal(err)
}
_, _ = nonempty.RowsAffected() // should not panic if lib/pq returned a non-nil result.
empty, err := db.Exec(" ; ; ;")
if err != nil {
t.Fatal(err)
}
rows, err := empty.RowsAffected()
if rows != 0 {
t.Fatalf("expected 0 rows, got %d", rows)
}
if err == nil {
t.Fatal("expected error")
}
}
示例11: getClient
func (t *parallelTest) getClient(nodeIdx, clientIdx int) *gosql.DB {
for len(t.clients[nodeIdx]) <= clientIdx {
// Add a client.
pgURL, cleanupFunc := sqlutils.PGUrl(t.T,
t.cluster.Server(nodeIdx).ServingAddr(),
"TestParallel",
url.User(security.RootUser))
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
sqlutils.MakeSQLRunner(t, db).Exec("SET DATABASE = test")
t.cluster.Stopper().AddCloser(
stop.CloserFn(func() {
_ = db.Close()
cleanupFunc()
}))
t.clients[nodeIdx] = append(t.clients[nodeIdx], db)
}
return t.clients[nodeIdx][clientIdx]
}
示例12: TestPrepareSyntax
func TestPrepareSyntax(t *testing.T) {
defer leaktest.AfterTest(t)()
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPrepareSyntax", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
const strTest = `SELECT """test"""`
if _, err := db.Exec(`SET SYNTAX = traditional`); err != nil {
t.Fatal(err)
}
if _, err := db.Prepare(strTest); err == nil {
t.Fatal("expected error")
}
if _, err := db.Exec(`SET SYNTAX = modern`); err != nil {
t.Fatal(err)
}
stmt, err := db.Prepare(strTest)
if err != nil {
t.Fatalf("unexpected error: %s", err)
}
var v string
if err := stmt.QueryRow().Scan(&v); err != nil {
t.Fatalf("unexpected error: %s", err)
} else if v != "test" {
t.Fatalf("unexpected result: %q", v)
}
}
示例13: StartServer
// StartServer creates a test server and sets up a gosql DB connection.
// The server should be stopped by calling server.Stopper().Stop().
func StartServer(
t testing.TB, params base.TestServerArgs,
) (TestServerInterface, *gosql.DB, *client.DB) {
server, err := StartServerRaw(params)
if err != nil {
t.Fatal(err)
}
kvClient := server.KVClient().(*client.DB)
pgURL, cleanupGoDB := sqlutils.PGUrl(
t, server.ServingAddr(), "StartServer", url.User(security.RootUser))
pgURL.Path = params.UseDatabase
goDB, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
server.Stopper().AddCloser(
stop.CloserFn(func() {
_ = goDB.Close()
cleanupGoDB()
}))
return server, goDB, kvClient
}
示例14: TestPGPreparedExec
//.........这里部分代码省略.........
[]preparedExecTest{
baseTest.RowsAffected(1).SetArgs(
int64(0),
float64(0),
"",
[]byte{},
time.Time{}, // date
time.Time{}, // timestamp
time.Time{}, // timestamptz
time.Hour.String(),
true,
"0.0", // decimal
),
},
},
{
"DROP DATABASE d",
[]preparedExecTest{
baseTest,
},
},
// An empty string is valid in postgres.
{
"",
[]preparedExecTest{
baseTest.RowsAffectedErr("no RowsAffected available after the empty statement"),
},
},
// Empty statements are permitted.
{
";",
[]preparedExecTest{
baseTest.RowsAffectedErr("no RowsAffected available after the empty statement"),
},
},
// Any number of empty statements are permitted with a single statement
// anywhere.
{
"; ; SET DATABASE = system; ;",
[]preparedExecTest{
baseTest,
},
},
}
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPGPreparedExec", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
runTests := func(query string, tests []preparedExecTest, execFunc func(...interface{}) (gosql.Result, error)) {
for _, test := range tests {
if testing.Verbose() || log.V(1) {
log.Infof(context.Background(), "exec: %s", query)
}
if result, err := execFunc(test.qargs...); err != nil {
if test.error == "" {
t.Errorf("%s: %v: unexpected error: %s", query, test.qargs, err)
} else if err.Error() != test.error {
t.Errorf("%s: %v: expected error: %s, got %s", query, test.qargs, test.error, err)
}
} else {
rowsAffected, err := result.RowsAffected()
if !testutils.IsError(err, test.rowsAffectedErr) {
t.Errorf("%s: %v: expected %q, got %v", query, test.qargs, test.rowsAffectedErr, err)
} else if rowsAffected != test.rowsAffected {
t.Errorf("%s: %v: expected %v, got %v", query, test.qargs, test.rowsAffected, rowsAffected)
}
}
}
}
for _, execTest := range execTests {
runTests(execTest.query, execTest.tests, func(args ...interface{}) (gosql.Result, error) {
return db.Exec(execTest.query, args...)
})
}
for _, execTest := range execTests {
if testing.Verbose() || log.V(1) {
log.Infof(context.Background(), "prepare: %s", execTest.query)
}
if stmt, err := db.Prepare(execTest.query); err != nil {
t.Errorf("%s: prepare error: %s", execTest.query, err)
} else {
func() {
defer stmt.Close()
runTests(execTest.query, execTest.tests, stmt.Exec)
}()
}
}
}
示例15: TestPGPreparedQuery
//.........这里部分代码省略.........
"SELECT ANNOTATE_TYPE($1, int)": {
baseTest.SetArgs(12).Results(12),
},
"SELECT $1 + $2, ANNOTATE_TYPE($2, float)": {
baseTest.SetArgs(12, 23).Results(35, 23),
},
"INSERT INTO d.T VALUES ($1 + 1) RETURNING a": {
baseTest.SetArgs(1).Results(2),
baseTest.SetArgs(11).Results(12),
},
"INSERT INTO d.T VALUES (-$1) RETURNING a": {
baseTest.SetArgs(1).Results(-1),
baseTest.SetArgs(-999).Results(999),
},
"INSERT INTO d.two (a, b) VALUES (~$1, $1 + $2) RETURNING a, b": {
baseTest.SetArgs(5, 6).Results(-6, 11),
},
"INSERT INTO d.str (s) VALUES (LEFT($1, 3)) RETURNING s": {
baseTest.SetArgs("abcdef").Results("abc"),
baseTest.SetArgs("123456").Results("123"),
},
"INSERT INTO d.str (b) VALUES (COALESCE($1, 'strLit')) RETURNING b": {
baseTest.SetArgs(nil).Results("strLit"),
baseTest.SetArgs("123456").Results("123456"),
},
"INSERT INTO d.intStr VALUES ($1, 'hello ' || $1::TEXT) RETURNING *": {
baseTest.SetArgs(123).Results(123, "hello 123"),
},
}
s, _, _ := serverutils.StartServer(t, base.TestServerArgs{})
defer s.Stopper().Stop()
pgURL, cleanupFn := sqlutils.PGUrl(t, s.ServingAddr(), "TestPGPreparedQuery", url.User(security.RootUser))
defer cleanupFn()
db, err := gosql.Open("postgres", pgURL.String())
if err != nil {
t.Fatal(err)
}
defer db.Close()
runTests := func(query string, prepared bool, tests []preparedQueryTest, queryFunc func(...interface{}) (*gosql.Rows, error)) {
for _, test := range tests {
if testing.Verbose() || log.V(1) {
log.Infof(context.Background(), "query: %s", query)
}
rows, err := queryFunc(test.qargs...)
if err != nil {
if test.error == "" {
t.Errorf("%s: %v: unexpected error: %s", query, test.qargs, err)
} else {
expectedErr := test.error
if prepared && test.preparedError != "" {
expectedErr = test.preparedError
}
if err.Error() != expectedErr {
t.Errorf("%s: %v: expected error: %s, got %s", query, test.qargs, expectedErr, err)
}
continue
}
}
defer rows.Close()
if test.error != "" {
t.Errorf("expected error: %s: %v", query, test.qargs)