From addb461cb63a6e54934760161cd8a31c49bec4c3 Mon Sep 17 00:00:00 2001
From: Volker Schukai <volker.schukai@schukai.com>
Date: Mon, 23 Oct 2023 22:02:04 +0200
Subject: [PATCH] feat: job serialize and unserialize #1
---
.gitlab-ci.yml | 4 --
devenv.nix | 4 --
errors.go | 1 +
import.go | 105 +++++++++++++++++++++++---------------
job-handler.go | 30 +++++++++++
job.go | 51 ++++++++++++++++--
job_test.go | 48 +++++++++++++++++
manager.go | 14 +++++
manager_test.go | 49 ++++++++++++------
runnable-counter.go | 10 +++-
runnable-dummy.go | 4 ++
runnable-fileoperation.go | 20 ++++++++
runnable-gorm.go | 23 +++++++++
runnable-http.go | 29 +++++++++++
runnable-mail.go | 59 +++++++++++++++++++++
runnable-sftp.go | 22 ++++++++
runnable-shell.go | 11 ++++
state.go | 55 ++++++++++++++++++++
state_test.go | 40 +++++++++++++++
19 files changed, 512 insertions(+), 67 deletions(-)
create mode 100644 job-handler.go
create mode 100644 state.go
create mode 100644 state_test.go
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 6225215..ca7c8db 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -64,7 +64,3 @@ deploy:
paths:
- /nix/store
-
- artifacts:
- paths:
- - dist
diff --git a/devenv.nix b/devenv.nix
index 4d0389c..78bebff 100644
--- a/devenv.nix
+++ b/devenv.nix
@@ -363,10 +363,6 @@ deploy:
paths:
- /nix/store
-
- artifacts:
- paths:
- - dist
EOF
diff --git a/errors.go b/errors.go
index ec304a7..8493dd8 100644
--- a/errors.go
+++ b/errors.go
@@ -29,4 +29,5 @@ var (
ErrUnsupportedFileOption = fmt.Errorf("unsupported file option")
ErrUnsupportedCredentialType = fmt.Errorf("unsupported credential type")
ErrUnsupportedTransferDirection = fmt.Errorf("unsupported transfer direction")
+ ErrInvalidData = fmt.Errorf("invalid data")
)
diff --git a/import.go b/import.go
index 371f122..987ec12 100644
--- a/import.go
+++ b/import.go
@@ -60,62 +60,87 @@ func ReadJsonFile(filePath string) ([]JobImport, error) {
return jobs, nil
}
+func CreateGenericJobFromImport[T any](jobImport JobImport, runner Runnable[T]) GenericJob {
+ return &Job[T]{
+ id: JobID(jobImport.ID),
+ priority: Priority(jobImport.Priority),
+ timeout: jobImport.Timeout,
+ maxRetries: jobImport.MaxRetries,
+ RetryDelay: jobImport.RetryDelay,
+ runner: runner,
+ }
+}
+
func CreateJobAndSchedulerFromImport(jobImport JobImport) (GenericJob, Scheduler, error) {
var job GenericJob
switch jobImport.Runnable.Type {
- case "Shell":
+ case "Dummy":
- runner := &ShellRunnable{
- ScriptPath: jobImport.Runnable.Data["ScriptPath"].(string),
+ runner, err := NewDummyRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
}
- job = GenericJob(&Job[ShellResult]{
- id: JobID(jobImport.ID),
- priority: Priority(jobImport.Priority),
- timout: jobImport.Timeout,
- maxRetries: jobImport.MaxRetries,
- RetryDelay: jobImport.RetryDelay,
- runner: runner,
- })
+ job = CreateGenericJobFromImport[DummyResult](jobImport, runner)
case "Counter":
- runner := &CounterRunnable{
- Count: jobImport.Runnable.Data["Count"].(int),
+
+ runner, err := NewCounterRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
}
- job = GenericJob(&Job[CounterResult]{
- id: JobID(jobImport.ID),
- priority: Priority(jobImport.Priority),
- timout: jobImport.Timeout,
- maxRetries: jobImport.MaxRetries,
- RetryDelay: jobImport.RetryDelay,
- runner: runner,
- })
- case "HTTP":
- runner := &HTTPRunnable{
- URL: jobImport.Runnable.Data["URL"].(string),
+ job = CreateGenericJobFromImport[CounterResult](jobImport, runner)
+
+ case "FileOperation":
+ runner, err := NewFileOperationRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
}
- job = GenericJob(&Job[HTTPResult]{id: JobID(jobImport.ID),
- priority: Priority(jobImport.Priority),
- timout: jobImport.Timeout,
- maxRetries: jobImport.MaxRetries,
- RetryDelay: jobImport.RetryDelay,
- runner: runner,
- })
+
+ job = CreateGenericJobFromImport[FileOperationResult](jobImport, runner)
case "DB":
- runner := &DBRunnable{
- Query: jobImport.Runnable.Data["Query"].(string),
+ runner, err := NewDBRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
}
- job = GenericJob(&Job[DBResult]{id: JobID(jobImport.ID),
- priority: Priority(jobImport.Priority),
- timout: jobImport.Timeout,
- maxRetries: jobImport.MaxRetries,
- RetryDelay: jobImport.RetryDelay,
- runner: runner,
- })
+
+ job = CreateGenericJobFromImport[DBResult](jobImport, runner)
+
+ case "HTTP":
+ runner, err := NewHTTPRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
+ }
+
+ job = CreateGenericJobFromImport[HTTPResult](jobImport, runner)
+
+ case "Mail":
+ runner, err := NewMailRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
+ }
+
+ job = CreateGenericJobFromImport[MailResult](jobImport, runner)
+
+ case "SFTP":
+ runner, err := NewSFTPRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
+ }
+
+ job = CreateGenericJobFromImport[SFTPResult](jobImport, runner)
+
+ case "Shell":
+ runner, err := NewShellRunnableFromMap(jobImport.Runnable.Data)
+ if err != nil {
+ return nil, nil, err
+ }
+
+ job = CreateGenericJobFromImport[ShellResult](jobImport, runner)
default:
return nil, nil, ErrUnknownRunnableType
diff --git a/job-handler.go b/job-handler.go
new file mode 100644
index 0000000..8308aab
--- /dev/null
+++ b/job-handler.go
@@ -0,0 +1,30 @@
+package jobqueue
+
+type CompletedJobHandler interface {
+ HandleCompletedJob(job GenericJob) error
+}
+
+type DatabaseArchiver struct {
+ // ...
+}
+
+func (d *DatabaseArchiver) HandleCompletedJob(job GenericJob) error {
+
+ return nil
+}
+
+type FileLogger struct {
+}
+
+func (f *FileLogger) HandleCompletedJob(job GenericJob) error {
+
+ return nil
+}
+
+type MetricsPublisher struct {
+}
+
+func (m *MetricsPublisher) HandleCompletedJob(job GenericJob) error {
+
+ return nil
+}
diff --git a/job.go b/job.go
index 1dcdb80..86f6def 100644
--- a/job.go
+++ b/job.go
@@ -37,13 +37,15 @@ type GenericJob interface {
GetRetryDelay() time.Duration
GetTimeout() time.Duration
+
+ Archive() error
}
type Job[T any] struct {
id JobID
priority Priority
- timout time.Duration
+ timeout time.Duration
maxRetries uint
RetryDelay time.Duration
@@ -57,6 +59,17 @@ type Job[T any] struct {
logs []JobLog
}
+type JobSerializedState struct {
+ ID JobID `json:"id"`
+ Priority Priority `json:"priority"`
+ Timeout time.Duration `json:"timeout" `
+ MaxRetries uint `json:"maxRetries" `
+ RetryDelay time.Duration `json:"retryDelay"`
+ Dependencies []JobID `json:"dependencies" `
+ Stats JobStats `json:"stats"`
+ Logs []JobLog `json:"logs"`
+}
+
// NewJob creates a new job with the given id and runner
func NewJob[T any](id JobID, runner Runnable[T]) *Job[T] {
return &Job[T]{
@@ -66,6 +79,38 @@ func NewJob[T any](id JobID, runner Runnable[T]) *Job[T] {
}
}
+func (j *Job[T]) SerializeState() JobSerializedState {
+ j.mu.Lock()
+ defer j.mu.Unlock()
+ return JobSerializedState{
+ ID: j.id,
+ Priority: j.priority,
+ Timeout: j.timeout,
+ MaxRetries: j.maxRetries,
+ RetryDelay: j.RetryDelay,
+ Dependencies: j.dependencies,
+ Stats: j.stats,
+ Logs: j.logs,
+ }
+}
+
+func (j *Job[T]) UnserializeState(serializedState JobSerializedState) {
+ j.mu.Lock()
+ defer j.mu.Unlock()
+ j.id = serializedState.ID
+ j.priority = serializedState.Priority
+ j.timeout = serializedState.Timeout
+ j.maxRetries = serializedState.MaxRetries
+ j.RetryDelay = serializedState.RetryDelay
+ j.dependencies = serializedState.Dependencies
+ j.stats = serializedState.Stats
+ j.logs = serializedState.Logs
+}
+
+func (j *Job[T]) Archive() error {
+ return nil
+}
+
// Execute executes the job
func (j *Job[T]) Execute(ctx context.Context) (RunGenericResult, error) {
startTime := time.Now()
@@ -150,7 +195,7 @@ func (j *Job[T]) SetTimeout(timeout time.Duration) *Job[T] {
j.mu.Lock()
defer j.mu.Unlock()
- j.timout = timeout
+ j.timeout = timeout
return j
}
@@ -158,7 +203,7 @@ func (j *Job[T]) GetTimeout() time.Duration {
j.mu.Lock()
defer j.mu.Unlock()
- return j.timout
+ return j.timeout
}
// SetMaxRetries sets the max retries of the job
diff --git a/job_test.go b/job_test.go
index 3400cef..b95aac4 100644
--- a/job_test.go
+++ b/job_test.go
@@ -142,3 +142,51 @@ echo "Hello World"
assert.Equal(t, 2, stats.SuccessCount)
assert.Equal(t, 0, stats.ErrorCount)
}
+
+func TestSerializeState(t *testing.T) {
+ // Initialize job
+ job := NewJob[DummyResult]("testJob", &DummyRunnable{})
+
+ job.SetPriority(PriorityHigh)
+ job.SetTimeout(5 * time.Minute)
+ job.SetMaxRetries(3)
+ job.SetRetryDelay(1 * time.Minute)
+ job.SetDependencies([]JobID{"dep1", "dep2"})
+
+ // Serialize state
+ serializedState := job.SerializeState()
+
+ // Verify serialized state
+ assert.Equal(t, JobID("testJob"), serializedState.ID)
+ assert.Equal(t, PriorityHigh, serializedState.Priority)
+ assert.Equal(t, 5*time.Minute, serializedState.Timeout)
+ assert.Equal(t, uint(3), serializedState.MaxRetries)
+ assert.Equal(t, 1*time.Minute, serializedState.RetryDelay)
+ assert.ElementsMatch(t, []JobID{"dep1", "dep2"}, serializedState.Dependencies)
+}
+
+func TestUnserializeState(t *testing.T) {
+ // Initialize serialized state
+ serializedState := JobSerializedState{
+ ID: "testJob",
+ Priority: PriorityHigh,
+ Timeout: 5 * time.Minute,
+ MaxRetries: 3,
+ RetryDelay: 1 * time.Minute,
+ Dependencies: []JobID{"dep1", "dep2"},
+ }
+
+ // Initialize job
+ job := NewJob[DummyResult]("initialJob", &DummyRunnable{})
+
+ // Unserialize state
+ job.UnserializeState(serializedState)
+
+ // Verify job properties
+ assert.Equal(t, JobID("testJob"), job.GetID())
+ assert.Equal(t, PriorityHigh, job.GetPriority())
+ assert.Equal(t, 5*time.Minute, job.GetTimeout())
+ assert.Equal(t, uint(3), job.GetMaxRetries())
+ assert.Equal(t, 1*time.Minute, job.GetRetryDelay())
+ assert.ElementsMatch(t, []JobID{"dep1", "dep2"}, job.GetDependencies())
+}
diff --git a/manager.go b/manager.go
index 4b60620..5e7fee8 100644
--- a/manager.go
+++ b/manager.go
@@ -22,6 +22,8 @@ type Manager struct {
jobEventCh chan interface{}
+ stateManager StateManager
+
mu sync.Mutex
}
@@ -134,6 +136,12 @@ func (m *Manager) Start() error {
return ErrManagerAlreadyRunning
}
+ if m.stateManager != nil {
+ if err := m.stateManager.LoadState(); err != nil {
+ return err
+ }
+ }
+
if len(m.workerMap) == 0 {
return ErrNoWorkers
}
@@ -206,6 +214,12 @@ func (m *Manager) Stop() error {
wrappedErr = fmt.Errorf("%w\n%s", wrappedErr, err.Error())
}
+ if m.stateManager != nil {
+ if err := m.stateManager.LoadState(); err != nil {
+ return err
+ }
+ }
+
return wrappedErr
}
diff --git a/manager_test.go b/manager_test.go
index fa12cf1..e991a2d 100644
--- a/manager_test.go
+++ b/manager_test.go
@@ -106,60 +106,79 @@ func TestManager_AddWorker(t *testing.T) {
}
func TestManager_RemoveWorker(t *testing.T) {
+ var err error
m := NewManager()
w := &MockWorker{id: "worker1", status: WorkerStatusStopped}
- m.AddWorker(w)
-
- err := m.RemoveWorker(w)
+ err = m.AddWorker(w)
+ assert.Nil(t, err)
+ err = m.RemoveWorker(w)
assert.Nil(t, err)
assert.Equal(t, int(ManagerStateStopped), int(m.state))
}
func TestManager_Start(t *testing.T) {
+ var err error
m := NewManager()
w := &MockWorker{id: "worker1", status: WorkerStatusStopped}
- m.AddWorker(w)
+ err = m.AddWorker(w)
+ assert.Nil(t, err)
- err := m.Start()
+ err = m.Start()
+ assert.Nil(t, err)
assert.Nil(t, err)
assert.Equal(t, int(ManagerStateRunning), int(m.state))
}
func TestManager_Stop(t *testing.T) {
+ var err error
m := NewManager()
w := &MockWorker{id: "worker1", status: WorkerStatusStopped}
- m.AddWorker(w)
- m.Start()
+ err = m.AddWorker(w)
+ assert.Nil(t, err)
+
+ err = m.Start()
+ assert.Nil(t, err)
- err := m.Stop()
+ err = m.Stop()
assert.Nil(t, err)
assert.Equal(t, int(ManagerStateStopped), int(m.state))
}
func TestManager_ScheduleJob(t *testing.T) {
+ var err error
+
m := NewManager()
w := &MockWorker{id: "worker1", status: WorkerStatusStopped}
- m.AddWorker(w)
- m.Start()
+ err = m.AddWorker(w)
+ assert.Nil(t, err)
+
+ err = m.Start()
+ assert.Nil(t, err)
job := &MockGenericJob{ID: "job1"}
scheduler := InstantScheduler{}
- err := m.ScheduleJob(job, &scheduler)
+ err = m.ScheduleJob(job, &scheduler)
assert.Nil(t, err)
}
func TestManager_CancelJob(t *testing.T) {
+ var err error
+
m := NewManager()
w := &MockWorker{id: "worker1", status: WorkerStatusStopped}
- m.AddWorker(w)
- m.Start()
+ err = m.AddWorker(w)
+ assert.Nil(t, err)
+
+ err = m.Start()
+ assert.Nil(t, err)
job := &MockGenericJob{ID: "job1"}
scheduler := InstantScheduler{}
- m.ScheduleJob(job, &scheduler)
+ err = m.ScheduleJob(job, &scheduler)
+ assert.Nil(t, err)
- err := m.CancelJob("job1")
+ err = m.CancelJob("job1")
assert.Nil(t, err)
}
diff --git a/runnable-counter.go b/runnable-counter.go
index 780768c..d019ce0 100644
--- a/runnable-counter.go
+++ b/runnable-counter.go
@@ -4,6 +4,14 @@ import (
"sync"
)
+func NewCounterRunnableFromMap(data map[string]any) (*CounterRunnable, error) {
+ count, ok := data["Count"].(int)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+ return &CounterRunnable{Count: count}, nil
+}
+
// CounterResult is a result of a counter
type CounterResult struct {
Count int
@@ -11,7 +19,7 @@ type CounterResult struct {
// CounterRunnable is a runnable that counts
type CounterRunnable struct {
- Count int
+ Count int `json:"count" yaml:"count"`
mu sync.Mutex
}
diff --git a/runnable-dummy.go b/runnable-dummy.go
index 6e86990..795fa76 100644
--- a/runnable-dummy.go
+++ b/runnable-dummy.go
@@ -1,5 +1,9 @@
package jobqueue
+func NewDummyRunnableFromMap(data map[string]any) (*DummyRunnable, error) {
+ return &DummyRunnable{}, nil
+}
+
// DummyResult is a dummy result
type DummyResult struct {
}
diff --git a/runnable-fileoperation.go b/runnable-fileoperation.go
index 1aa1e0c..020bf73 100644
--- a/runnable-fileoperation.go
+++ b/runnable-fileoperation.go
@@ -4,6 +4,26 @@ import (
"os"
)
+func NewFileOperationRunnableFromMap(data map[string]interface{}) (*FileOperationRunnable, error) {
+ operation, ok := data["Operation"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ filePath, ok := data["FilePath"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ content, _ := data["Content"].(string) // Optional, so no error check
+
+ return &FileOperationRunnable{
+ Operation: operation,
+ FilePath: filePath,
+ Content: content,
+ }, nil
+}
+
type FileOperationResult struct {
Success bool
Content string // Optional, je nach Operation
diff --git a/runnable-gorm.go b/runnable-gorm.go
index 53cb581..0c66d2c 100644
--- a/runnable-gorm.go
+++ b/runnable-gorm.go
@@ -5,6 +5,29 @@ import (
"gorm.io/gorm"
)
+func NewDBRunnableFromMap(data map[string]interface{}) (*DBRunnable, error) {
+ t, ok := data["Type"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ dsn, ok := data["DSN"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ query, ok := data["Query"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ return &DBRunnable{
+ Type: t,
+ DSN: dsn,
+ Query: query,
+ }, nil
+}
+
// DBResult is a result of a db query
type DBResult struct {
RowsAffected int
diff --git a/runnable-http.go b/runnable-http.go
index b1f498a..07412a7 100644
--- a/runnable-http.go
+++ b/runnable-http.go
@@ -6,6 +6,35 @@ import (
"net/http"
)
+func NewHTTPRunnableFromMap(data map[string]interface{}) (*HTTPRunnable, error) {
+ url, ok := data["URL"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ method, ok := data["Method"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ header, ok := data["Header"].(map[string]string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ body, ok := data["Body"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ return &HTTPRunnable{
+ URL: url,
+ Method: method,
+ Header: header,
+ Body: body,
+ }, nil
+}
+
// HTTPResult is a result of a http request
type HTTPResult struct {
StatusCode int
diff --git a/runnable-mail.go b/runnable-mail.go
index 9927096..a6b7f5e 100644
--- a/runnable-mail.go
+++ b/runnable-mail.go
@@ -4,6 +4,65 @@ import (
"net/smtp"
)
+func NewMailRunnableFromMap(data map[string]interface{}) (*MailRunnable, error) {
+ to, ok := data["To"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ from, ok := data["From"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ subject, ok := data["Subject"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ body, ok := data["Body"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ server, ok := data["Server"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ port, ok := data["Port"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ username, ok := data["Username"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ password, ok := data["Password"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ headers, ok := data["Headers"].(map[string]string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ return &MailRunnable{
+ To: to,
+ From: from,
+ Subject: subject,
+ Body: body,
+ Server: server,
+ Port: port,
+ Username: username,
+ Password: password,
+ Headers: headers,
+ }, nil
+}
+
// MailResult is a result of a email
type MailResult struct {
Sent bool
diff --git a/runnable-sftp.go b/runnable-sftp.go
index c57400c..1fbdb7c 100644
--- a/runnable-sftp.go
+++ b/runnable-sftp.go
@@ -8,6 +8,28 @@ import (
"os"
)
+func NewSFTPRunnableFromMap(data map[string]interface{}) (*SFTPRunnable, error) {
+ // Your map to struct conversion logic here
+ // e.g.,
+
+ host, ok := data["Host"].(string)
+ if !ok {
+ return nil, fmt.Errorf("invalid Host")
+ }
+ //... (other fields)
+
+ transferDirection, ok := data["TransferDirection"].(string)
+ if !ok {
+ return nil, fmt.Errorf("invalid TransferDirection")
+ }
+
+ return &SFTPRunnable{
+ Host: host,
+ //... (other fields)
+ TransferDirection: Direction(transferDirection),
+ }, nil
+}
+
// SFTPResult is a result of a sftp
type SFTPResult struct {
FilesCopied []string
diff --git a/runnable-shell.go b/runnable-shell.go
index af60f25..5a6da9d 100644
--- a/runnable-shell.go
+++ b/runnable-shell.go
@@ -5,6 +5,17 @@ import (
"strings"
)
+func NewShellRunnableFromMap(data map[string]interface{}) (*ShellRunnable, error) {
+ scriptPath, ok := data["ScriptPath"].(string)
+ if !ok {
+ return nil, ErrInvalidData
+ }
+
+ return &ShellRunnable{
+ ScriptPath: scriptPath,
+ }, nil
+}
+
// ShellResult is a result of a shell script
type ShellResult struct {
Output string
diff --git a/state.go b/state.go
new file mode 100644
index 0000000..a9323ac
--- /dev/null
+++ b/state.go
@@ -0,0 +1,55 @@
+package jobqueue
+
+import (
+ "encoding/json"
+ "os"
+)
+
+// State repräsentiert den Zustand, den wir speichern wollen
+type State struct {
+ // Jobs enthält alle Jobs, die wir speichern wollen
+ Jobs []JobSerializedState `json:"jobs"`
+}
+
+type StateManager interface {
+ LoadState() error
+ SaveState() error
+}
+
+// FileStateManager implementiert StateManager
+type FileStateManager struct {
+ filePath string
+ state *State
+}
+
+// LoadState lädt den Zustand aus der Datei
+func (f *FileStateManager) LoadState() error {
+ file, err := os.Open(f.filePath)
+ if err != nil {
+ return err
+ }
+ defer file.Close()
+
+ decoder := json.NewDecoder(file)
+ if err := decoder.Decode(&f.state); err != nil {
+ return err
+ }
+
+ return nil
+}
+
+// SaveState speichert den Zustand in der Datei
+func (f *FileStateManager) SaveState() error {
+ file, err := os.Create(f.filePath)
+ if err != nil {
+ return err
+ }
+ defer file.Close()
+
+ encoder := json.NewEncoder(file)
+ if err := encoder.Encode(f.state); err != nil {
+ return err
+ }
+
+ return nil
+}
diff --git a/state_test.go b/state_test.go
new file mode 100644
index 0000000..8a14b16
--- /dev/null
+++ b/state_test.go
@@ -0,0 +1,40 @@
+package jobqueue
+
+import (
+ "os"
+ "path"
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+)
+
+func TestFileStateManager(t *testing.T) {
+
+ tmpDir := t.TempDir()
+
+ tempFile := path.Join(tmpDir, "state.json")
+ defer os.Remove(tempFile)
+
+ manager := &FileStateManager{
+ filePath: tempFile,
+ state: &State{
+ Jobs: []JobSerializedState{
+ {ID: "1"},
+ {ID: "2"},
+ },
+ },
+ }
+
+ err := manager.SaveState()
+ assert.NoError(t, err)
+
+ manager2 := &FileStateManager{
+ filePath: tempFile,
+ state: &State{},
+ }
+
+ err = manager2.LoadState()
+ assert.NoError(t, err)
+
+ assert.Equal(t, manager.state, manager2.state)
+}
--
GitLab