feat: implement refresh method for repos (#237)

core/codegen/base.builder.go

@@ -300,6 +300,11 @@ func (b *build) buildBaseFile(node *field.NodeTable) error {
 			jen.Id("user").Op("*").Add(b.input.SourceQual(node.NameGo())),
 		).Error(),
 
+		jen.Id("Refresh").Call(
+			jen.Id("ctx").Qual("context", "Context"),
+			jen.Id("user").Op("*").Add(b.input.SourceQual(node.NameGo())),
+		).Error(),
+
 		jen.Id("Relate").Call().Op("*").Qual(b.subPkg(def.PkgRelate), node.NameGo()),
 	)
 
@@ -498,6 +503,35 @@ func (b *build) buildBaseFile(node *field.NodeTable) error {
 			),
 		)
 
+	f.Line()
+	f.Func().
+		Params(jen.Id("r").Op("*").Id(node.NameGoLower())).
+		Id("Refresh").
+		Params(
+			jen.Id("ctx").Qual("context", "Context"),
+			jen.Id(node.NameGoLower()).Op("*").Add(b.input.SourceQual(node.NameGo())),
+		).
+		Error().
+		Block(
+			jen.If(jen.Id(node.NameGoLower()).Op("==").Nil()).
+				Block(
+					jen.Return(jen.Qual("errors", "New").Call(jen.Lit("the passed node must not be nil"))),
+				),
+
+			jen.If(jen.Id(node.NameGoLower()).Dot("ID").Call().Op("==").Lit("")).
+				Block(
+					jen.Return(jen.Qual("errors", "New").Call(jen.Lit("cannot refresh "+node.NameGo()+" without existing record ID"))),
+				),
+
+			jen.Return(
+				jen.Id("r").Dot("refresh").Call(
+					jen.Id("ctx"),
+					jen.Id(node.NameGoLower()).Dot("ID").Call(),
+					jen.Id(node.NameGoLower()),
+				),
+			),
+		)
+
 	f.Line()
 	f.Func().
 		Params(jen.Id("r").Op("*").Id(node.NameGoLower())).

core/embed/som/som.node.go

@@ -4,6 +4,7 @@ package som
 
 import (
 	"context"
+	"errors"
 	"fmt"
 )
 
@@ -86,3 +87,18 @@ func (r *repo[N, C]) delete(ctx context.Context, id string, node *N) error {
 	}
 	return nil
 }
+
+func (r *repo[N, C]) refresh(ctx context.Context, id string, node *N) error {
+	read, exists, err := r.read(ctx, id)
+	if err != nil {
+		return fmt.Errorf("failed to read node: %w", err)
+	}
+
+	if !exists {
+		return errors.New("given node does not exist")
+	}
+
+	*node = *read
+
+	return nil
+}

tests/basic/som_base_test.go

@@ -463,3 +463,28 @@ func TestAsync(t *testing.T) {
 	assert.NilError(t, <-resCh.Err())
 	assert.Equal(t, 2, <-resCh.Val())
 }
+
+func TestRefresh(t *testing.T) {
+	ctx := context.Background()
+
+	client, cleanup := prepareDatabase(ctx, t)
+	defer cleanup()
+
+	allFieldTypes := &model.AllFieldTypes{
+		String: "some value",
+	}
+
+	err := client.AllFieldTypesRepo().Create(ctx, allFieldTypes)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	allFieldTypes.String = "some other value"
+
+	err = client.AllFieldTypesRepo().Refresh(ctx, allFieldTypes)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	assert.Equal(t, "some value", allFieldTypes.String)
+}