Add missing metric drain config attrs

aptible · Dec 24, 2024 · 45cd2ec · 45cd2ec
1 parent cc04d04
commit 45cd2ec
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/aptible/metric_drain.go b/aptible/metric_drain.go
@@ -22,6 +22,9 @@ type MetricDrain struct {
 	APIKey     string
 	SeriesURL  strfmt.URI
 	AccountID  int64
+	AuthToken  string
+	Bucket     string
+	Org        string
 }
 
 type MetricDrainCreateAttrs struct {
@@ -34,6 +37,9 @@ type MetricDrainCreateAttrs struct {
 	Database     string
 	APIKey       string
 	SeriesURL    strfmt.URI
+	AuthToken    string
+	Bucket       string
+	Org          string
 }
 
 func (c *Client) CreateMetricDrain(handle string, accountID int64, attrs *MetricDrainCreateAttrs) (*MetricDrain, error) {
@@ -56,6 +62,9 @@ func (c *Client) CreateMetricDrain(handle string, accountID int64, attrs *Metric
 			Database:  attrs.Database,
 			APIKey:    attrs.APIKey,
 			SeriesURL: attrs.SeriesURL,
+			AuthToken: attrs.AuthToken,
+			Bucket:    attrs.Bucket,
+			Org:       attrs.Org,
 		}
 	}
 
@@ -130,6 +139,9 @@ func (c *Client) GetMetricDrain(metricDrainID int64) (*MetricDrain, error) {
 		metricDrain.Database = response.Payload.DrainConfiguration.Database
 		metricDrain.APIKey = response.Payload.DrainConfiguration.APIKey
 		metricDrain.SeriesURL = response.Payload.DrainConfiguration.SeriesURL
+		metricDrain.AuthToken = response.Payload.DrainConfiguration.AuthToken
+		metricDrain.Bucket = response.Payload.DrainConfiguration.Bucket
+		metricDrain.Org = response.Payload.DrainConfiguration.Org
 	}
 
 	if response.Payload.Links.Database != nil {

diff --git a/models/accountsaccount_idmetric_drains_drain_configuration.go b/models/accountsaccount_idmetric_drains_drain_configuration.go
diff --git a/models/inline_response20029_embedded_drain_configuration.go b/models/inline_response20029_embedded_drain_configuration.go