gopkg/pgx
2
0
Fork 0
Code Activity

Add the drop replication slot functionality

Browse Source
This commit is contained in:
Kris Wehner
2017-01-09 14:19:08 -08:00
parent c88c110169
commit af01afca00
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
replication.go
+1 -1
View File
@@ -362,7 +362,7 @@ func (rc *ReplicationConn) CreateReplicationSlot(slotName, outputPlugin string)
} }
// Drop the replication slot for the given name // Drop the replication slot for the given name
func (rc *ReplicationConn) DropReplicationSlot(slotName, outputPlugin string) (err error) { func (rc *ReplicationConn) DropReplicationSlot(slotName string) (err error) {
_, err = rc.c.Exec(fmt.Sprintf("DROP_REPLICATION_SLOT %s", slotName)) _, err = rc.c.Exec(fmt.Sprintf("DROP_REPLICATION_SLOT %s", slotName))
return return
} }
replication_test.go
+4 -6
View File
@@ -173,6 +173,10 @@ func TestSimpleReplicationConnection(t *testing.T) {
t.Errorf("Unexpected write position %d", status.WalWritePosition) t.Errorf("Unexpected write position %d", status.WalWritePosition)
} }
err = replicationConn.DropReplicationSlot("pgx_test")
if err != nil {
t.Fatalf("Failed to drop replication slot %v", err)
}
err = replicationConn.Close() err = replicationConn.Close()
if err != nil { if err != nil {
t.Fatalf("Replication connection close failed: %v", err) t.Fatalf("Replication connection close failed: %v", err)
@@ -183,10 +187,4 @@ func TestSimpleReplicationConnection(t *testing.T) {
if integerRestartLsn != maxWal { if integerRestartLsn != maxWal {
t.Fatalf("Wal offset update failed, expected %s found %s", pgx.FormatLSN(maxWal), restartLsn) t.Fatalf("Wal offset update failed, expected %s found %s", pgx.FormatLSN(maxWal), restartLsn)
} }
_, err = conn.Exec("select pg_drop_replication_slot($1)", "pgx_test")
if err != nil {
t.Fatalf("Failed to drop replication slot: %v", err)
}
} }