contexty/store_helpers_test.go at main · skosovsky/contexty · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package contexty_test

import (
	"context"

	"github.com/skosovsky/contexty"
)

func loadState(
	ctx context.Context,
	store contexty.ConversationStateStore,
	conversationID string,
) (contexty.ConversationState, error) {
	return store.LoadState(ctx, conversationID)
}

func updateSegment(
	ctx context.Context,
	store contexty.ConversationStateStore,
	conversationID string,
	expectedVersion int64,
	name contexty.SegmentName,
	msgs []contexty.Message,
) error {
	return store.ApplyDelta(ctx, conversationID, expectedVersion, contexty.ConversationDelta{
		Operation: contexty.DeltaReplaceSegment,
		Segment:   name,
		Messages:  msgs,
	})
}

func appendSegment(
	ctx context.Context,
	store contexty.ConversationStateStore,
	conversationID string,
	expectedVersion int64,
	name contexty.SegmentName,
	msgs ...contexty.Message,
) error {
	return store.ApplyDelta(ctx, conversationID, expectedVersion, contexty.ConversationDelta{
		Operation: contexty.DeltaAppendMessages,
		Segment:   name,
		Messages:  msgs,
	})
}