Commitfa799a6

committed

comments

1 parentd2cd8d7 commitfa799a6Copy full SHA for fa799a6

File tree

+10

-6

lines changed

+10

-6

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,7 @@ func (d *awsIamRdsDriver) Open(name string) (driver.Conn, error) {`
`69`	`69`	`returnconn,nil`
`70`	`70`	`}`
`71`	`71`
	`72`	`+// Connector returns a driver.Connector that fetches a new authentication token for each connection.`
`72`	`73`	`func (d*awsIamRdsDriver)Connector(namestring) (driver.Connector,error) {`
`73`	`74`	`connector:=&connector{`
`74`	`75`	`url:name,`
`@@ -122,7 +123,7 @@ func (c *connector) Connect(ctx context.Context) (driver.Conn, error) {`
`122`	`123`	`returnnc.Connect(ctx)`
`123`	`124`	`}`
`124`	`125`
`125`		`-func (c*connector)Driver() driver.Driver {`
	`126`	`+func (*connector)Driver() driver.Driver {`
`126`	`127`	`return&pq.Driver{}`
`127`	`128`	`}`
`128`	`129`

Lines changed: 2 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -7,10 +7,12 @@ import (`
`7`	`7`	`"github.com/lib/pq"`
`8`	`8`	`)`
`9`	`9`
	`10`	`+// ConnectorCreator can create a driver.Connector.`
`10`	`11`	`typeConnectorCreatorinterface {`
`11`	`12`	`Connector(namestring) (driver.Connector,error)`
`12`	`13`	`}`
`13`	`14`
	`15`	`+// DialerConnector can create a driver.Connector and set a pq.Dialer.`
`14`	`16`	`typeDialerConnectorinterface {`
`15`	`17`	`Connect(context.Context) (driver.Conn,error)`
`16`	`18`	`Dialer(dialer pq.Dialer)`

Lines changed: 6 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -441,15 +441,16 @@ func (p *PGPubsub) startListener(ctx context.Context, connectURL string) error {`
`441`	`441`
`442`	`442`	`// Create a custom connector if the database driver supports it.`
`443`	`443`	`connectorCreator,ok:=p.db.Driver().(database.ConnectorCreator)`
`444`		`-if!ok {`
`445`		`-connector,err=pq.NewConnector(connectURL)`
	`444`	`+ifok {`
	`445`	`+connector,err=connectorCreator.Connector(connectURL)`
`446`	`446`	`iferr!=nil {`
`447`		`-returnxerrors.Errorf("createpq connector: %w",err)`
	`447`	`+returnxerrors.Errorf("createcustom connector: %w",err)`
`448`	`448`	`}`
`449`	`449`	`}else {`
`450`		`-connector,err=connectorCreator.Connector(connectURL)`
	`450`	`+// use the default pq connector otherwise`
	`451`	`+connector,err=pq.NewConnector(connectURL)`
`451`	`452`	`iferr!=nil {`
`452`		`-returnxerrors.Errorf("createcustom connector: %w",err)`
	`453`	`+returnxerrors.Errorf("createpq connector: %w",err)`
`453`	`454`	`}`
`454`	`455`	`}`
`455`	`456`

Comments

(0)