[extension/storage/dbstorage] Create component (#7061)

* add dbstorage extension

* lint

* fix import

* make extensions test pass

* goimports

* add CODEOWNERS

* address code review

* code review - add missing link

.github/CODEOWNERS

@@ -65,6 +65,7 @@ extension/observer/                                  @open-telemetry/collector-c
 extension/observer/ecstaskobserver/                  @open-telemetry/collector-contrib-approvers @rmfitzpatrick
 extension/observer/k8sobserver/                      @open-telemetry/collector-contrib-approvers @rmfitzpatrick @dmitryax
 extension/oidcauthextension/                         @open-telemetry/collector-contrib-approvers @jpkrohling
+extension/storage/dbstorage/                         @open-telemetry/collector-contrib-approvers @dmitryax @atoulme
 extension/storage/filestorage/                       @open-telemetry/collector-contrib-approvers @djaglowski
 
 internal/aws/                                        @open-telemetry/collector-contrib-approvers @anuraaga @mxiamxia

CHANGELOG.md

@@ -52,6 +52,7 @@
 - `ecs_task_observer`: Discover running containers in AWS ECS tasks (#6894)
 - `mongodbreceiver`: Establish codebase for MongoDB metrics receiver (#6972)
 - `couchbasereceiver`: Establish codebase for Couchbase metrics receiver (#7046)
+- `dbstorage`: New experimental dbstorage extension (#7061)
 
 ## 🧰 Bug fixes 🧰
 

cmd/configschema/go.mod

@@ -150,6 +150,14 @@ require (
 	github.com/influxdata/influxdb-observability/influx2otel v0.2.10 // indirect
 	github.com/influxdata/influxdb-observability/otel2influx v0.2.10 // indirect
 	github.com/influxdata/line-protocol/v2 v2.2.1 // indirect
+	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
+	github.com/jackc/pgconn v1.10.1 // indirect
+	github.com/jackc/pgio v1.0.0 // indirect
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgproto3/v2 v2.2.0 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20200714003250-2b9c44734f2b // indirect
+	github.com/jackc/pgtype v1.9.1 // indirect
+	github.com/jackc/pgx/v4 v4.14.1 // indirect
 	github.com/jaegertracing/jaeger v1.29.0 // indirect
 	github.com/jcmturner/aescts/v2 v2.0.0 // indirect
 	github.com/jcmturner/dnsutils/v2 v2.0.0 // indirect
@@ -173,6 +181,7 @@ require (
 	github.com/mailru/easyjson v0.7.7 // indirect
 	github.com/mattn/go-colorable v0.1.12 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect
+	github.com/mattn/go-sqlite3 v1.14.10 // indirect
 	github.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect
 	github.com/microsoft/ApplicationInsights-Go v0.4.4 // indirect
 	github.com/miekg/dns v1.1.43 // indirect

extension/storage/dbstorage/README.md

@@ -0,0 +1,38 @@
+# Database Storage
+
+> :construction: This extension is in alpha. Configuration and functionality are subject to change.
+
+The Database Storage extension can persist state to a relational database. 
+
+The extension requires read and write access to a database table.
+
+`driver`: the name of the database driver to use. By default, the storage client supports "sqlite3" and "pgx".
+
+Implementors can add additional driver support by importing SQL drivers into the program.
+See [Golang database/sql package documentation](https://pkg.go.dev/database/sql) for more information.
+
+`datasource`: the url of the database, in the format accepted by the driver.
+
+
+```
+extensions:
+  db_storage:
+    driver: "sqlite3"
+    datasource: "foo.db?_busy_timeout=10000&_journal=WAL&_sync=NORMAL"
+
+service:
+  extensions: [db_storage]
+  pipelines:
+    traces:
+      receivers: [nop]
+      processors: [nop]
+      exporters: [nop]
+
+# Data pipeline is required to load the config.
+receivers:
+  nop:
+processors:
+  nop:
+exporters:
+  nop:
+```

extension/storage/dbstorage/client.go

@@ -0,0 +1,130 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage // import "github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"fmt"
+
+	// Postgres driver
+	_ "github.com/jackc/pgx/v4/stdlib"
+	// SQLite driver
+	_ "github.com/mattn/go-sqlite3"
+	"go.opentelemetry.io/collector/extension/experimental/storage"
+)
+
+const (
+	createTable     = "create table if not exists %s (key text primary key, value blob)"
+	getQueryText    = "select value from %s where key=?"
+	setQueryText    = "insert into %s(key, value) values(?,?) on conflict(key) do update set value=?"
+	deleteQueryText = "delete from %s where key=?"
+)
+
+type dbStorageClient struct {
+	db          *sql.DB
+	getQuery    *sql.Stmt
+	setQuery    *sql.Stmt
+	deleteQuery *sql.Stmt
+}
+
+func newClient(ctx context.Context, db *sql.DB, tableName string) (*dbStorageClient, error) {
+	var err error
+	_, err = db.ExecContext(ctx, fmt.Sprintf(createTable, tableName))
+	if err != nil {
+		return nil, err
+	}
+
+	selectQuery, err := db.PrepareContext(ctx, fmt.Sprintf(getQueryText, tableName))
+	if err != nil {
+		return nil, err
+	}
+	setQuery, err := db.PrepareContext(ctx, fmt.Sprintf(setQueryText, tableName))
+	if err != nil {
+		return nil, err
+	}
+	deleteQuery, err := db.PrepareContext(ctx, fmt.Sprintf(deleteQueryText, tableName))
+	if err != nil {
+		return nil, err
+	}
+	return &dbStorageClient{db, selectQuery, setQuery, deleteQuery}, nil
+}
+
+// Get will retrieve data from storage that corresponds to the specified key
+func (c *dbStorageClient) Get(ctx context.Context, key string) ([]byte, error) {
+	rows, err := c.getQuery.QueryContext(ctx, key)
+	if err != nil {
+		return nil, err
+	}
+	if !rows.Next() {
+		return nil, nil
+	}
+	var result []byte
+	err = rows.Scan(&result)
+	if err != nil {
+		return result, err
+	}
+	err = rows.Close()
+	return result, err
+}
+
+// Set will store data. The data can be retrieved using the same key
+func (c *dbStorageClient) Set(ctx context.Context, key string, value []byte) error {
+	_, err := c.setQuery.ExecContext(ctx, key, value, value)
+	return err
+}
+
+// Delete will delete data associated with the specified key
+func (c *dbStorageClient) Delete(ctx context.Context, key string) error {
+	_, err := c.deleteQuery.ExecContext(ctx, key)
+	return err
+}
+
+// Batch executes the specified operations in order. Get operation results are updated in place
+func (c *dbStorageClient) Batch(ctx context.Context, ops ...storage.Operation) error {
+	var err error
+	for _, op := range ops {
+		switch op.Type {
+		case storage.Get:
+			op.Value, err = c.Get(ctx, op.Key)
+		case storage.Set:
+			err = c.Set(ctx, op.Key, op.Value)
+		case storage.Delete:
+			err = c.Delete(ctx, op.Key)
+		default:
+			return errors.New("wrong operation type")
+		}
+
+		if err != nil {
+			return err
+		}
+	}
+	return err
+}
+
+// Close will close the database
+func (c *dbStorageClient) Close(_ context.Context) error {
+	if err := c.setQuery.Close(); err != nil {
+		return err
+	}
+	if err := c.deleteQuery.Close(); err != nil {
+		return err
+	}
+	if err := c.getQuery.Close(); err != nil {
+		return err
+	}
+	return nil
+}

extension/storage/dbstorage/config.go

@@ -0,0 +1,39 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage // import "github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
+
+import (
+	"fmt"
+
+	"go.opentelemetry.io/collector/config"
+)
+
+// Config defines configuration for dbstorage extension.
+type Config struct {
+	config.ExtensionSettings `mapstructure:",squash"`
+	DriverName               string `mapstructure:"driver,omitempty"`
+	DataSource               string `mapstructure:"datasource,omitempty"`
+}
+
+func (cfg *Config) Validate() error {
+	if cfg.DataSource == "" {
+		return fmt.Errorf(fmt.Sprintf("missing datasource for %s", cfg.ID()))
+	}
+	if cfg.DriverName == "" {
+		return fmt.Errorf(fmt.Sprintf("missing driver name for %s", cfg.ID()))
+	}
+
+	return nil
+}

extension/storage/dbstorage/config_test.go

@@ -0,0 +1,55 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage // import "github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
+import (
+	"errors"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConfig_Validate(t *testing.T) {
+	tests := []struct {
+		name      string
+		config    Config
+		errWanted error
+	}{
+		{
+			"Missing driver name",
+			Config{DataSource: "foo"},
+			errors.New("missing driver name for /blah"),
+		},
+		{
+			"Missing datasource",
+			Config{DriverName: "foo"},
+			errors.New("missing datasource for /blah"),
+		},
+		{
+			"valid",
+			Config{DriverName: "foo", DataSource: "bar"},
+			nil,
+		},
+	}
+
+	for _, test := range tests {
+		test.config.SetIDName("blah")
+		err := test.config.Validate()
+		if test.errWanted == nil {
+			assert.NoError(t, err)
+		} else {
+			assert.Equal(t, test.errWanted, err)
+		}
+	}
+}

extension/storage/dbstorage/extension.go

@@ -0,0 +1,91 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage // import "github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"strings"
+
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/config"
+	"go.opentelemetry.io/collector/extension/experimental/storage"
+	"go.uber.org/zap"
+)
+
+type databaseStorage struct {
+	driverName     string
+	datasourceName string
+	logger         *zap.Logger
+	db             *sql.DB
+}
+
+// Ensure this storage extension implements the appropriate interface
+var _ storage.Extension = (*databaseStorage)(nil)
+
+func newDBStorage(logger *zap.Logger, config *Config) (component.Extension, error) {
+	return &databaseStorage{
+		driverName:     config.DriverName,
+		datasourceName: config.DataSource,
+		logger:         logger,
+	}, nil
+}
+
+// Start opens a connection to the database
+func (ds *databaseStorage) Start(context.Context, component.Host) error {
+	db, err := sql.Open(ds.driverName, ds.datasourceName)
+	if err != nil {
+		return err
+	}
+
+	if err := db.Ping(); err != nil {
+		return err
+	}
+	ds.db = db
+	return nil
+}
+
+// Shutdown closes the connection to the database
+func (ds *databaseStorage) Shutdown(context.Context) error {
+	return ds.db.Close()
+}
+
+// GetClient returns a storage client for an individual component
+func (ds *databaseStorage) GetClient(ctx context.Context, kind component.Kind, ent config.ComponentID, name string) (storage.Client, error) {
+	var fullName string
+	if name == "" {
+		fullName = fmt.Sprintf("%s_%s_%s", kindString(kind), ent.Type(), ent.Name())
+	} else {
+		fullName = fmt.Sprintf("%s_%s_%s_%s", kindString(kind), ent.Type(), ent.Name(), name)
+	}
+	fullName = strings.ReplaceAll(fullName, " ", "")
+	return newClient(ctx, ds.db, fullName)
+}
+
+func kindString(k component.Kind) string {
+	switch k {
+	case component.KindReceiver:
+		return "receiver"
+	case component.KindProcessor:
+		return "processor"
+	case component.KindExporter:
+		return "exporter"
+	case component.KindExtension:
+		return "extension"
+	default:
+		return "other" // not expected
+	}
+}

extension/storage/dbstorage/extension_test.go

@@ -0,0 +1,129 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage
+
+import (
+	"context"
+	"fmt"
+	"io/ioutil"
+	"sync"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/component/componenttest"
+	"go.opentelemetry.io/collector/config"
+	"go.opentelemetry.io/collector/extension/experimental/storage"
+)
+
+func TestExtensionIntegrity(t *testing.T) {
+	ctx := context.Background()
+	se := newTestExtension(t)
+	err := se.Start(context.Background(), componenttest.NewNopHost())
+	defer se.Shutdown(context.Background())
+	assert.NoError(t, err)
+
+	type mockComponent struct {
+		kind component.Kind
+		name config.ComponentID
+	}
+
+	components := []mockComponent{
+		{kind: component.KindReceiver, name: newTestEntity("receiver_one")},
+		{kind: component.KindReceiver, name: newTestEntity("receiver_two")},
+		{kind: component.KindProcessor, name: newTestEntity("processor_one")},
+		{kind: component.KindProcessor, name: newTestEntity("processor_two")},
+		{kind: component.KindExporter, name: newTestEntity("exporter_one")},
+		{kind: component.KindExporter, name: newTestEntity("exporter_two")},
+		{kind: component.KindExtension, name: newTestEntity("extension_one")},
+		{kind: component.KindExtension, name: newTestEntity("extension_two")},
+	}
+
+	// Make a client for each component
+	clients := make(map[config.ComponentID]storage.Client)
+	for _, c := range components {
+		client, err := se.GetClient(ctx, c.kind, c.name, "")
+		require.NoError(t, err)
+		clients[c.name] = client
+	}
+
+	thrashClient := func(wg *sync.WaitGroup, n config.ComponentID, c storage.Client) {
+		// keys and values
+		keys := []string{"a", "b", "c", "d", "e"}
+		myBytes := []byte(n.Name())
+
+		// Set my values
+		for i := 0; i < len(keys); i++ {
+			err := c.Set(ctx, keys[i], myBytes)
+			require.NoError(t, err)
+		}
+
+		// Repeatedly thrash client
+		for j := 0; j < 100; j++ {
+
+			// Make sure my values are still mine
+			for i := 0; i < len(keys); i++ {
+				v, err := c.Get(ctx, keys[i])
+				require.NoError(t, err)
+				require.Equal(t, myBytes, v)
+			}
+
+			// Delete my values
+			for i := 0; i < len(keys); i++ {
+				err := c.Delete(ctx, keys[i])
+				require.NoError(t, err)
+			}
+
+			// Reset my values
+			for i := 0; i < len(keys); i++ {
+				err := c.Set(ctx, keys[i], myBytes)
+				require.NoError(t, err)
+			}
+		}
+		c.Close(ctx)
+		wg.Done()
+	}
+
+	// Use clients concurrently
+	var wg sync.WaitGroup
+	for name, client := range clients {
+		wg.Add(1)
+		go thrashClient(&wg, name, client)
+	}
+	wg.Wait()
+}
+
+func newTestExtension(t *testing.T) storage.Extension {
+	tempDir, err := ioutil.TempDir("", "")
+	require.NoError(t, err)
+
+	f := NewFactory()
+	cfg := f.CreateDefaultConfig().(*Config)
+	cfg.DriverName = "sqlite3"
+	cfg.DataSource = fmt.Sprintf("file:%s/foo.db?_busy_timeout=10000&_journal=WAL&_sync=NORMAL", tempDir)
+
+	extension, err := f.CreateExtension(context.Background(), componenttest.NewNopExtensionCreateSettings(), cfg)
+	require.NoError(t, err)
+
+	se, ok := extension.(storage.Extension)
+	require.True(t, ok)
+
+	return se
+}
+
+func newTestEntity(name string) config.ComponentID {
+	return config.NewComponentIDWithName("nop", name)
+}

extension/storage/dbstorage/factory.go

@@ -0,0 +1,48 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package dbstorage // import "github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
+
+import (
+	"context"
+
+	"go.opentelemetry.io/collector/component"
+	"go.opentelemetry.io/collector/config"
+	"go.opentelemetry.io/collector/extension/extensionhelper"
+)
+
+// The value of extension "type" in configuration.
+const typeStr config.Type = "db_storage"
+
+// NewFactory creates a factory for DBStorage extension.
+func NewFactory() component.ExtensionFactory {
+	return extensionhelper.NewFactory(
+		typeStr,
+		createDefaultConfig,
+		createExtension)
+}
+
+func createDefaultConfig() config.Extension {
+	return &Config{
+		ExtensionSettings: config.NewExtensionSettings(config.NewComponentID(typeStr)),
+	}
+}
+
+func createExtension(
+	_ context.Context,
+	params component.ExtensionCreateSettings,
+	cfg config.Extension,
+) (component.Extension, error) {
+	return newDBStorage(params.Logger, cfg.(*Config))
+}

extension/storage/go.mod

@@ -15,9 +15,18 @@ require (
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/fsnotify/fsnotify v1.5.1 // indirect
 	github.com/gogo/protobuf v1.3.2 // indirect
+	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
+	github.com/jackc/pgconn v1.10.1 // indirect
+	github.com/jackc/pgio v1.0.0 // indirect
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgproto3/v2 v2.2.0 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20200714003250-2b9c44734f2b // indirect
+	github.com/jackc/pgtype v1.9.1 // indirect
+	github.com/jackc/pgx/v4 v4.14.1 // indirect
 	github.com/knadh/koanf v1.4.0 // indirect
 	github.com/kr/pretty v0.3.0 // indirect
 	github.com/magiconair/properties v1.8.5 // indirect
+	github.com/mattn/go-sqlite3 v1.14.10 // indirect
 	github.com/mitchellh/copystructure v1.2.0 // indirect
 	github.com/mitchellh/mapstructure v1.4.3 // indirect
 	github.com/mitchellh/reflectwalk v1.0.2 // indirect
@@ -31,7 +40,9 @@ require (
 	go.opentelemetry.io/otel/trace v1.3.0 // indirect
 	go.uber.org/atomic v1.9.0 // indirect
 	go.uber.org/multierr v1.7.0 // indirect
+	golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97 // indirect
 	golang.org/x/sys v0.0.0-20211205182925-97ca703d548d // indirect
+	golang.org/x/text v0.3.7 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b // indirect
 )

go.mod

@@ -257,6 +257,14 @@ require (
 	github.com/influxdata/influxdb-observability/influx2otel v0.2.10 // indirect
 	github.com/influxdata/influxdb-observability/otel2influx v0.2.10 // indirect
 	github.com/influxdata/line-protocol/v2 v2.2.1 // indirect
+	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
+	github.com/jackc/pgconn v1.10.1 // indirect
+	github.com/jackc/pgio v1.0.0 // indirect
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgproto3/v2 v2.2.0 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20200714003250-2b9c44734f2b // indirect
+	github.com/jackc/pgtype v1.9.1 // indirect
+	github.com/jackc/pgx/v4 v4.14.1 // indirect
 	github.com/jaegertracing/jaeger v1.29.0 // indirect
 	github.com/jcmturner/aescts/v2 v2.0.0 // indirect
 	github.com/jcmturner/dnsutils/v2 v2.0.0 // indirect
@@ -280,6 +288,7 @@ require (
 	github.com/mailru/easyjson v0.7.7 // indirect
 	github.com/mattn/go-colorable v0.1.12 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect
+	github.com/mattn/go-sqlite3 v1.14.10 // indirect
 	github.com/matttproud/golang_protobuf_extensions v1.0.2-0.20181231171920-c182affec369 // indirect
 	github.com/microsoft/ApplicationInsights-Go v0.4.4 // indirect
 	github.com/miekg/dns v1.1.43 // indirect

internal/components/components.go

@@ -75,6 +75,7 @@ import (
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/observer/k8sobserver"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/oidcauthextension"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/pprofextension"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/filestorage"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/attributesprocessor"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/processor/cumulativetodeltaprocessor"
@@ -143,6 +144,7 @@ func Components() (component.Factories, error) {
 		awsproxy.NewFactory(),
 		ballastextension.NewFactory(),
 		bearertokenauthextension.NewFactory(),
+		dbstorage.NewFactory(),
 		ecstaskobserver.NewFactory(),
 		filestorage.NewFactory(),
 		fluentbitextension.NewFactory(),

internal/components/extensions_test.go

@@ -36,6 +36,7 @@ import (
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/observer/ecstaskobserver"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/observer/hostobserver"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/pprofextension"
+	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/dbstorage"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/extension/storage/filestorage"
 	"github.com/open-telemetry/opentelemetry-collector-contrib/internal/coreinternal/testutil"
 )
@@ -152,6 +153,16 @@ func TestDefaultExtensions(t *testing.T) {
 			extension:     "oidc",
 			skipLifecycle: true, // Requires a running OIDC server in order to complete life cycle testing
 		},
+		{
+			extension: "db_storage",
+			getConfigFn: func() config.Extension {
+				cfg := extFactories["db_storage"].CreateDefaultConfig().(*dbstorage.Config)
+				cfg.DriverName = "sqlite3"
+				tempFolder := testutil.NewTemporaryDirectory(t)
+				cfg.DataSource = tempFolder + "/foo.db"
+				return cfg
+			},
+		},
 		{
 			extension: "file_storage",
 			getConfigFn: func() config.Extension {