-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add appuio_control_organization_info metric
- Loading branch information
1 parent
5c79acf
commit 1f808f7
Showing
3 changed files
with
110 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,54 @@ | ||
package controllers | ||
|
||
import ( | ||
"context" | ||
|
||
"github.com/prometheus/client_golang/prometheus" | ||
"sigs.k8s.io/controller-runtime/pkg/client" | ||
|
||
orgv1 "github.com/appuio/control-api/apis/organization/v1" | ||
) | ||
|
||
//+kubebuilder:rbac:groups="rbac.appuio.io",resources=organizations,verbs=get;list;watch | ||
//+kubebuilder:rbac:groups="organization.appuio.io",resources=organizations,verbs=get;list;watch | ||
|
||
var orgInfoMetricDesc = prometheus.NewDesc( | ||
"appuio_control_organization_info", | ||
"Information about APPUiO Cloud organizations", | ||
[]string{"organization", "billing_entity", "sales_order"}, | ||
nil, | ||
) | ||
|
||
// OrgInfoMetric is a Prometheus collector that exposes the link between an organization and a billing entity. | ||
type OrgInfoMetric struct { | ||
client.Client | ||
} | ||
|
||
var _ prometheus.Collector = &OrgInfoMetric{} | ||
|
||
// Describe implements prometheus.Collector. | ||
// Sends the static description of the metric to the provided channel. | ||
func (o *OrgInfoMetric) Describe(ch chan<- *prometheus.Desc) { | ||
ch <- orgInfoMetricDesc | ||
} | ||
|
||
// Collect implements prometheus.Collector. | ||
// Sends a metric for each organization and its billing entity to the provided channel. | ||
func (o *OrgInfoMetric) Collect(ch chan<- prometheus.Metric) { | ||
orgs := &orgv1.OrganizationList{} | ||
if err := o.List(context.Background(), orgs); err != nil { | ||
ch <- prometheus.NewInvalidMetric(desc, err) | ||
return | ||
} | ||
|
||
for _, org := range orgs.Items { | ||
ch <- prometheus.MustNewConstMetric( | ||
orgInfoMetricDesc, | ||
prometheus.GaugeValue, | ||
1, | ||
org.Name, | ||
org.Spec.BillingEntityRef, | ||
org.Status.SaleOrderName, | ||
) | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,52 @@ | ||
package controllers_test | ||
|
||
import ( | ||
"strings" | ||
"testing" | ||
|
||
"github.com/prometheus/client_golang/prometheus/testutil" | ||
"github.com/stretchr/testify/require" | ||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
|
||
orgv1 "github.com/appuio/control-api/apis/organization/v1" | ||
"github.com/appuio/control-api/controllers" | ||
) | ||
|
||
func TestOrgInfoMetric(t *testing.T) { | ||
c := prepareTest(t, &orgv1.Organization{ | ||
ObjectMeta: metav1.ObjectMeta{ | ||
Name: "test-org", | ||
}, | ||
Spec: orgv1.OrganizationSpec{ | ||
BillingEntityRef: "test-billing-entity", | ||
}, | ||
}, &orgv1.Organization{ | ||
ObjectMeta: metav1.ObjectMeta{ | ||
Name: "blub-org", | ||
}, | ||
Spec: orgv1.OrganizationSpec{ | ||
BillingEntityRef: "be-1734", | ||
}, | ||
}, &orgv1.Organization{ | ||
ObjectMeta: metav1.ObjectMeta{ | ||
Name: "foo-org", | ||
}, | ||
Spec: orgv1.OrganizationSpec{ | ||
BillingEntityRef: "be-234", | ||
}, | ||
Status: orgv1.OrganizationStatus{ | ||
SaleOrderName: "SO9999", | ||
}, | ||
}) | ||
|
||
require.NoError(t, | ||
testutil.CollectAndCompare(&controllers.OrgInfoMetric{c}, strings.NewReader(` | ||
# HELP appuio_control_organization_info Information about APPUiO Cloud organizations | ||
# TYPE appuio_control_organization_info gauge | ||
appuio_control_organization_info{billing_entity="be-1734",organization="blub-org",sales_order=""} 1 | ||
appuio_control_organization_info{billing_entity="be-234",organization="foo-org",sales_order="SO9999"} 1 | ||
appuio_control_organization_info{billing_entity="test-billing-entity",organization="test-org",sales_order=""} 1 | ||
`), | ||
"appuio_control_organization_info"), | ||
) | ||
} |