influxdb/remotes/service.go

package remotes

import (
	"context"
	"database/sql"
	"errors"

	sq "github.com/Masterminds/squirrel"
	"github.com/influxdata/influxdb/v2"
	"github.com/influxdata/influxdb/v2/kit/platform"
	ierrors "github.com/influxdata/influxdb/v2/kit/platform/errors"
	"github.com/influxdata/influxdb/v2/snowflake"
	"github.com/influxdata/influxdb/v2/sqlite"
)

var (
	errRemoteNotFound = &ierrors.Error{
		Code: ierrors.ENotFound,
		Msg:  "remote connection not found",
	}
)

func NewService(store *sqlite.SqlStore) *service {
	return &service{
		store:       store,
		idGenerator: snowflake.NewIDGenerator(),
	}
}

type service struct {
	store       *sqlite.SqlStore
	idGenerator platform.IDGenerator
}

func (s service) ListRemoteConnections(ctx context.Context, filter influxdb.RemoteConnectionListFilter) (*influxdb.RemoteConnections, error) {
	q := sq.Select("id", "org_id", "name", "description", "remote_url", "remote_org_id", "allow_insecure_tls").
		From("remotes").
		Where(sq.Eq{"org_id": filter.OrgID})

	if filter.Name != nil {
		q = q.Where(sq.Eq{"name": *filter.Name})
	}
	if filter.RemoteURL != nil {
		q = q.Where(sq.Eq{"remote_url": *filter.RemoteURL})
	}

	query, args, err := q.ToSql()
	if err != nil {
		return nil, err
	}

	var rcs influxdb.RemoteConnections
	if err := s.store.DB.SelectContext(ctx, &rcs.Remotes, query, args...); err != nil {
		return nil, err
	}

	return &rcs, nil
}

func (s service) CreateRemoteConnection(ctx context.Context, request influxdb.CreateRemoteConnectionRequest) (*influxdb.RemoteConnection, error) {
	s.store.Mu.Lock()
	defer s.store.Mu.Unlock()

	q := sq.Insert("remotes").
		SetMap(sq.Eq{
			"id":                 s.idGenerator.ID(),
			"org_id":             request.OrgID,
			"name":               request.Name,
			"description":        request.Description,
			"remote_url":         request.RemoteURL,
			"remote_api_token":   request.RemoteToken,
			"remote_org_id":      request.RemoteOrgID,
			"allow_insecure_tls": request.AllowInsecureTLS,
			"created_at":         "datetime('now')",
			"updated_at":         "datetime('now')",
		}).
		Suffix("RETURNING id, org_id, name, description, remote_url, remote_org_id, allow_insecure_tls")

	query, args, err := q.ToSql()
	if err != nil {
		return nil, err
	}

	var rc influxdb.RemoteConnection
	if err := s.store.DB.GetContext(ctx, &rc, query, args...); err != nil {
		return nil, err
	}
	return &rc, nil
}

func (s service) GetRemoteConnection(ctx context.Context, id platform.ID) (*influxdb.RemoteConnection, error) {
	q := sq.Select("id", "org_id", "name", "description", "remote_url", "remote_org_id", "allow_insecure_tls").
		From("remotes").
		Where(sq.Eq{"id": id})

	query, args, err := q.ToSql()
	if err != nil {
		return nil, err
	}

	var rc influxdb.RemoteConnection
	if err := s.store.DB.GetContext(ctx, &rc, query, args...); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return nil, errRemoteNotFound
		}
		return nil, err
	}
	return &rc, nil
}

func (s service) UpdateRemoteConnection(ctx context.Context, id platform.ID, request influxdb.UpdateRemoteConnectionRequest) (*influxdb.RemoteConnection, error) {
	s.store.Mu.Lock()
	defer s.store.Mu.Unlock()

	updates := sq.Eq{"updated_at": sq.Expr("datetime('now')")}
	if request.AllowInsecureTLS != nil {
		updates["allow_insecure_tls"] = *request.AllowInsecureTLS
	}
	if request.RemoteOrgID != nil {
		updates["remote_org_id"] = *request.RemoteOrgID
	}
	if request.RemoteURL != nil {
		updates["remote_url"] = *request.RemoteURL
	}
	if request.RemoteToken != nil {
		updates["remote_api_token"] = *request.RemoteToken
	}
	if request.Name != nil {
		updates["name"] = *request.Name
	}
	if request.Description != nil {
		updates["description"] = *request.Description
	}

	q := sq.Update("remotes").SetMap(updates).Where(sq.Eq{"id": id}).
		Suffix("RETURNING id, org_id, name, description, remote_url, remote_org_id, allow_insecure_tls")

	query, args, err := q.ToSql()
	if err != nil {
		return nil, err
	}

	var rc influxdb.RemoteConnection
	if err := s.store.DB.GetContext(ctx, &rc, query, args...); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return nil, errRemoteNotFound
		}
		return nil, err
	}
	return &rc, nil
}

func (s service) DeleteRemoteConnection(ctx context.Context, id platform.ID) error {
	s.store.Mu.Lock()
	defer s.store.Mu.Unlock()

	q := sq.Delete("remotes").Where(sq.Eq{"id": id}).Suffix("RETURNING id")
	query, args, err := q.ToSql()
	if err != nil {
		return err
	}

	var d platform.ID
	if err := s.store.DB.GetContext(ctx, &d, query, args...); err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return errRemoteNotFound
		}
		return err
	}
	return nil
}