feat: introducing streams management pages #166

2025-05-11 02:15:48 +02:00 · 2023-12-15 21:43:13 +08:00 · 2023-12-15 21:43:13 +08:00 · 2649b710bb
commit 2649b710bb
parent 9c16b55dde
32 changed files with 3170 additions and 606 deletions
--- a/query/gen.go
+++ b/query/gen.go
@ -26,6 +26,7 @@ var (
 	Environment   *environment
 	Notification  *notification
 	Site          *site
+	Stream        *stream
 )

 func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
@ -39,6 +40,7 @@ func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
 	Environment = &Q.Environment
 	Notification = &Q.Notification
 	Site = &Q.Site
+	Stream = &Q.Stream
 }

 func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
@ -53,6 +55,7 @@ func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
 		Environment:   newEnvironment(db, opts...),
 		Notification:  newNotification(db, opts...),
 		Site:          newSite(db, opts...),
+		Stream:        newStream(db, opts...),
 	}
 }

@ -68,6 +71,7 @@ type Query struct {
 	Environment   environment
 	Notification  notification
 	Site          site
+	Stream        stream
 }

 func (q *Query) Available() bool { return q.db != nil }
@ -84,6 +88,7 @@ func (q *Query) clone(db *gorm.DB) *Query {
 		Environment:   q.Environment.clone(db),
 		Notification:  q.Notification.clone(db),
 		Site:          q.Site.clone(db),
+		Stream:        q.Stream.clone(db),
 	}
 }

@ -107,6 +112,7 @@ func (q *Query) ReplaceDB(db *gorm.DB) *Query {
 		Environment:   q.Environment.replaceDB(db),
 		Notification:  q.Notification.replaceDB(db),
 		Site:          q.Site.replaceDB(db),
+		Stream:        q.Stream.replaceDB(db),
 	}
 }

@ -120,6 +126,7 @@ type queryCtx struct {
 	Environment   *environmentDo
 	Notification  *notificationDo
 	Site          *siteDo
+	Stream        *streamDo
 }

 func (q *Query) WithContext(ctx context.Context) *queryCtx {
@ -133,6 +140,7 @@ func (q *Query) WithContext(ctx context.Context) *queryCtx {
 		Environment:   q.Environment.WithContext(ctx),
 		Notification:  q.Notification.WithContext(ctx),
 		Site:          q.Site.WithContext(ctx),
+		Stream:        q.Stream.WithContext(ctx),
 	}
 }

