luxolus/open-vault
2
0
Fork 0
Code Issues 1 Pull requests Releases Activity

Add lock and close check on cassandra as well

Browse source
This commit is contained in:
Jeff Mitchell 2017-12-19 10:26:46 -05:00
parent 3cf61a5d8f
commit 15df4d1f36
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
plugins/database/cassandra/connection_producer.go
View file

@ -116,12 +116,15 @@ func (c *cassandraConnectionProducer) Initialize(ctx context.Context, conf map[s
}
func (c *cassandraConnectionProducer) Connection(_ context.Context) (interface{}, error) {
c.Lock()
defer c.Unlock()
if !c.Initialized {
return nil, connutil.ErrNotInitialized
}
// If we already have a DB, return it
if c.session != nil {
if c.session != nil && !c.session.Closed() {
return c.session, nil
}