WIP - Add SetDefaultSchemaVersion

client/db.go

@@ -111,6 +111,8 @@ type Store interface {
 	// [FieldKindStringToEnumMapping].
 	PatchSchema(context.Context, string) error
 
+	SetDefaultSchemaVersion(context.Context, string) error
+
 	// SetMigration sets the migration for the given source-destination schema version IDs. Is equivilent to
 	// calling `LensRegistry().SetMigration(ctx, cfg)`.
 	//

db/collection.go

@@ -300,7 +300,7 @@ func (db *db) updateCollection(
 		return nil, err
 	}
 
-	err = db.setDefaultSchemaVersion(ctx, txn, desc.Name, desc.Schema.SchemaID, schemaVersionID)
+	err = db.setDefaultSchemaVersionExplicit(ctx, txn, desc.Name, desc.Schema.SchemaID, schemaVersionID)
 	if err != nil {
 		return nil, err
 	}
@@ -585,6 +585,20 @@ func validateUpdateCollectionIndexes(
 }
 
 func (db *db) setDefaultSchemaVersion(
+	ctx context.Context,
+	txn datastore.Txn,
+	schemaVersionID string,
+) error {
+	col, err := db.getCollectionByVersionID(ctx, txn, schemaVersionID)
+	if err != nil {
+		return err
+	}
+
+	desc := col.Description()
+	return db.setDefaultSchemaVersionExplicit(ctx, txn, desc.Name, desc.Schema.SchemaID, schemaVersionID)
+}
+
+func (db *db) setDefaultSchemaVersionExplicit(
 	ctx context.Context,
 	txn datastore.Txn,
 	collectionName string,

db/txn_db.go

@@ -280,6 +280,25 @@ func (db *explicitTxnDB) PatchSchema(ctx context.Context, patchString string) er
 	return db.patchSchema(ctx, db.txn, patchString)
 }
 
+func (db *implicitTxnDB) SetDefaultSchemaVersion(ctx context.Context, schemaVersionID string) error {
+	txn, err := db.NewTxn(ctx, false)
+	if err != nil {
+		return err
+	}
+	defer txn.Discard(ctx)
+
+	err = db.setDefaultSchemaVersion(ctx, txn, schemaVersionID)
+	if err != nil {
+		return err
+	}
+
+	return txn.Commit(ctx)
+}
+
+func (db *explicitTxnDB) SetDefaultSchemaVersion(ctx context.Context, schemaVersionID string) error {
+	return db.setDefaultSchemaVersion(ctx, db.txn, schemaVersionID)
+}
+
 func (db *implicitTxnDB) SetMigration(ctx context.Context, cfg client.LensConfig) error {
 	txn, err := db.NewTxn(ctx, false)
 	if err != nil {

http/client.go

@@ -223,6 +223,18 @@ func (c *Client) PatchSchema(ctx context.Context, patch string) error {
 	return err
 }
 
+func (c *Client) SetDefaultSchemaVersion(ctx context.Context, schemaVersionID string) error {
+	methodURL := c.http.baseURL.JoinPath("schema")
+	panic("todo")
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPatch, methodURL.String(), strings.NewReader(schemaVersionID))
+	if err != nil {
+		return err
+	}
+	_, err = c.http.request(req)
+	return err
+}
+
 func (c *Client) SetMigration(ctx context.Context, config client.LensConfig) error {
 	return c.LensRegistry().SetMigration(ctx, config)
 }

http/wrapper.go

@@ -90,6 +90,10 @@ func (w *Wrapper) PatchSchema(ctx context.Context, patch string) error {
 	return w.client.PatchSchema(ctx, patch)
 }
 
+func (w *Wrapper) SetDefaultSchemaVersion(ctx context.Context, schemaVersionID string) error {
+	return w.client.SetDefaultSchemaVersion(ctx, schemaVersionID)
+}
+
 func (w *Wrapper) SetMigration(ctx context.Context, config client.LensConfig) error {
 	return w.client.SetMigration(ctx, config)
 }

tests/integration/test_case.go

@@ -85,6 +85,16 @@ type SchemaPatch struct {
 	ExpectedError string
 }
 
+type SetDefaultSchemaVersion struct {
+	// NodeID may hold the ID (index) of a node to set the default schema version on.
+	//
+	// If a value is not provided the default will be set on all nodes.
+	NodeID immutable.Option[int]
+
+	SchemaVersionID string
+	ExpectedError   string
+}
+
 // CreateDoc will attempt to create the given document in the given collection
 // using the set [MutationType].
 type CreateDoc struct {

tests/integration/utils2.go

@@ -418,6 +418,9 @@ func executeTestCase(
 		case SchemaPatch:
 			patchSchema(s, action)
 
+		case SetDefaultSchemaVersion:
+			setDefaultSchemaVersion(s, action)
+
 		case ConfigureMigration:
 			configureMigration(s, action)
 
@@ -1070,6 +1073,21 @@ func patchSchema(
 	refreshIndexes(s)
 }
 
+func setDefaultSchemaVersion(
+	s *state,
+	action SetDefaultSchemaVersion,
+) {
+	for _, node := range getNodes(action.NodeID, s.nodes) {
+		err := node.DB.SetDefaultSchemaVersion(s.ctx, action.SchemaVersionID)
+		expectedErrorRaised := AssertError(s.t, s.testCase.Description, err, action.ExpectedError)
+
+		assertExpectedErrorRaised(s.t, s.testCase.Description, action.ExpectedError, expectedErrorRaised)
+	}
+
+	refreshCollections(s)
+	refreshIndexes(s)
+}
+
 // createDoc creates a document using the chosen [mutationType] and caches it in the
 // test state object.
 func createDoc(