--- a/query/streams.gen.go
+++ b/query/streams.gen.go
@ -0,0 +1,374 @@
+// Code generated by gorm.io/gen. DO NOT EDIT.
+// Code generated by gorm.io/gen. DO NOT EDIT.
+// Code generated by gorm.io/gen. DO NOT EDIT.
+
+package query
+
+import (
+	"context"
+	"strings"
+
+	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
+	"gorm.io/gorm/schema"
+
+	"gorm.io/gen"
+	"gorm.io/gen/field"
+
+	"gorm.io/plugin/dbresolver"
+
+	"github.com/0xJacky/Nginx-UI/model"
+)
+
+func newStream(db *gorm.DB, opts ...gen.DOOption) stream {
+	_stream := stream{}
+
+	_stream.streamDo.UseDB(db, opts...)
+	_stream.streamDo.UseModel(&model.Stream{})
+
+	tableName := _stream.streamDo.TableName()
+	_stream.ALL = field.NewAsterisk(tableName)
+	_stream.ID = field.NewInt(tableName, "id")
+	_stream.CreatedAt = field.NewTime(tableName, "created_at")
+	_stream.UpdatedAt = field.NewTime(tableName, "updated_at")
+	_stream.DeletedAt = field.NewField(tableName, "deleted_at")
+	_stream.Path = field.NewString(tableName, "path")
+	_stream.Advanced = field.NewBool(tableName, "advanced")
+	_stream.ChatGPTMessages = field.NewField(tableName, "chat_gpt_messages")
+
+	_stream.fillFieldMap()
+
+	return _stream
+}
+
+type stream struct {
+	streamDo
+
+	ALL             field.Asterisk
+	ID              field.Int
+	CreatedAt       field.Time
+	UpdatedAt       field.Time
+	DeletedAt       field.Field
+	Path            field.String
+	Advanced        field.Bool
+	ChatGPTMessages field.Field
+
+	fieldMap map[string]field.Expr
+}
+
+func (s stream) Table(newTableName string) *stream {
+	s.streamDo.UseTable(newTableName)
+	return s.updateTableName(newTableName)
+}
+
+func (s stream) As(alias string) *stream {
+	s.streamDo.DO = *(s.streamDo.As(alias).(*gen.DO))
+	return s.updateTableName(alias)
+}
+
+func (s *stream) updateTableName(table string) *stream {
+	s.ALL = field.NewAsterisk(table)
+	s.ID = field.NewInt(table, "id")
+	s.CreatedAt = field.NewTime(table, "created_at")
+	s.UpdatedAt = field.NewTime(table, "updated_at")
+	s.DeletedAt = field.NewField(table, "deleted_at")
+	s.Path = field.NewString(table, "path")
+	s.Advanced = field.NewBool(table, "advanced")
+	s.ChatGPTMessages = field.NewField(table, "chat_gpt_messages")
+
+	s.fillFieldMap()
+
+	return s
+}
+
+func (s *stream) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
+	_f, ok := s.fieldMap[fieldName]
+	if !ok || _f == nil {
+		return nil, false
+	}
+	_oe, ok := _f.(field.OrderExpr)
+	return _oe, ok
+}
+
+func (s *stream) fillFieldMap() {
+	s.fieldMap = make(map[string]field.Expr, 7)
+	s.fieldMap["id"] = s.ID
+	s.fieldMap["created_at"] = s.CreatedAt
+	s.fieldMap["updated_at"] = s.UpdatedAt
+	s.fieldMap["deleted_at"] = s.DeletedAt
+	s.fieldMap["path"] = s.Path
+	s.fieldMap["advanced"] = s.Advanced
+	s.fieldMap["chat_gpt_messages"] = s.ChatGPTMessages
+}
+
+func (s stream) clone(db *gorm.DB) stream {
+	s.streamDo.ReplaceConnPool(db.Statement.ConnPool)
+	return s
+}
+
+func (s stream) replaceDB(db *gorm.DB) stream {
+	s.streamDo.ReplaceDB(db)
+	return s
+}
+
+type streamDo struct{ gen.DO }
+
+// FirstByID Where("id=@id")
+func (s streamDo) FirstByID(id int) (result *model.Stream, err error) {
+	var params []interface{}
+
+	var generateSQL strings.Builder
+	params = append(params, id)
+	generateSQL.WriteString("id=? ")
+
+	var executeSQL *gorm.DB
+	executeSQL = s.UnderlyingDB().Where(generateSQL.String(), params...).Take(&result) // ignore_security_alert
+	err = executeSQL.Error
+
+	return
+}
+
+// DeleteByID update @@table set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=@id
+func (s streamDo) DeleteByID(id int) (err error) {
+	var params []interface{}
+
+	var generateSQL strings.Builder
+	params = append(params, id)
+	generateSQL.WriteString("update streams set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=? ")
+
+	var executeSQL *gorm.DB
+	executeSQL = s.UnderlyingDB().Exec(generateSQL.String(), params...) // ignore_security_alert
+	err = executeSQL.Error
+
+	return
+}
+
+func (s streamDo) Debug() *streamDo {
+	return s.withDO(s.DO.Debug())
+}
+
+func (s streamDo) WithContext(ctx context.Context) *streamDo {
+	return s.withDO(s.DO.WithContext(ctx))
+}
+
+func (s streamDo) ReadDB() *streamDo {
+	return s.Clauses(dbresolver.Read)
+}
+
+func (s streamDo) WriteDB() *streamDo {
+	return s.Clauses(dbresolver.Write)
+}
+
+func (s streamDo) Session(config *gorm.Session) *streamDo {
+	return s.withDO(s.DO.Session(config))
+}
+
+func (s streamDo) Clauses(conds ...clause.Expression) *streamDo {
+	return s.withDO(s.DO.Clauses(conds...))
+}
+
+func (s streamDo) Returning(value interface{}, columns ...string) *streamDo {
+	return s.withDO(s.DO.Returning(value, columns...))
+}
+
+func (s streamDo) Not(conds ...gen.Condition) *streamDo {
+	return s.withDO(s.DO.Not(conds...))
+}
+
+func (s streamDo) Or(conds ...gen.Condition) *streamDo {
+	return s.withDO(s.DO.Or(conds...))
+}
+
+func (s streamDo) Select(conds ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Select(conds...))
+}
+
+func (s streamDo) Where(conds ...gen.Condition) *streamDo {
+	return s.withDO(s.DO.Where(conds...))
+}
+
+func (s streamDo) Order(conds ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Order(conds...))
+}
+
+func (s streamDo) Distinct(cols ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Distinct(cols...))
+}
+
+func (s streamDo) Omit(cols ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Omit(cols...))
+}
+
+func (s streamDo) Join(table schema.Tabler, on ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Join(table, on...))
+}
+
+func (s streamDo) LeftJoin(table schema.Tabler, on ...field.Expr) *streamDo {
+	return s.withDO(s.DO.LeftJoin(table, on...))
+}
+
+func (s streamDo) RightJoin(table schema.Tabler, on ...field.Expr) *streamDo {
+	return s.withDO(s.DO.RightJoin(table, on...))
+}
+
+func (s streamDo) Group(cols ...field.Expr) *streamDo {
+	return s.withDO(s.DO.Group(cols...))
+}
+
+func (s streamDo) Having(conds ...gen.Condition) *streamDo {
+	return s.withDO(s.DO.Having(conds...))
+}
+
+func (s streamDo) Limit(limit int) *streamDo {
+	return s.withDO(s.DO.Limit(limit))
+}
+
+func (s streamDo) Offset(offset int) *streamDo {
+	return s.withDO(s.DO.Offset(offset))
+}
+
+func (s streamDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *streamDo {
+	return s.withDO(s.DO.Scopes(funcs...))
+}
+
+func (s streamDo) Unscoped() *streamDo {
+	return s.withDO(s.DO.Unscoped())
+}
+
+func (s streamDo) Create(values ...*model.Stream) error {
+	if len(values) == 0 {
+		return nil
+	}
+	return s.DO.Create(values)
+}
+
+func (s streamDo) CreateInBatches(values []*model.Stream, batchSize int) error {
+	return s.DO.CreateInBatches(values, batchSize)
+}
+
+// Save : !!! underlying implementation is different with GORM
+// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
+func (s streamDo) Save(values ...*model.Stream) error {
+	if len(values) == 0 {
+		return nil
+	}
+	return s.DO.Save(values)
+}
+
+func (s streamDo) First() (*model.Stream, error) {
+	if result, err := s.DO.First(); err != nil {
+		return nil, err
+	} else {
+		return result.(*model.Stream), nil
+	}
+}
+
+func (s streamDo) Take() (*model.Stream, error) {
+	if result, err := s.DO.Take(); err != nil {
+		return nil, err
+	} else {
+		return result.(*model.Stream), nil
+	}
+}
+
+func (s streamDo) Last() (*model.Stream, error) {
+	if result, err := s.DO.Last(); err != nil {
+		return nil, err
+	} else {
+		return result.(*model.Stream), nil
+	}
+}
+
+func (s streamDo) Find() ([]*model.Stream, error) {
+	result, err := s.DO.Find()
+	return result.([]*model.Stream), err
+}
+
+func (s streamDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.Stream, err error) {
+	buf := make([]*model.Stream, 0, batchSize)
+	err = s.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
+		defer func() { results = append(results, buf...) }()
+		return fc(tx, batch)
+	})
+	return results, err
+}
+
+func (s streamDo) FindInBatches(result *[]*model.Stream, batchSize int, fc func(tx gen.Dao, batch int) error) error {
+	return s.DO.FindInBatches(result, batchSize, fc)
+}
+
+func (s streamDo) Attrs(attrs ...field.AssignExpr) *streamDo {
+	return s.withDO(s.DO.Attrs(attrs...))
+}
+
+func (s streamDo) Assign(attrs ...field.AssignExpr) *streamDo {
+	return s.withDO(s.DO.Assign(attrs...))
+}
+
+func (s streamDo) Joins(fields ...field.RelationField) *streamDo {
+	for _, _f := range fields {
+		s = *s.withDO(s.DO.Joins(_f))
+	}
+	return &s
+}
+
+func (s streamDo) Preload(fields ...field.RelationField) *streamDo {
+	for _, _f := range fields {
+		s = *s.withDO(s.DO.Preload(_f))
+	}
+	return &s
+}
+
+func (s streamDo) FirstOrInit() (*model.Stream, error) {
+	if result, err := s.DO.FirstOrInit(); err != nil {
+		return nil, err
+	} else {
+		return result.(*model.Stream), nil
+	}
+}
+
+func (s streamDo) FirstOrCreate() (*model.Stream, error) {
+	if result, err := s.DO.FirstOrCreate(); err != nil {
+		return nil, err
+	} else {
+		return result.(*model.Stream), nil
+	}
+}
+
+func (s streamDo) FindByPage(offset int, limit int) (result []*model.Stream, count int64, err error) {
+	result, err = s.Offset(offset).Limit(limit).Find()
+	if err != nil {
+		return
+	}
+
+	if size := len(result); 0 < limit && 0 < size && size < limit {
+		count = int64(size + offset)
+		return
+	}
+
+	count, err = s.Offset(-1).Limit(-1).Count()
+	return
+}
+
+func (s streamDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
+	count, err = s.Count()
+	if err != nil {
+		return
+	}
+
+	err = s.Offset(offset).Limit(limit).Scan(result)
+	return
+}
+
+func (s streamDo) Scan(result interface{}) (err error) {
+	return s.DO.Scan(result)
+}
+
+func (s streamDo) Delete(models ...*model.Stream) (result gen.ResultInfo, err error) {
+	return s.DO.Delete(models)
+}
+
+func (s *streamDo) withDO(do gen.Dao) *streamDo {
+	s.DO = *do.(*gen.DO)
+	return s
+}