feat(pgdriver): allow setting Network in config

uptrace · Oct 10, 2021 · b24b5d8 · b24b5d8
1 parent e621318
commit b24b5d8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/driver/pgdriver/config.go b/driver/pgdriver/config.go
@@ -71,6 +71,15 @@ func newDefaultConfig() *Config {
 
 type DriverOption func(cfg *Config)
 
+func WithNetwork(network string) DriverOption {
+	if network == "" {
+		panic("network is empty")
+	}
+	return func(cfg *Config) {
+		cfg.Network = network
+	}
+}
+
 func WithAddr(addr string) DriverOption {
 	if addr == "" {
 		panic("addr is empty")