mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b9f9cdf918
Re-sync the entire SigNoz-derived Go tree to a clean fork of latest SigNoz main (upstream 3e6339019), eliminating the version-skew that broke `go build ./...` (mixed-era packages, ~14 imported-but-absent packages, partial signoz->o11y / clickhouse->datastore rebrands, self-import cycle, duplicate package decls). - pkg/ + cmd/ wholesale replaced with latest SigNoz; imports rewritten github.com/SigNoz/signoz/ -> github.com/hanzoai/o11y/. - Adopt SigNoz's CURRENT structure (resourcefilter->telemetryresourcefilter, integrationtypes->cloudintegrationtypes, roletypes/grammar/times folded in upstream); old paths dropped. - Fork deps repathed to hanzoai + bumped: signoz-otel-collector v0.144.6, clickhouse-go-mock v0.14.1, govaluate v0.1.0; expr-lang/expr => hanzoai/expr v1.17.8 (replace). go.mod reconciled from latest SigNoz; go.sum re-tidied. - ee/ + cmd/enterprise stripped — o11y stays MIT-only; community edition imports no ee packages. - Preserved o11y additions: module path github.com/hanzoai/o11y, mount.go (cloud.Register/zip mount adapter, HIP-0106), NOTICE + LICENSE (SigNoz §4c attribution retained; upstream pkg files are headerless per SigNoz convention). - Reverted o11y cosmetic renames (o11y*/datastore*) and experimental originals (authz/iamauthz, billing, zapreceiver, zapmetricreceiver) to SigNoz canonical to kill the skew; snapshots retained in git history for re-layering. go build ./... = exit 0. go vet clean except benign "unreachable code" in generated ANTLR parser files (inherited from upstream).
112 lines
3.3 KiB
Go
112 lines
3.3 KiB
Go
package integrations
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/hanzoai/o11y/pkg/modules/dashboard"
|
|
"github.com/hanzoai/o11y/pkg/query-service/agentConf"
|
|
"github.com/hanzoai/o11y/pkg/query-service/model"
|
|
"github.com/hanzoai/o11y/pkg/sqlstore"
|
|
"github.com/hanzoai/o11y/pkg/types/pipelinetypes"
|
|
"github.com/hanzoai/o11y/pkg/valuer"
|
|
)
|
|
|
|
type Controller struct {
|
|
mgr *Manager
|
|
}
|
|
|
|
func NewController(sqlStore sqlstore.SQLStore, dashboardModule dashboard.Module) (*Controller, error) {
|
|
mgr, err := NewManager(sqlStore, dashboardModule)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("couldn't create integrations manager: %w", err)
|
|
}
|
|
|
|
return &Controller{
|
|
mgr: mgr,
|
|
}, nil
|
|
}
|
|
|
|
type IntegrationsListResponse struct {
|
|
Integrations []IntegrationsListItem `json:"integrations"`
|
|
|
|
// Pagination details to come later
|
|
}
|
|
|
|
func (c *Controller) ListIntegrations(ctx context.Context, orgId string, params map[string]string) (*IntegrationsListResponse, *model.ApiError) {
|
|
var filters *IntegrationsFilter
|
|
if isInstalledFilter, exists := params["is_installed"]; exists {
|
|
isInstalled := !(isInstalledFilter == "false")
|
|
filters = &IntegrationsFilter{
|
|
IsInstalled: &isInstalled,
|
|
}
|
|
}
|
|
|
|
integrations, apiErr := c.mgr.ListIntegrations(ctx, orgId, filters)
|
|
if apiErr != nil {
|
|
return nil, apiErr
|
|
}
|
|
|
|
return &IntegrationsListResponse{
|
|
Integrations: integrations,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Controller) GetIntegration(ctx context.Context, orgId string, integrationId string) (*Integration, *model.ApiError) {
|
|
return c.mgr.GetIntegration(ctx, orgId, integrationId)
|
|
}
|
|
|
|
func (c *Controller) IsIntegrationInstalled(ctx context.Context, orgId string, integrationId string) (bool, *model.ApiError) {
|
|
installation, apiErr := c.mgr.getInstalledIntegration(ctx, orgId, integrationId)
|
|
if apiErr != nil {
|
|
return false, apiErr
|
|
}
|
|
isInstalled := installation != nil
|
|
return isInstalled, nil
|
|
}
|
|
|
|
func (c *Controller) GetIntegrationConnectionTests(ctx context.Context, orgId string, integrationId string) (*IntegrationConnectionTests, *model.ApiError) {
|
|
return c.mgr.GetIntegrationConnectionTests(ctx, orgId, integrationId)
|
|
}
|
|
|
|
type InstallIntegrationRequest struct {
|
|
IntegrationId string `json:"integration_id"`
|
|
Config map[string]interface{} `json:"config"`
|
|
}
|
|
|
|
func (c *Controller) Install(ctx context.Context, orgId string, req *InstallIntegrationRequest, createdBy string, creator valuer.UUID) (*IntegrationsListItem, *model.ApiError) {
|
|
res, apiErr := c.mgr.InstallIntegration(
|
|
ctx, orgId, req.IntegrationId, req.Config, createdBy, creator,
|
|
)
|
|
if apiErr != nil {
|
|
return nil, apiErr
|
|
}
|
|
agentConf.NotifyConfigUpdate(ctx)
|
|
return res, nil
|
|
}
|
|
|
|
type UninstallIntegrationRequest struct {
|
|
IntegrationId string `json:"integration_id"`
|
|
}
|
|
|
|
func (c *Controller) Uninstall(ctx context.Context, orgId string, req *UninstallIntegrationRequest) *model.ApiError {
|
|
if len(req.IntegrationId) < 1 {
|
|
return model.BadRequest(fmt.Errorf(
|
|
"integration_id is required",
|
|
))
|
|
}
|
|
|
|
apiErr := c.mgr.UninstallIntegration(
|
|
ctx, orgId, req.IntegrationId,
|
|
)
|
|
if apiErr != nil {
|
|
return apiErr
|
|
}
|
|
agentConf.NotifyConfigUpdate(ctx)
|
|
return nil
|
|
}
|
|
|
|
func (c *Controller) GetPipelinesForInstalledIntegrations(ctx context.Context, orgId string) ([]pipelinetypes.GettablePipeline, error) {
|
|
return c.mgr.GetPipelinesForInstalledIntegrations(ctx, orgId)
|
|
}
|