mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package system
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
|
|
"github.com/hanzoai/builder"
|
|
)
|
|
|
|
// AppState represents a state record in database
|
|
// if one day we would run this as a cluster,
|
|
// we can introduce a new field `Scope` here to store different states for different nodes
|
|
type AppState struct {
|
|
ID string `xorm:"pk varchar(200)"`
|
|
Revision int64
|
|
Content string `xorm:"LONGTEXT"`
|
|
}
|
|
|
|
func init() {
|
|
db.RegisterModel(new(AppState))
|
|
}
|
|
|
|
// SaveAppStateContent saves the app state item to database
|
|
func SaveAppStateContent(ctx context.Context, key, content string) error {
|
|
return db.WithTx(ctx, func(ctx context.Context) error {
|
|
eng := db.GetEngine(ctx)
|
|
// try to update existing row
|
|
res, err := eng.Exec("UPDATE app_state SET revision=revision+1, content=? WHERE id=?", content, key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
rows, _ := res.RowsAffected()
|
|
if rows != 0 {
|
|
// the existing row is updated, so we can return
|
|
return nil
|
|
}
|
|
// if no existing row, insert a new row
|
|
_, err = eng.Insert(&AppState{ID: key, Content: content})
|
|
return err
|
|
})
|
|
}
|
|
|
|
// GetAppStateContent gets an app state from database
|
|
func GetAppStateContent(ctx context.Context, key string) (content string, err error) {
|
|
appState, has, err := db.Get[AppState](ctx, builder.Eq{"id": key})
|
|
if err != nil {
|
|
return "", err
|
|
} else if !has {
|
|
return "", nil
|
|
}
|
|
return appState.Content, nil
|
|
}
|