From 42377fcb4c45905f8960573bd1edf7908cfec410 Mon Sep 17 00:00:00 2001 From: SandeepDev1 Date: Tue, 30 May 2023 18:43:27 +0530 Subject: [PATCH] added timeout flag for the cassandra server connection timeout --- cmd/schemagen/schemagen.go | 4 ++++ go.sum | 5 ----- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/cmd/schemagen/schemagen.go b/cmd/schemagen/schemagen.go index 9cff543..72ba81e 100644 --- a/cmd/schemagen/schemagen.go +++ b/cmd/schemagen/schemagen.go @@ -12,6 +12,7 @@ import ( "os" "path" "strings" + "time" "github.com/gocql/gocql" "github.com/scylladb/gocqlx/v2" @@ -26,6 +27,7 @@ var ( flagOutput = cmd.String("output", "models", "the name of the folder to output to") flagUser = cmd.String("user", "", "user for password authentication") flagPassword = cmd.String("password", "", "password for password authentication") + flagTimeout = cmd.Int("timeout", 5, "connection timeout for the connection to the database ( in seconds )") ) var ( @@ -119,6 +121,8 @@ func createSession() (gocqlx.Session, error) { Password: *flagPassword, } } + + cluster.ConnectTimeout = time.Second * time.Duration(*flagTimeout) return gocqlx.WrapSession(cluster.CreateSession()) } diff --git a/go.sum b/go.sum index 00894b5..13ad649 100644 --- a/go.sum +++ b/go.sum @@ -3,13 +3,8 @@ github.com/bitly/go-hostpool v0.0.0-20171023180738-a3a6125de932/go.mod h1:NOuUCS github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869 h1:DDGfHa7BWjL4YnC6+E63dPcxHo2sUxDIu8g3QgEJdRY= github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869/go.mod h1:Ekp36dRnpXw/yCqJaO+ZrUyxD+3VXMFFr56k5XYrpB4= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= -github.com/gocql/gocql v0.0.0-20200131111108-92af2e088537 h1:NaMut1fdw76YYX/TPinSAbai4DShF5tPort3bHpET6g= -github.com/gocql/gocql v0.0.0-20200131111108-92af2e088537/go.mod h1:DL0ekTmBSTdlNF25Orwt/JMzqIq3EJ4MVa/J/uK64OY= github.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1 h1:px9qUCy/RNJNsfCam4m2IxWGxNuimkrioEF0vrrbPsg= github.com/gocql/gocql v0.0.0-20211015133455-b225f9b53fa1/go.mod h1:3gM2c4D3AnkISwBxGnMMsS8Oy4y2lhbPRsH4xnJrHG8= -github.com/golang/snappy v0.0.0-20170215233205-553a64147049/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= -github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4= -github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= github.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= github.com/golang/snappy v0.0.4 h1:yAGX7huGHXlcLOEtBnF4w7FQwA26wojNCwOYAEhLjQM= github.com/golang/snappy v0.0.4/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=