feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
fanlv
2025-07-20 17:36:12 +08:00
commit 890153324f
14811 changed files with 1923430 additions and 0 deletions

View File

@@ -0,0 +1,400 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newConnectorWorkflowVersion(db *gorm.DB, opts ...gen.DOOption) connectorWorkflowVersion {
_connectorWorkflowVersion := connectorWorkflowVersion{}
_connectorWorkflowVersion.connectorWorkflowVersionDo.UseDB(db, opts...)
_connectorWorkflowVersion.connectorWorkflowVersionDo.UseModel(&model.ConnectorWorkflowVersion{})
tableName := _connectorWorkflowVersion.connectorWorkflowVersionDo.TableName()
_connectorWorkflowVersion.ALL = field.NewAsterisk(tableName)
_connectorWorkflowVersion.ID = field.NewInt64(tableName, "id")
_connectorWorkflowVersion.AppID = field.NewInt64(tableName, "app_id")
_connectorWorkflowVersion.ConnectorID = field.NewInt64(tableName, "connector_id")
_connectorWorkflowVersion.WorkflowID = field.NewInt64(tableName, "workflow_id")
_connectorWorkflowVersion.Version = field.NewString(tableName, "version")
_connectorWorkflowVersion.CreatedAt = field.NewInt64(tableName, "created_at")
_connectorWorkflowVersion.fillFieldMap()
return _connectorWorkflowVersion
}
type connectorWorkflowVersion struct {
connectorWorkflowVersionDo
ALL field.Asterisk
ID field.Int64 // id
AppID field.Int64 // app id
ConnectorID field.Int64 // connector id
WorkflowID field.Int64 // workflow id
Version field.String // version
CreatedAt field.Int64 // create time in millisecond
fieldMap map[string]field.Expr
}
func (c connectorWorkflowVersion) Table(newTableName string) *connectorWorkflowVersion {
c.connectorWorkflowVersionDo.UseTable(newTableName)
return c.updateTableName(newTableName)
}
func (c connectorWorkflowVersion) As(alias string) *connectorWorkflowVersion {
c.connectorWorkflowVersionDo.DO = *(c.connectorWorkflowVersionDo.As(alias).(*gen.DO))
return c.updateTableName(alias)
}
func (c *connectorWorkflowVersion) updateTableName(table string) *connectorWorkflowVersion {
c.ALL = field.NewAsterisk(table)
c.ID = field.NewInt64(table, "id")
c.AppID = field.NewInt64(table, "app_id")
c.ConnectorID = field.NewInt64(table, "connector_id")
c.WorkflowID = field.NewInt64(table, "workflow_id")
c.Version = field.NewString(table, "version")
c.CreatedAt = field.NewInt64(table, "created_at")
c.fillFieldMap()
return c
}
func (c *connectorWorkflowVersion) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := c.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (c *connectorWorkflowVersion) fillFieldMap() {
c.fieldMap = make(map[string]field.Expr, 6)
c.fieldMap["id"] = c.ID
c.fieldMap["app_id"] = c.AppID
c.fieldMap["connector_id"] = c.ConnectorID
c.fieldMap["workflow_id"] = c.WorkflowID
c.fieldMap["version"] = c.Version
c.fieldMap["created_at"] = c.CreatedAt
}
func (c connectorWorkflowVersion) clone(db *gorm.DB) connectorWorkflowVersion {
c.connectorWorkflowVersionDo.ReplaceConnPool(db.Statement.ConnPool)
return c
}
func (c connectorWorkflowVersion) replaceDB(db *gorm.DB) connectorWorkflowVersion {
c.connectorWorkflowVersionDo.ReplaceDB(db)
return c
}
type connectorWorkflowVersionDo struct{ gen.DO }
type IConnectorWorkflowVersionDo interface {
gen.SubQuery
Debug() IConnectorWorkflowVersionDo
WithContext(ctx context.Context) IConnectorWorkflowVersionDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IConnectorWorkflowVersionDo
WriteDB() IConnectorWorkflowVersionDo
As(alias string) gen.Dao
Session(config *gorm.Session) IConnectorWorkflowVersionDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IConnectorWorkflowVersionDo
Not(conds ...gen.Condition) IConnectorWorkflowVersionDo
Or(conds ...gen.Condition) IConnectorWorkflowVersionDo
Select(conds ...field.Expr) IConnectorWorkflowVersionDo
Where(conds ...gen.Condition) IConnectorWorkflowVersionDo
Order(conds ...field.Expr) IConnectorWorkflowVersionDo
Distinct(cols ...field.Expr) IConnectorWorkflowVersionDo
Omit(cols ...field.Expr) IConnectorWorkflowVersionDo
Join(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo
LeftJoin(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo
RightJoin(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo
Group(cols ...field.Expr) IConnectorWorkflowVersionDo
Having(conds ...gen.Condition) IConnectorWorkflowVersionDo
Limit(limit int) IConnectorWorkflowVersionDo
Offset(offset int) IConnectorWorkflowVersionDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IConnectorWorkflowVersionDo
Unscoped() IConnectorWorkflowVersionDo
Create(values ...*model.ConnectorWorkflowVersion) error
CreateInBatches(values []*model.ConnectorWorkflowVersion, batchSize int) error
Save(values ...*model.ConnectorWorkflowVersion) error
First() (*model.ConnectorWorkflowVersion, error)
Take() (*model.ConnectorWorkflowVersion, error)
Last() (*model.ConnectorWorkflowVersion, error)
Find() ([]*model.ConnectorWorkflowVersion, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.ConnectorWorkflowVersion, err error)
FindInBatches(result *[]*model.ConnectorWorkflowVersion, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.ConnectorWorkflowVersion) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IConnectorWorkflowVersionDo
Assign(attrs ...field.AssignExpr) IConnectorWorkflowVersionDo
Joins(fields ...field.RelationField) IConnectorWorkflowVersionDo
Preload(fields ...field.RelationField) IConnectorWorkflowVersionDo
FirstOrInit() (*model.ConnectorWorkflowVersion, error)
FirstOrCreate() (*model.ConnectorWorkflowVersion, error)
FindByPage(offset int, limit int) (result []*model.ConnectorWorkflowVersion, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IConnectorWorkflowVersionDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (c connectorWorkflowVersionDo) Debug() IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Debug())
}
func (c connectorWorkflowVersionDo) WithContext(ctx context.Context) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.WithContext(ctx))
}
func (c connectorWorkflowVersionDo) ReadDB() IConnectorWorkflowVersionDo {
return c.Clauses(dbresolver.Read)
}
func (c connectorWorkflowVersionDo) WriteDB() IConnectorWorkflowVersionDo {
return c.Clauses(dbresolver.Write)
}
func (c connectorWorkflowVersionDo) Session(config *gorm.Session) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Session(config))
}
func (c connectorWorkflowVersionDo) Clauses(conds ...clause.Expression) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Clauses(conds...))
}
func (c connectorWorkflowVersionDo) Returning(value interface{}, columns ...string) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Returning(value, columns...))
}
func (c connectorWorkflowVersionDo) Not(conds ...gen.Condition) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Not(conds...))
}
func (c connectorWorkflowVersionDo) Or(conds ...gen.Condition) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Or(conds...))
}
func (c connectorWorkflowVersionDo) Select(conds ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Select(conds...))
}
func (c connectorWorkflowVersionDo) Where(conds ...gen.Condition) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Where(conds...))
}
func (c connectorWorkflowVersionDo) Order(conds ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Order(conds...))
}
func (c connectorWorkflowVersionDo) Distinct(cols ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Distinct(cols...))
}
func (c connectorWorkflowVersionDo) Omit(cols ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Omit(cols...))
}
func (c connectorWorkflowVersionDo) Join(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Join(table, on...))
}
func (c connectorWorkflowVersionDo) LeftJoin(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.LeftJoin(table, on...))
}
func (c connectorWorkflowVersionDo) RightJoin(table schema.Tabler, on ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.RightJoin(table, on...))
}
func (c connectorWorkflowVersionDo) Group(cols ...field.Expr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Group(cols...))
}
func (c connectorWorkflowVersionDo) Having(conds ...gen.Condition) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Having(conds...))
}
func (c connectorWorkflowVersionDo) Limit(limit int) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Limit(limit))
}
func (c connectorWorkflowVersionDo) Offset(offset int) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Offset(offset))
}
func (c connectorWorkflowVersionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Scopes(funcs...))
}
func (c connectorWorkflowVersionDo) Unscoped() IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Unscoped())
}
func (c connectorWorkflowVersionDo) Create(values ...*model.ConnectorWorkflowVersion) error {
if len(values) == 0 {
return nil
}
return c.DO.Create(values)
}
func (c connectorWorkflowVersionDo) CreateInBatches(values []*model.ConnectorWorkflowVersion, batchSize int) error {
return c.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 (c connectorWorkflowVersionDo) Save(values ...*model.ConnectorWorkflowVersion) error {
if len(values) == 0 {
return nil
}
return c.DO.Save(values)
}
func (c connectorWorkflowVersionDo) First() (*model.ConnectorWorkflowVersion, error) {
if result, err := c.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.ConnectorWorkflowVersion), nil
}
}
func (c connectorWorkflowVersionDo) Take() (*model.ConnectorWorkflowVersion, error) {
if result, err := c.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.ConnectorWorkflowVersion), nil
}
}
func (c connectorWorkflowVersionDo) Last() (*model.ConnectorWorkflowVersion, error) {
if result, err := c.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.ConnectorWorkflowVersion), nil
}
}
func (c connectorWorkflowVersionDo) Find() ([]*model.ConnectorWorkflowVersion, error) {
result, err := c.DO.Find()
return result.([]*model.ConnectorWorkflowVersion), err
}
func (c connectorWorkflowVersionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.ConnectorWorkflowVersion, err error) {
buf := make([]*model.ConnectorWorkflowVersion, 0, batchSize)
err = c.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 (c connectorWorkflowVersionDo) FindInBatches(result *[]*model.ConnectorWorkflowVersion, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return c.DO.FindInBatches(result, batchSize, fc)
}
func (c connectorWorkflowVersionDo) Attrs(attrs ...field.AssignExpr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Attrs(attrs...))
}
func (c connectorWorkflowVersionDo) Assign(attrs ...field.AssignExpr) IConnectorWorkflowVersionDo {
return c.withDO(c.DO.Assign(attrs...))
}
func (c connectorWorkflowVersionDo) Joins(fields ...field.RelationField) IConnectorWorkflowVersionDo {
for _, _f := range fields {
c = *c.withDO(c.DO.Joins(_f))
}
return &c
}
func (c connectorWorkflowVersionDo) Preload(fields ...field.RelationField) IConnectorWorkflowVersionDo {
for _, _f := range fields {
c = *c.withDO(c.DO.Preload(_f))
}
return &c
}
func (c connectorWorkflowVersionDo) FirstOrInit() (*model.ConnectorWorkflowVersion, error) {
if result, err := c.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.ConnectorWorkflowVersion), nil
}
}
func (c connectorWorkflowVersionDo) FirstOrCreate() (*model.ConnectorWorkflowVersion, error) {
if result, err := c.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.ConnectorWorkflowVersion), nil
}
}
func (c connectorWorkflowVersionDo) FindByPage(offset int, limit int) (result []*model.ConnectorWorkflowVersion, count int64, err error) {
result, err = c.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 = c.Offset(-1).Limit(-1).Count()
return
}
func (c connectorWorkflowVersionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = c.Count()
if err != nil {
return
}
err = c.Offset(offset).Limit(limit).Scan(result)
return
}
func (c connectorWorkflowVersionDo) Scan(result interface{}) (err error) {
return c.DO.Scan(result)
}
func (c connectorWorkflowVersionDo) Delete(models ...*model.ConnectorWorkflowVersion) (result gen.ResultInfo, err error) {
return c.DO.Delete(models)
}
func (c *connectorWorkflowVersionDo) withDO(do gen.Dao) *connectorWorkflowVersionDo {
c.DO = *do.(*gen.DO)
return c
}

View File

@@ -0,0 +1,159 @@
// 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"
"database/sql"
"gorm.io/gorm"
"gorm.io/gen"
"gorm.io/plugin/dbresolver"
)
var (
Q = new(Query)
ConnectorWorkflowVersion *connectorWorkflowVersion
NodeExecution *nodeExecution
WorkflowDraft *workflowDraft
WorkflowExecution *workflowExecution
WorkflowMeta *workflowMeta
WorkflowReference *workflowReference
WorkflowSnapshot *workflowSnapshot
WorkflowVersion *workflowVersion
)
func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
*Q = *Use(db, opts...)
ConnectorWorkflowVersion = &Q.ConnectorWorkflowVersion
NodeExecution = &Q.NodeExecution
WorkflowDraft = &Q.WorkflowDraft
WorkflowExecution = &Q.WorkflowExecution
WorkflowMeta = &Q.WorkflowMeta
WorkflowReference = &Q.WorkflowReference
WorkflowSnapshot = &Q.WorkflowSnapshot
WorkflowVersion = &Q.WorkflowVersion
}
func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
return &Query{
db: db,
ConnectorWorkflowVersion: newConnectorWorkflowVersion(db, opts...),
NodeExecution: newNodeExecution(db, opts...),
WorkflowDraft: newWorkflowDraft(db, opts...),
WorkflowExecution: newWorkflowExecution(db, opts...),
WorkflowMeta: newWorkflowMeta(db, opts...),
WorkflowReference: newWorkflowReference(db, opts...),
WorkflowSnapshot: newWorkflowSnapshot(db, opts...),
WorkflowVersion: newWorkflowVersion(db, opts...),
}
}
type Query struct {
db *gorm.DB
ConnectorWorkflowVersion connectorWorkflowVersion
NodeExecution nodeExecution
WorkflowDraft workflowDraft
WorkflowExecution workflowExecution
WorkflowMeta workflowMeta
WorkflowReference workflowReference
WorkflowSnapshot workflowSnapshot
WorkflowVersion workflowVersion
}
func (q *Query) Available() bool { return q.db != nil }
func (q *Query) clone(db *gorm.DB) *Query {
return &Query{
db: db,
ConnectorWorkflowVersion: q.ConnectorWorkflowVersion.clone(db),
NodeExecution: q.NodeExecution.clone(db),
WorkflowDraft: q.WorkflowDraft.clone(db),
WorkflowExecution: q.WorkflowExecution.clone(db),
WorkflowMeta: q.WorkflowMeta.clone(db),
WorkflowReference: q.WorkflowReference.clone(db),
WorkflowSnapshot: q.WorkflowSnapshot.clone(db),
WorkflowVersion: q.WorkflowVersion.clone(db),
}
}
func (q *Query) ReadDB() *Query {
return q.ReplaceDB(q.db.Clauses(dbresolver.Read))
}
func (q *Query) WriteDB() *Query {
return q.ReplaceDB(q.db.Clauses(dbresolver.Write))
}
func (q *Query) ReplaceDB(db *gorm.DB) *Query {
return &Query{
db: db,
ConnectorWorkflowVersion: q.ConnectorWorkflowVersion.replaceDB(db),
NodeExecution: q.NodeExecution.replaceDB(db),
WorkflowDraft: q.WorkflowDraft.replaceDB(db),
WorkflowExecution: q.WorkflowExecution.replaceDB(db),
WorkflowMeta: q.WorkflowMeta.replaceDB(db),
WorkflowReference: q.WorkflowReference.replaceDB(db),
WorkflowSnapshot: q.WorkflowSnapshot.replaceDB(db),
WorkflowVersion: q.WorkflowVersion.replaceDB(db),
}
}
type queryCtx struct {
ConnectorWorkflowVersion IConnectorWorkflowVersionDo
NodeExecution INodeExecutionDo
WorkflowDraft IWorkflowDraftDo
WorkflowExecution IWorkflowExecutionDo
WorkflowMeta IWorkflowMetaDo
WorkflowReference IWorkflowReferenceDo
WorkflowSnapshot IWorkflowSnapshotDo
WorkflowVersion IWorkflowVersionDo
}
func (q *Query) WithContext(ctx context.Context) *queryCtx {
return &queryCtx{
ConnectorWorkflowVersion: q.ConnectorWorkflowVersion.WithContext(ctx),
NodeExecution: q.NodeExecution.WithContext(ctx),
WorkflowDraft: q.WorkflowDraft.WithContext(ctx),
WorkflowExecution: q.WorkflowExecution.WithContext(ctx),
WorkflowMeta: q.WorkflowMeta.WithContext(ctx),
WorkflowReference: q.WorkflowReference.WithContext(ctx),
WorkflowSnapshot: q.WorkflowSnapshot.WithContext(ctx),
WorkflowVersion: q.WorkflowVersion.WithContext(ctx),
}
}
func (q *Query) Transaction(fc func(tx *Query) error, opts ...*sql.TxOptions) error {
return q.db.Transaction(func(tx *gorm.DB) error { return fc(q.clone(tx)) }, opts...)
}
func (q *Query) Begin(opts ...*sql.TxOptions) *QueryTx {
tx := q.db.Begin(opts...)
return &QueryTx{Query: q.clone(tx), Error: tx.Error}
}
type QueryTx struct {
*Query
Error error
}
func (q *QueryTx) Commit() error {
return q.db.Commit().Error
}
func (q *QueryTx) Rollback() error {
return q.db.Rollback().Error
}
func (q *QueryTx) SavePoint(name string) error {
return q.db.SavePoint(name).Error
}
func (q *QueryTx) RollbackTo(name string) error {
return q.db.RollbackTo(name).Error
}

View File

@@ -0,0 +1,461 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newNodeExecution(db *gorm.DB, opts ...gen.DOOption) nodeExecution {
_nodeExecution := nodeExecution{}
_nodeExecution.nodeExecutionDo.UseDB(db, opts...)
_nodeExecution.nodeExecutionDo.UseModel(&model.NodeExecution{})
tableName := _nodeExecution.nodeExecutionDo.TableName()
_nodeExecution.ALL = field.NewAsterisk(tableName)
_nodeExecution.ID = field.NewInt64(tableName, "id")
_nodeExecution.ExecuteID = field.NewInt64(tableName, "execute_id")
_nodeExecution.NodeID = field.NewString(tableName, "node_id")
_nodeExecution.NodeName = field.NewString(tableName, "node_name")
_nodeExecution.NodeType = field.NewString(tableName, "node_type")
_nodeExecution.CreatedAt = field.NewInt64(tableName, "created_at")
_nodeExecution.Status = field.NewInt32(tableName, "status")
_nodeExecution.Duration = field.NewInt64(tableName, "duration")
_nodeExecution.Input = field.NewString(tableName, "input")
_nodeExecution.Output = field.NewString(tableName, "output")
_nodeExecution.RawOutput = field.NewString(tableName, "raw_output")
_nodeExecution.ErrorInfo = field.NewString(tableName, "error_info")
_nodeExecution.ErrorLevel = field.NewString(tableName, "error_level")
_nodeExecution.InputTokens = field.NewInt64(tableName, "input_tokens")
_nodeExecution.OutputTokens = field.NewInt64(tableName, "output_tokens")
_nodeExecution.UpdatedAt = field.NewInt64(tableName, "updated_at")
_nodeExecution.CompositeNodeIndex = field.NewInt64(tableName, "composite_node_index")
_nodeExecution.CompositeNodeItems = field.NewString(tableName, "composite_node_items")
_nodeExecution.ParentNodeID = field.NewString(tableName, "parent_node_id")
_nodeExecution.SubExecuteID = field.NewInt64(tableName, "sub_execute_id")
_nodeExecution.Extra = field.NewString(tableName, "extra")
_nodeExecution.fillFieldMap()
return _nodeExecution
}
// nodeExecution node 节点运行记录用于记录每次workflow执行时每个节点的状态信息
type nodeExecution struct {
nodeExecutionDo
ALL field.Asterisk
ID field.Int64 // node execution id
ExecuteID field.Int64 // the workflow execute id this node execution belongs to
NodeID field.String // node key
NodeName field.String // name of the node
NodeType field.String // the type of the node, in string
CreatedAt field.Int64 // create time in millisecond
Status field.Int32 // 1=waiting 2=running 3=success 4=fail
Duration field.Int64 // execution duration in millisecond
Input field.String // actual input of the node
Output field.String // actual output of the node
RawOutput field.String // the original output of the node
ErrorInfo field.String // error info
ErrorLevel field.String // level of the error
InputTokens field.Int64 // number of input tokens
OutputTokens field.Int64 // number of output tokens
UpdatedAt field.Int64 // update time in millisecond
CompositeNodeIndex field.Int64 // loop or batch's execution index
CompositeNodeItems field.String // the items extracted from parent composite node for this index
ParentNodeID field.String // when as inner node for loop or batch, this is the parent node's key
SubExecuteID field.Int64 // if this node is sub_workflow, the exe id of the sub workflow
Extra field.String // extra info
fieldMap map[string]field.Expr
}
func (n nodeExecution) Table(newTableName string) *nodeExecution {
n.nodeExecutionDo.UseTable(newTableName)
return n.updateTableName(newTableName)
}
func (n nodeExecution) As(alias string) *nodeExecution {
n.nodeExecutionDo.DO = *(n.nodeExecutionDo.As(alias).(*gen.DO))
return n.updateTableName(alias)
}
func (n *nodeExecution) updateTableName(table string) *nodeExecution {
n.ALL = field.NewAsterisk(table)
n.ID = field.NewInt64(table, "id")
n.ExecuteID = field.NewInt64(table, "execute_id")
n.NodeID = field.NewString(table, "node_id")
n.NodeName = field.NewString(table, "node_name")
n.NodeType = field.NewString(table, "node_type")
n.CreatedAt = field.NewInt64(table, "created_at")
n.Status = field.NewInt32(table, "status")
n.Duration = field.NewInt64(table, "duration")
n.Input = field.NewString(table, "input")
n.Output = field.NewString(table, "output")
n.RawOutput = field.NewString(table, "raw_output")
n.ErrorInfo = field.NewString(table, "error_info")
n.ErrorLevel = field.NewString(table, "error_level")
n.InputTokens = field.NewInt64(table, "input_tokens")
n.OutputTokens = field.NewInt64(table, "output_tokens")
n.UpdatedAt = field.NewInt64(table, "updated_at")
n.CompositeNodeIndex = field.NewInt64(table, "composite_node_index")
n.CompositeNodeItems = field.NewString(table, "composite_node_items")
n.ParentNodeID = field.NewString(table, "parent_node_id")
n.SubExecuteID = field.NewInt64(table, "sub_execute_id")
n.Extra = field.NewString(table, "extra")
n.fillFieldMap()
return n
}
func (n *nodeExecution) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := n.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (n *nodeExecution) fillFieldMap() {
n.fieldMap = make(map[string]field.Expr, 21)
n.fieldMap["id"] = n.ID
n.fieldMap["execute_id"] = n.ExecuteID
n.fieldMap["node_id"] = n.NodeID
n.fieldMap["node_name"] = n.NodeName
n.fieldMap["node_type"] = n.NodeType
n.fieldMap["created_at"] = n.CreatedAt
n.fieldMap["status"] = n.Status
n.fieldMap["duration"] = n.Duration
n.fieldMap["input"] = n.Input
n.fieldMap["output"] = n.Output
n.fieldMap["raw_output"] = n.RawOutput
n.fieldMap["error_info"] = n.ErrorInfo
n.fieldMap["error_level"] = n.ErrorLevel
n.fieldMap["input_tokens"] = n.InputTokens
n.fieldMap["output_tokens"] = n.OutputTokens
n.fieldMap["updated_at"] = n.UpdatedAt
n.fieldMap["composite_node_index"] = n.CompositeNodeIndex
n.fieldMap["composite_node_items"] = n.CompositeNodeItems
n.fieldMap["parent_node_id"] = n.ParentNodeID
n.fieldMap["sub_execute_id"] = n.SubExecuteID
n.fieldMap["extra"] = n.Extra
}
func (n nodeExecution) clone(db *gorm.DB) nodeExecution {
n.nodeExecutionDo.ReplaceConnPool(db.Statement.ConnPool)
return n
}
func (n nodeExecution) replaceDB(db *gorm.DB) nodeExecution {
n.nodeExecutionDo.ReplaceDB(db)
return n
}
type nodeExecutionDo struct{ gen.DO }
type INodeExecutionDo interface {
gen.SubQuery
Debug() INodeExecutionDo
WithContext(ctx context.Context) INodeExecutionDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() INodeExecutionDo
WriteDB() INodeExecutionDo
As(alias string) gen.Dao
Session(config *gorm.Session) INodeExecutionDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) INodeExecutionDo
Not(conds ...gen.Condition) INodeExecutionDo
Or(conds ...gen.Condition) INodeExecutionDo
Select(conds ...field.Expr) INodeExecutionDo
Where(conds ...gen.Condition) INodeExecutionDo
Order(conds ...field.Expr) INodeExecutionDo
Distinct(cols ...field.Expr) INodeExecutionDo
Omit(cols ...field.Expr) INodeExecutionDo
Join(table schema.Tabler, on ...field.Expr) INodeExecutionDo
LeftJoin(table schema.Tabler, on ...field.Expr) INodeExecutionDo
RightJoin(table schema.Tabler, on ...field.Expr) INodeExecutionDo
Group(cols ...field.Expr) INodeExecutionDo
Having(conds ...gen.Condition) INodeExecutionDo
Limit(limit int) INodeExecutionDo
Offset(offset int) INodeExecutionDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) INodeExecutionDo
Unscoped() INodeExecutionDo
Create(values ...*model.NodeExecution) error
CreateInBatches(values []*model.NodeExecution, batchSize int) error
Save(values ...*model.NodeExecution) error
First() (*model.NodeExecution, error)
Take() (*model.NodeExecution, error)
Last() (*model.NodeExecution, error)
Find() ([]*model.NodeExecution, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.NodeExecution, err error)
FindInBatches(result *[]*model.NodeExecution, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.NodeExecution) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) INodeExecutionDo
Assign(attrs ...field.AssignExpr) INodeExecutionDo
Joins(fields ...field.RelationField) INodeExecutionDo
Preload(fields ...field.RelationField) INodeExecutionDo
FirstOrInit() (*model.NodeExecution, error)
FirstOrCreate() (*model.NodeExecution, error)
FindByPage(offset int, limit int) (result []*model.NodeExecution, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) INodeExecutionDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (n nodeExecutionDo) Debug() INodeExecutionDo {
return n.withDO(n.DO.Debug())
}
func (n nodeExecutionDo) WithContext(ctx context.Context) INodeExecutionDo {
return n.withDO(n.DO.WithContext(ctx))
}
func (n nodeExecutionDo) ReadDB() INodeExecutionDo {
return n.Clauses(dbresolver.Read)
}
func (n nodeExecutionDo) WriteDB() INodeExecutionDo {
return n.Clauses(dbresolver.Write)
}
func (n nodeExecutionDo) Session(config *gorm.Session) INodeExecutionDo {
return n.withDO(n.DO.Session(config))
}
func (n nodeExecutionDo) Clauses(conds ...clause.Expression) INodeExecutionDo {
return n.withDO(n.DO.Clauses(conds...))
}
func (n nodeExecutionDo) Returning(value interface{}, columns ...string) INodeExecutionDo {
return n.withDO(n.DO.Returning(value, columns...))
}
func (n nodeExecutionDo) Not(conds ...gen.Condition) INodeExecutionDo {
return n.withDO(n.DO.Not(conds...))
}
func (n nodeExecutionDo) Or(conds ...gen.Condition) INodeExecutionDo {
return n.withDO(n.DO.Or(conds...))
}
func (n nodeExecutionDo) Select(conds ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Select(conds...))
}
func (n nodeExecutionDo) Where(conds ...gen.Condition) INodeExecutionDo {
return n.withDO(n.DO.Where(conds...))
}
func (n nodeExecutionDo) Order(conds ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Order(conds...))
}
func (n nodeExecutionDo) Distinct(cols ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Distinct(cols...))
}
func (n nodeExecutionDo) Omit(cols ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Omit(cols...))
}
func (n nodeExecutionDo) Join(table schema.Tabler, on ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Join(table, on...))
}
func (n nodeExecutionDo) LeftJoin(table schema.Tabler, on ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.LeftJoin(table, on...))
}
func (n nodeExecutionDo) RightJoin(table schema.Tabler, on ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.RightJoin(table, on...))
}
func (n nodeExecutionDo) Group(cols ...field.Expr) INodeExecutionDo {
return n.withDO(n.DO.Group(cols...))
}
func (n nodeExecutionDo) Having(conds ...gen.Condition) INodeExecutionDo {
return n.withDO(n.DO.Having(conds...))
}
func (n nodeExecutionDo) Limit(limit int) INodeExecutionDo {
return n.withDO(n.DO.Limit(limit))
}
func (n nodeExecutionDo) Offset(offset int) INodeExecutionDo {
return n.withDO(n.DO.Offset(offset))
}
func (n nodeExecutionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) INodeExecutionDo {
return n.withDO(n.DO.Scopes(funcs...))
}
func (n nodeExecutionDo) Unscoped() INodeExecutionDo {
return n.withDO(n.DO.Unscoped())
}
func (n nodeExecutionDo) Create(values ...*model.NodeExecution) error {
if len(values) == 0 {
return nil
}
return n.DO.Create(values)
}
func (n nodeExecutionDo) CreateInBatches(values []*model.NodeExecution, batchSize int) error {
return n.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 (n nodeExecutionDo) Save(values ...*model.NodeExecution) error {
if len(values) == 0 {
return nil
}
return n.DO.Save(values)
}
func (n nodeExecutionDo) First() (*model.NodeExecution, error) {
if result, err := n.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.NodeExecution), nil
}
}
func (n nodeExecutionDo) Take() (*model.NodeExecution, error) {
if result, err := n.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.NodeExecution), nil
}
}
func (n nodeExecutionDo) Last() (*model.NodeExecution, error) {
if result, err := n.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.NodeExecution), nil
}
}
func (n nodeExecutionDo) Find() ([]*model.NodeExecution, error) {
result, err := n.DO.Find()
return result.([]*model.NodeExecution), err
}
func (n nodeExecutionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.NodeExecution, err error) {
buf := make([]*model.NodeExecution, 0, batchSize)
err = n.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 (n nodeExecutionDo) FindInBatches(result *[]*model.NodeExecution, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return n.DO.FindInBatches(result, batchSize, fc)
}
func (n nodeExecutionDo) Attrs(attrs ...field.AssignExpr) INodeExecutionDo {
return n.withDO(n.DO.Attrs(attrs...))
}
func (n nodeExecutionDo) Assign(attrs ...field.AssignExpr) INodeExecutionDo {
return n.withDO(n.DO.Assign(attrs...))
}
func (n nodeExecutionDo) Joins(fields ...field.RelationField) INodeExecutionDo {
for _, _f := range fields {
n = *n.withDO(n.DO.Joins(_f))
}
return &n
}
func (n nodeExecutionDo) Preload(fields ...field.RelationField) INodeExecutionDo {
for _, _f := range fields {
n = *n.withDO(n.DO.Preload(_f))
}
return &n
}
func (n nodeExecutionDo) FirstOrInit() (*model.NodeExecution, error) {
if result, err := n.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.NodeExecution), nil
}
}
func (n nodeExecutionDo) FirstOrCreate() (*model.NodeExecution, error) {
if result, err := n.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.NodeExecution), nil
}
}
func (n nodeExecutionDo) FindByPage(offset int, limit int) (result []*model.NodeExecution, count int64, err error) {
result, err = n.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 = n.Offset(-1).Limit(-1).Count()
return
}
func (n nodeExecutionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = n.Count()
if err != nil {
return
}
err = n.Offset(offset).Limit(limit).Scan(result)
return
}
func (n nodeExecutionDo) Scan(result interface{}) (err error) {
return n.DO.Scan(result)
}
func (n nodeExecutionDo) Delete(models ...*model.NodeExecution) (result gen.ResultInfo, err error) {
return n.DO.Delete(models)
}
func (n *nodeExecutionDo) withDO(do gen.Dao) *nodeExecutionDo {
n.DO = *do.(*gen.DO)
return n
}

View File

@@ -0,0 +1,413 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowDraft(db *gorm.DB, opts ...gen.DOOption) workflowDraft {
_workflowDraft := workflowDraft{}
_workflowDraft.workflowDraftDo.UseDB(db, opts...)
_workflowDraft.workflowDraftDo.UseModel(&model.WorkflowDraft{})
tableName := _workflowDraft.workflowDraftDo.TableName()
_workflowDraft.ALL = field.NewAsterisk(tableName)
_workflowDraft.ID = field.NewInt64(tableName, "id")
_workflowDraft.Canvas = field.NewString(tableName, "canvas")
_workflowDraft.InputParams = field.NewString(tableName, "input_params")
_workflowDraft.OutputParams = field.NewString(tableName, "output_params")
_workflowDraft.TestRunSuccess = field.NewBool(tableName, "test_run_success")
_workflowDraft.Modified = field.NewBool(tableName, "modified")
_workflowDraft.UpdatedAt = field.NewInt64(tableName, "updated_at")
_workflowDraft.DeletedAt = field.NewField(tableName, "deleted_at")
_workflowDraft.CommitID = field.NewString(tableName, "commit_id")
_workflowDraft.fillFieldMap()
return _workflowDraft
}
// workflowDraft workflow 画布草稿表用于记录workflow最新的草稿画布信息
type workflowDraft struct {
workflowDraftDo
ALL field.Asterisk
ID field.Int64 // workflow ID
Canvas field.String // 前端 schema
InputParams field.String // 入参 schema
OutputParams field.String // 出参 schema
TestRunSuccess field.Bool // 0 未运行, 1 运行成功
Modified field.Bool // 0 未被修改, 1 已被修改
UpdatedAt field.Int64
DeletedAt field.Field
CommitID field.String // used to uniquely identify a draft snapshot
fieldMap map[string]field.Expr
}
func (w workflowDraft) Table(newTableName string) *workflowDraft {
w.workflowDraftDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowDraft) As(alias string) *workflowDraft {
w.workflowDraftDo.DO = *(w.workflowDraftDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowDraft) updateTableName(table string) *workflowDraft {
w.ALL = field.NewAsterisk(table)
w.ID = field.NewInt64(table, "id")
w.Canvas = field.NewString(table, "canvas")
w.InputParams = field.NewString(table, "input_params")
w.OutputParams = field.NewString(table, "output_params")
w.TestRunSuccess = field.NewBool(table, "test_run_success")
w.Modified = field.NewBool(table, "modified")
w.UpdatedAt = field.NewInt64(table, "updated_at")
w.DeletedAt = field.NewField(table, "deleted_at")
w.CommitID = field.NewString(table, "commit_id")
w.fillFieldMap()
return w
}
func (w *workflowDraft) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowDraft) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 9)
w.fieldMap["id"] = w.ID
w.fieldMap["canvas"] = w.Canvas
w.fieldMap["input_params"] = w.InputParams
w.fieldMap["output_params"] = w.OutputParams
w.fieldMap["test_run_success"] = w.TestRunSuccess
w.fieldMap["modified"] = w.Modified
w.fieldMap["updated_at"] = w.UpdatedAt
w.fieldMap["deleted_at"] = w.DeletedAt
w.fieldMap["commit_id"] = w.CommitID
}
func (w workflowDraft) clone(db *gorm.DB) workflowDraft {
w.workflowDraftDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowDraft) replaceDB(db *gorm.DB) workflowDraft {
w.workflowDraftDo.ReplaceDB(db)
return w
}
type workflowDraftDo struct{ gen.DO }
type IWorkflowDraftDo interface {
gen.SubQuery
Debug() IWorkflowDraftDo
WithContext(ctx context.Context) IWorkflowDraftDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowDraftDo
WriteDB() IWorkflowDraftDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowDraftDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowDraftDo
Not(conds ...gen.Condition) IWorkflowDraftDo
Or(conds ...gen.Condition) IWorkflowDraftDo
Select(conds ...field.Expr) IWorkflowDraftDo
Where(conds ...gen.Condition) IWorkflowDraftDo
Order(conds ...field.Expr) IWorkflowDraftDo
Distinct(cols ...field.Expr) IWorkflowDraftDo
Omit(cols ...field.Expr) IWorkflowDraftDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo
Group(cols ...field.Expr) IWorkflowDraftDo
Having(conds ...gen.Condition) IWorkflowDraftDo
Limit(limit int) IWorkflowDraftDo
Offset(offset int) IWorkflowDraftDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowDraftDo
Unscoped() IWorkflowDraftDo
Create(values ...*model.WorkflowDraft) error
CreateInBatches(values []*model.WorkflowDraft, batchSize int) error
Save(values ...*model.WorkflowDraft) error
First() (*model.WorkflowDraft, error)
Take() (*model.WorkflowDraft, error)
Last() (*model.WorkflowDraft, error)
Find() ([]*model.WorkflowDraft, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowDraft, err error)
FindInBatches(result *[]*model.WorkflowDraft, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowDraft) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowDraftDo
Assign(attrs ...field.AssignExpr) IWorkflowDraftDo
Joins(fields ...field.RelationField) IWorkflowDraftDo
Preload(fields ...field.RelationField) IWorkflowDraftDo
FirstOrInit() (*model.WorkflowDraft, error)
FirstOrCreate() (*model.WorkflowDraft, error)
FindByPage(offset int, limit int) (result []*model.WorkflowDraft, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowDraftDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowDraftDo) Debug() IWorkflowDraftDo {
return w.withDO(w.DO.Debug())
}
func (w workflowDraftDo) WithContext(ctx context.Context) IWorkflowDraftDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowDraftDo) ReadDB() IWorkflowDraftDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowDraftDo) WriteDB() IWorkflowDraftDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowDraftDo) Session(config *gorm.Session) IWorkflowDraftDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowDraftDo) Clauses(conds ...clause.Expression) IWorkflowDraftDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowDraftDo) Returning(value interface{}, columns ...string) IWorkflowDraftDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowDraftDo) Not(conds ...gen.Condition) IWorkflowDraftDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowDraftDo) Or(conds ...gen.Condition) IWorkflowDraftDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowDraftDo) Select(conds ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowDraftDo) Where(conds ...gen.Condition) IWorkflowDraftDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowDraftDo) Order(conds ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowDraftDo) Distinct(cols ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowDraftDo) Omit(cols ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowDraftDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowDraftDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowDraftDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowDraftDo) Group(cols ...field.Expr) IWorkflowDraftDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowDraftDo) Having(conds ...gen.Condition) IWorkflowDraftDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowDraftDo) Limit(limit int) IWorkflowDraftDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowDraftDo) Offset(offset int) IWorkflowDraftDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowDraftDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowDraftDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowDraftDo) Unscoped() IWorkflowDraftDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowDraftDo) Create(values ...*model.WorkflowDraft) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowDraftDo) CreateInBatches(values []*model.WorkflowDraft, batchSize int) error {
return w.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 (w workflowDraftDo) Save(values ...*model.WorkflowDraft) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowDraftDo) First() (*model.WorkflowDraft, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowDraft), nil
}
}
func (w workflowDraftDo) Take() (*model.WorkflowDraft, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowDraft), nil
}
}
func (w workflowDraftDo) Last() (*model.WorkflowDraft, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowDraft), nil
}
}
func (w workflowDraftDo) Find() ([]*model.WorkflowDraft, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowDraft), err
}
func (w workflowDraftDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowDraft, err error) {
buf := make([]*model.WorkflowDraft, 0, batchSize)
err = w.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 (w workflowDraftDo) FindInBatches(result *[]*model.WorkflowDraft, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowDraftDo) Attrs(attrs ...field.AssignExpr) IWorkflowDraftDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowDraftDo) Assign(attrs ...field.AssignExpr) IWorkflowDraftDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowDraftDo) Joins(fields ...field.RelationField) IWorkflowDraftDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowDraftDo) Preload(fields ...field.RelationField) IWorkflowDraftDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowDraftDo) FirstOrInit() (*model.WorkflowDraft, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowDraft), nil
}
}
func (w workflowDraftDo) FirstOrCreate() (*model.WorkflowDraft, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowDraft), nil
}
}
func (w workflowDraftDo) FindByPage(offset int, limit int) (result []*model.WorkflowDraft, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowDraftDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowDraftDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowDraftDo) Delete(models ...*model.WorkflowDraft) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowDraftDo) withDO(do gen.Dao) *workflowDraftDo {
w.DO = *do.(*gen.DO)
return w
}

View File

@@ -0,0 +1,485 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowExecution(db *gorm.DB, opts ...gen.DOOption) workflowExecution {
_workflowExecution := workflowExecution{}
_workflowExecution.workflowExecutionDo.UseDB(db, opts...)
_workflowExecution.workflowExecutionDo.UseModel(&model.WorkflowExecution{})
tableName := _workflowExecution.workflowExecutionDo.TableName()
_workflowExecution.ALL = field.NewAsterisk(tableName)
_workflowExecution.ID = field.NewInt64(tableName, "id")
_workflowExecution.WorkflowID = field.NewInt64(tableName, "workflow_id")
_workflowExecution.Version = field.NewString(tableName, "version")
_workflowExecution.SpaceID = field.NewInt64(tableName, "space_id")
_workflowExecution.Mode = field.NewInt32(tableName, "mode")
_workflowExecution.OperatorID = field.NewInt64(tableName, "operator_id")
_workflowExecution.ConnectorID = field.NewInt64(tableName, "connector_id")
_workflowExecution.ConnectorUID = field.NewString(tableName, "connector_uid")
_workflowExecution.CreatedAt = field.NewInt64(tableName, "created_at")
_workflowExecution.LogID = field.NewString(tableName, "log_id")
_workflowExecution.Status = field.NewInt32(tableName, "status")
_workflowExecution.Duration = field.NewInt64(tableName, "duration")
_workflowExecution.Input = field.NewString(tableName, "input")
_workflowExecution.Output = field.NewString(tableName, "output")
_workflowExecution.ErrorCode = field.NewString(tableName, "error_code")
_workflowExecution.FailReason = field.NewString(tableName, "fail_reason")
_workflowExecution.InputTokens = field.NewInt64(tableName, "input_tokens")
_workflowExecution.OutputTokens = field.NewInt64(tableName, "output_tokens")
_workflowExecution.UpdatedAt = field.NewInt64(tableName, "updated_at")
_workflowExecution.RootExecutionID = field.NewInt64(tableName, "root_execution_id")
_workflowExecution.ParentNodeID = field.NewString(tableName, "parent_node_id")
_workflowExecution.AppID = field.NewInt64(tableName, "app_id")
_workflowExecution.NodeCount = field.NewInt32(tableName, "node_count")
_workflowExecution.ResumeEventID = field.NewInt64(tableName, "resume_event_id")
_workflowExecution.AgentID = field.NewInt64(tableName, "agent_id")
_workflowExecution.SyncPattern = field.NewInt32(tableName, "sync_pattern")
_workflowExecution.CommitID = field.NewString(tableName, "commit_id")
_workflowExecution.fillFieldMap()
return _workflowExecution
}
// workflowExecution workflow 执行记录表用于记录每次workflow执行时的状态
type workflowExecution struct {
workflowExecutionDo
ALL field.Asterisk
ID field.Int64 // execute id
WorkflowID field.Int64 // workflow_id
Version field.String // workflow version. empty if is draft
SpaceID field.Int64 // the space id the workflow belongs to
Mode field.Int32 // the execution mode: 1. debug run 2. release run 3. node debug
OperatorID field.Int64 // the user id that runs this workflow
ConnectorID field.Int64 // the connector on which this execution happened
ConnectorUID field.String // user id of the connector
CreatedAt field.Int64 // create time in millisecond
LogID field.String // log id
Status field.Int32 // 1=running 2=success 3=fail 4=interrupted
Duration field.Int64 // execution duration in millisecond
Input field.String // actual input of this execution
Output field.String // the actual output of this execution
ErrorCode field.String // error code if any
FailReason field.String // the reason for failure
InputTokens field.Int64 // number of input tokens
OutputTokens field.Int64 // number of output tokens
UpdatedAt field.Int64 // update time in millisecond
RootExecutionID field.Int64 // the top level execution id. Null if this is the root
ParentNodeID field.String // the node key for the sub_workflow node that executes this workflow
AppID field.Int64 // app id this workflow execution belongs to
NodeCount field.Int32 // the total node count of the workflow
ResumeEventID field.Int64 // the current event ID which is resuming
AgentID field.Int64 // the agent that this execution binds to
SyncPattern field.Int32 // the sync pattern 1. sync 2. async 3. stream
CommitID field.String // draft commit id this execution belongs to
fieldMap map[string]field.Expr
}
func (w workflowExecution) Table(newTableName string) *workflowExecution {
w.workflowExecutionDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowExecution) As(alias string) *workflowExecution {
w.workflowExecutionDo.DO = *(w.workflowExecutionDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowExecution) updateTableName(table string) *workflowExecution {
w.ALL = field.NewAsterisk(table)
w.ID = field.NewInt64(table, "id")
w.WorkflowID = field.NewInt64(table, "workflow_id")
w.Version = field.NewString(table, "version")
w.SpaceID = field.NewInt64(table, "space_id")
w.Mode = field.NewInt32(table, "mode")
w.OperatorID = field.NewInt64(table, "operator_id")
w.ConnectorID = field.NewInt64(table, "connector_id")
w.ConnectorUID = field.NewString(table, "connector_uid")
w.CreatedAt = field.NewInt64(table, "created_at")
w.LogID = field.NewString(table, "log_id")
w.Status = field.NewInt32(table, "status")
w.Duration = field.NewInt64(table, "duration")
w.Input = field.NewString(table, "input")
w.Output = field.NewString(table, "output")
w.ErrorCode = field.NewString(table, "error_code")
w.FailReason = field.NewString(table, "fail_reason")
w.InputTokens = field.NewInt64(table, "input_tokens")
w.OutputTokens = field.NewInt64(table, "output_tokens")
w.UpdatedAt = field.NewInt64(table, "updated_at")
w.RootExecutionID = field.NewInt64(table, "root_execution_id")
w.ParentNodeID = field.NewString(table, "parent_node_id")
w.AppID = field.NewInt64(table, "app_id")
w.NodeCount = field.NewInt32(table, "node_count")
w.ResumeEventID = field.NewInt64(table, "resume_event_id")
w.AgentID = field.NewInt64(table, "agent_id")
w.SyncPattern = field.NewInt32(table, "sync_pattern")
w.CommitID = field.NewString(table, "commit_id")
w.fillFieldMap()
return w
}
func (w *workflowExecution) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowExecution) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 27)
w.fieldMap["id"] = w.ID
w.fieldMap["workflow_id"] = w.WorkflowID
w.fieldMap["version"] = w.Version
w.fieldMap["space_id"] = w.SpaceID
w.fieldMap["mode"] = w.Mode
w.fieldMap["operator_id"] = w.OperatorID
w.fieldMap["connector_id"] = w.ConnectorID
w.fieldMap["connector_uid"] = w.ConnectorUID
w.fieldMap["created_at"] = w.CreatedAt
w.fieldMap["log_id"] = w.LogID
w.fieldMap["status"] = w.Status
w.fieldMap["duration"] = w.Duration
w.fieldMap["input"] = w.Input
w.fieldMap["output"] = w.Output
w.fieldMap["error_code"] = w.ErrorCode
w.fieldMap["fail_reason"] = w.FailReason
w.fieldMap["input_tokens"] = w.InputTokens
w.fieldMap["output_tokens"] = w.OutputTokens
w.fieldMap["updated_at"] = w.UpdatedAt
w.fieldMap["root_execution_id"] = w.RootExecutionID
w.fieldMap["parent_node_id"] = w.ParentNodeID
w.fieldMap["app_id"] = w.AppID
w.fieldMap["node_count"] = w.NodeCount
w.fieldMap["resume_event_id"] = w.ResumeEventID
w.fieldMap["agent_id"] = w.AgentID
w.fieldMap["sync_pattern"] = w.SyncPattern
w.fieldMap["commit_id"] = w.CommitID
}
func (w workflowExecution) clone(db *gorm.DB) workflowExecution {
w.workflowExecutionDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowExecution) replaceDB(db *gorm.DB) workflowExecution {
w.workflowExecutionDo.ReplaceDB(db)
return w
}
type workflowExecutionDo struct{ gen.DO }
type IWorkflowExecutionDo interface {
gen.SubQuery
Debug() IWorkflowExecutionDo
WithContext(ctx context.Context) IWorkflowExecutionDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowExecutionDo
WriteDB() IWorkflowExecutionDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowExecutionDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowExecutionDo
Not(conds ...gen.Condition) IWorkflowExecutionDo
Or(conds ...gen.Condition) IWorkflowExecutionDo
Select(conds ...field.Expr) IWorkflowExecutionDo
Where(conds ...gen.Condition) IWorkflowExecutionDo
Order(conds ...field.Expr) IWorkflowExecutionDo
Distinct(cols ...field.Expr) IWorkflowExecutionDo
Omit(cols ...field.Expr) IWorkflowExecutionDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo
Group(cols ...field.Expr) IWorkflowExecutionDo
Having(conds ...gen.Condition) IWorkflowExecutionDo
Limit(limit int) IWorkflowExecutionDo
Offset(offset int) IWorkflowExecutionDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowExecutionDo
Unscoped() IWorkflowExecutionDo
Create(values ...*model.WorkflowExecution) error
CreateInBatches(values []*model.WorkflowExecution, batchSize int) error
Save(values ...*model.WorkflowExecution) error
First() (*model.WorkflowExecution, error)
Take() (*model.WorkflowExecution, error)
Last() (*model.WorkflowExecution, error)
Find() ([]*model.WorkflowExecution, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowExecution, err error)
FindInBatches(result *[]*model.WorkflowExecution, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowExecution) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowExecutionDo
Assign(attrs ...field.AssignExpr) IWorkflowExecutionDo
Joins(fields ...field.RelationField) IWorkflowExecutionDo
Preload(fields ...field.RelationField) IWorkflowExecutionDo
FirstOrInit() (*model.WorkflowExecution, error)
FirstOrCreate() (*model.WorkflowExecution, error)
FindByPage(offset int, limit int) (result []*model.WorkflowExecution, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowExecutionDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowExecutionDo) Debug() IWorkflowExecutionDo {
return w.withDO(w.DO.Debug())
}
func (w workflowExecutionDo) WithContext(ctx context.Context) IWorkflowExecutionDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowExecutionDo) ReadDB() IWorkflowExecutionDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowExecutionDo) WriteDB() IWorkflowExecutionDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowExecutionDo) Session(config *gorm.Session) IWorkflowExecutionDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowExecutionDo) Clauses(conds ...clause.Expression) IWorkflowExecutionDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowExecutionDo) Returning(value interface{}, columns ...string) IWorkflowExecutionDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowExecutionDo) Not(conds ...gen.Condition) IWorkflowExecutionDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowExecutionDo) Or(conds ...gen.Condition) IWorkflowExecutionDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowExecutionDo) Select(conds ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowExecutionDo) Where(conds ...gen.Condition) IWorkflowExecutionDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowExecutionDo) Order(conds ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowExecutionDo) Distinct(cols ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowExecutionDo) Omit(cols ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowExecutionDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowExecutionDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowExecutionDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowExecutionDo) Group(cols ...field.Expr) IWorkflowExecutionDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowExecutionDo) Having(conds ...gen.Condition) IWorkflowExecutionDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowExecutionDo) Limit(limit int) IWorkflowExecutionDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowExecutionDo) Offset(offset int) IWorkflowExecutionDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowExecutionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowExecutionDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowExecutionDo) Unscoped() IWorkflowExecutionDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowExecutionDo) Create(values ...*model.WorkflowExecution) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowExecutionDo) CreateInBatches(values []*model.WorkflowExecution, batchSize int) error {
return w.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 (w workflowExecutionDo) Save(values ...*model.WorkflowExecution) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowExecutionDo) First() (*model.WorkflowExecution, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowExecution), nil
}
}
func (w workflowExecutionDo) Take() (*model.WorkflowExecution, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowExecution), nil
}
}
func (w workflowExecutionDo) Last() (*model.WorkflowExecution, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowExecution), nil
}
}
func (w workflowExecutionDo) Find() ([]*model.WorkflowExecution, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowExecution), err
}
func (w workflowExecutionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowExecution, err error) {
buf := make([]*model.WorkflowExecution, 0, batchSize)
err = w.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 (w workflowExecutionDo) FindInBatches(result *[]*model.WorkflowExecution, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowExecutionDo) Attrs(attrs ...field.AssignExpr) IWorkflowExecutionDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowExecutionDo) Assign(attrs ...field.AssignExpr) IWorkflowExecutionDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowExecutionDo) Joins(fields ...field.RelationField) IWorkflowExecutionDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowExecutionDo) Preload(fields ...field.RelationField) IWorkflowExecutionDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowExecutionDo) FirstOrInit() (*model.WorkflowExecution, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowExecution), nil
}
}
func (w workflowExecutionDo) FirstOrCreate() (*model.WorkflowExecution, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowExecution), nil
}
}
func (w workflowExecutionDo) FindByPage(offset int, limit int) (result []*model.WorkflowExecution, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowExecutionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowExecutionDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowExecutionDo) Delete(models ...*model.WorkflowExecution) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowExecutionDo) withDO(do gen.Dao) *workflowExecutionDo {
w.DO = *do.(*gen.DO)
return w
}

View File

@@ -0,0 +1,453 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowMeta(db *gorm.DB, opts ...gen.DOOption) workflowMeta {
_workflowMeta := workflowMeta{}
_workflowMeta.workflowMetaDo.UseDB(db, opts...)
_workflowMeta.workflowMetaDo.UseModel(&model.WorkflowMeta{})
tableName := _workflowMeta.workflowMetaDo.TableName()
_workflowMeta.ALL = field.NewAsterisk(tableName)
_workflowMeta.ID = field.NewInt64(tableName, "id")
_workflowMeta.Name = field.NewString(tableName, "name")
_workflowMeta.Description = field.NewString(tableName, "description")
_workflowMeta.IconURI = field.NewString(tableName, "icon_uri")
_workflowMeta.Status = field.NewInt32(tableName, "status")
_workflowMeta.ContentType = field.NewInt32(tableName, "content_type")
_workflowMeta.Mode = field.NewInt32(tableName, "mode")
_workflowMeta.CreatedAt = field.NewInt64(tableName, "created_at")
_workflowMeta.UpdatedAt = field.NewInt64(tableName, "updated_at")
_workflowMeta.DeletedAt = field.NewField(tableName, "deleted_at")
_workflowMeta.CreatorID = field.NewInt64(tableName, "creator_id")
_workflowMeta.Tag = field.NewInt32(tableName, "tag")
_workflowMeta.AuthorID = field.NewInt64(tableName, "author_id")
_workflowMeta.SpaceID = field.NewInt64(tableName, "space_id")
_workflowMeta.UpdaterID = field.NewInt64(tableName, "updater_id")
_workflowMeta.SourceID = field.NewInt64(tableName, "source_id")
_workflowMeta.AppID = field.NewInt64(tableName, "app_id")
_workflowMeta.LatestVersion = field.NewString(tableName, "latest_version")
_workflowMeta.LatestVersionTs = field.NewInt64(tableName, "latest_version_ts")
_workflowMeta.fillFieldMap()
return _workflowMeta
}
// workflowMeta workflow 元信息表用于记录workflow基本的元信息
type workflowMeta struct {
workflowMetaDo
ALL field.Asterisk
ID field.Int64 // workflow id
Name field.String // workflow name
Description field.String // workflow description
IconURI field.String // icon uri
Status field.Int32 // 0:未发布过, 1:已发布过
ContentType field.Int32 // 0用户 1官方
Mode field.Int32 // 0:workflow, 3:chat_flow
CreatedAt field.Int64 // create time in millisecond
UpdatedAt field.Int64 // update time in millisecond
DeletedAt field.Field // delete time in millisecond
CreatorID field.Int64 // user id for creator
Tag field.Int32 // template tag: Tag: 1=All, 2=Hot, 3=Information, 4=Music, 5=Picture, 6=UtilityTool, 7=Life, 8=Traval, 9=Network, 10=System, 11=Movie, 12=Office, 13=Shopping, 14=Education, 15=Health, 16=Social, 17=Entertainment, 18=Finance, 100=Hidden
AuthorID field.Int64 // 原作者用户 ID
SpaceID field.Int64 // 空间 ID
UpdaterID field.Int64 // 更新元信息的用户 ID
SourceID field.Int64 // 复制来源的 workflow ID
AppID field.Int64 // 应用 ID
LatestVersion field.String // the version of the most recent publish
LatestVersionTs field.Int64 // create time of latest version
fieldMap map[string]field.Expr
}
func (w workflowMeta) Table(newTableName string) *workflowMeta {
w.workflowMetaDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowMeta) As(alias string) *workflowMeta {
w.workflowMetaDo.DO = *(w.workflowMetaDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowMeta) updateTableName(table string) *workflowMeta {
w.ALL = field.NewAsterisk(table)
w.ID = field.NewInt64(table, "id")
w.Name = field.NewString(table, "name")
w.Description = field.NewString(table, "description")
w.IconURI = field.NewString(table, "icon_uri")
w.Status = field.NewInt32(table, "status")
w.ContentType = field.NewInt32(table, "content_type")
w.Mode = field.NewInt32(table, "mode")
w.CreatedAt = field.NewInt64(table, "created_at")
w.UpdatedAt = field.NewInt64(table, "updated_at")
w.DeletedAt = field.NewField(table, "deleted_at")
w.CreatorID = field.NewInt64(table, "creator_id")
w.Tag = field.NewInt32(table, "tag")
w.AuthorID = field.NewInt64(table, "author_id")
w.SpaceID = field.NewInt64(table, "space_id")
w.UpdaterID = field.NewInt64(table, "updater_id")
w.SourceID = field.NewInt64(table, "source_id")
w.AppID = field.NewInt64(table, "app_id")
w.LatestVersion = field.NewString(table, "latest_version")
w.LatestVersionTs = field.NewInt64(table, "latest_version_ts")
w.fillFieldMap()
return w
}
func (w *workflowMeta) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowMeta) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 19)
w.fieldMap["id"] = w.ID
w.fieldMap["name"] = w.Name
w.fieldMap["description"] = w.Description
w.fieldMap["icon_uri"] = w.IconURI
w.fieldMap["status"] = w.Status
w.fieldMap["content_type"] = w.ContentType
w.fieldMap["mode"] = w.Mode
w.fieldMap["created_at"] = w.CreatedAt
w.fieldMap["updated_at"] = w.UpdatedAt
w.fieldMap["deleted_at"] = w.DeletedAt
w.fieldMap["creator_id"] = w.CreatorID
w.fieldMap["tag"] = w.Tag
w.fieldMap["author_id"] = w.AuthorID
w.fieldMap["space_id"] = w.SpaceID
w.fieldMap["updater_id"] = w.UpdaterID
w.fieldMap["source_id"] = w.SourceID
w.fieldMap["app_id"] = w.AppID
w.fieldMap["latest_version"] = w.LatestVersion
w.fieldMap["latest_version_ts"] = w.LatestVersionTs
}
func (w workflowMeta) clone(db *gorm.DB) workflowMeta {
w.workflowMetaDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowMeta) replaceDB(db *gorm.DB) workflowMeta {
w.workflowMetaDo.ReplaceDB(db)
return w
}
type workflowMetaDo struct{ gen.DO }
type IWorkflowMetaDo interface {
gen.SubQuery
Debug() IWorkflowMetaDo
WithContext(ctx context.Context) IWorkflowMetaDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowMetaDo
WriteDB() IWorkflowMetaDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowMetaDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowMetaDo
Not(conds ...gen.Condition) IWorkflowMetaDo
Or(conds ...gen.Condition) IWorkflowMetaDo
Select(conds ...field.Expr) IWorkflowMetaDo
Where(conds ...gen.Condition) IWorkflowMetaDo
Order(conds ...field.Expr) IWorkflowMetaDo
Distinct(cols ...field.Expr) IWorkflowMetaDo
Omit(cols ...field.Expr) IWorkflowMetaDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo
Group(cols ...field.Expr) IWorkflowMetaDo
Having(conds ...gen.Condition) IWorkflowMetaDo
Limit(limit int) IWorkflowMetaDo
Offset(offset int) IWorkflowMetaDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowMetaDo
Unscoped() IWorkflowMetaDo
Create(values ...*model.WorkflowMeta) error
CreateInBatches(values []*model.WorkflowMeta, batchSize int) error
Save(values ...*model.WorkflowMeta) error
First() (*model.WorkflowMeta, error)
Take() (*model.WorkflowMeta, error)
Last() (*model.WorkflowMeta, error)
Find() ([]*model.WorkflowMeta, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowMeta, err error)
FindInBatches(result *[]*model.WorkflowMeta, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowMeta) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowMetaDo
Assign(attrs ...field.AssignExpr) IWorkflowMetaDo
Joins(fields ...field.RelationField) IWorkflowMetaDo
Preload(fields ...field.RelationField) IWorkflowMetaDo
FirstOrInit() (*model.WorkflowMeta, error)
FirstOrCreate() (*model.WorkflowMeta, error)
FindByPage(offset int, limit int) (result []*model.WorkflowMeta, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowMetaDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowMetaDo) Debug() IWorkflowMetaDo {
return w.withDO(w.DO.Debug())
}
func (w workflowMetaDo) WithContext(ctx context.Context) IWorkflowMetaDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowMetaDo) ReadDB() IWorkflowMetaDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowMetaDo) WriteDB() IWorkflowMetaDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowMetaDo) Session(config *gorm.Session) IWorkflowMetaDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowMetaDo) Clauses(conds ...clause.Expression) IWorkflowMetaDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowMetaDo) Returning(value interface{}, columns ...string) IWorkflowMetaDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowMetaDo) Not(conds ...gen.Condition) IWorkflowMetaDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowMetaDo) Or(conds ...gen.Condition) IWorkflowMetaDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowMetaDo) Select(conds ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowMetaDo) Where(conds ...gen.Condition) IWorkflowMetaDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowMetaDo) Order(conds ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowMetaDo) Distinct(cols ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowMetaDo) Omit(cols ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowMetaDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowMetaDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowMetaDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowMetaDo) Group(cols ...field.Expr) IWorkflowMetaDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowMetaDo) Having(conds ...gen.Condition) IWorkflowMetaDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowMetaDo) Limit(limit int) IWorkflowMetaDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowMetaDo) Offset(offset int) IWorkflowMetaDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowMetaDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowMetaDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowMetaDo) Unscoped() IWorkflowMetaDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowMetaDo) Create(values ...*model.WorkflowMeta) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowMetaDo) CreateInBatches(values []*model.WorkflowMeta, batchSize int) error {
return w.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 (w workflowMetaDo) Save(values ...*model.WorkflowMeta) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowMetaDo) First() (*model.WorkflowMeta, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowMeta), nil
}
}
func (w workflowMetaDo) Take() (*model.WorkflowMeta, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowMeta), nil
}
}
func (w workflowMetaDo) Last() (*model.WorkflowMeta, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowMeta), nil
}
}
func (w workflowMetaDo) Find() ([]*model.WorkflowMeta, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowMeta), err
}
func (w workflowMetaDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowMeta, err error) {
buf := make([]*model.WorkflowMeta, 0, batchSize)
err = w.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 (w workflowMetaDo) FindInBatches(result *[]*model.WorkflowMeta, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowMetaDo) Attrs(attrs ...field.AssignExpr) IWorkflowMetaDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowMetaDo) Assign(attrs ...field.AssignExpr) IWorkflowMetaDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowMetaDo) Joins(fields ...field.RelationField) IWorkflowMetaDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowMetaDo) Preload(fields ...field.RelationField) IWorkflowMetaDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowMetaDo) FirstOrInit() (*model.WorkflowMeta, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowMeta), nil
}
}
func (w workflowMetaDo) FirstOrCreate() (*model.WorkflowMeta, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowMeta), nil
}
}
func (w workflowMetaDo) FindByPage(offset int, limit int) (result []*model.WorkflowMeta, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowMetaDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowMetaDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowMetaDo) Delete(models ...*model.WorkflowMeta) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowMetaDo) withDO(do gen.Dao) *workflowMetaDo {
w.DO = *do.(*gen.DO)
return w
}

View File

@@ -0,0 +1,409 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowReference(db *gorm.DB, opts ...gen.DOOption) workflowReference {
_workflowReference := workflowReference{}
_workflowReference.workflowReferenceDo.UseDB(db, opts...)
_workflowReference.workflowReferenceDo.UseModel(&model.WorkflowReference{})
tableName := _workflowReference.workflowReferenceDo.TableName()
_workflowReference.ALL = field.NewAsterisk(tableName)
_workflowReference.ID = field.NewInt64(tableName, "id")
_workflowReference.ReferredID = field.NewInt64(tableName, "referred_id")
_workflowReference.ReferringID = field.NewInt64(tableName, "referring_id")
_workflowReference.ReferType = field.NewInt32(tableName, "refer_type")
_workflowReference.ReferringBizType = field.NewInt32(tableName, "referring_biz_type")
_workflowReference.CreatedAt = field.NewInt64(tableName, "created_at")
_workflowReference.Status = field.NewInt32(tableName, "status")
_workflowReference.DeletedAt = field.NewField(tableName, "deleted_at")
_workflowReference.fillFieldMap()
return _workflowReference
}
// workflowReference workflow 关联关系表用于记录workflow 直接互相引用关系
type workflowReference struct {
workflowReferenceDo
ALL field.Asterisk
ID field.Int64 // workflow id
ReferredID field.Int64 // the id of the workflow that is referred by other entities
ReferringID field.Int64 // the entity id that refers this workflow
ReferType field.Int32 // 1 subworkflow 2 tool
ReferringBizType field.Int32 // the biz type the referring entity belongs to: 1. workflow 2. agent
CreatedAt field.Int64 // create time in millisecond
Status field.Int32 // whether this reference currently takes effect. 0: disabled 1: enabled
DeletedAt field.Field
fieldMap map[string]field.Expr
}
func (w workflowReference) Table(newTableName string) *workflowReference {
w.workflowReferenceDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowReference) As(alias string) *workflowReference {
w.workflowReferenceDo.DO = *(w.workflowReferenceDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowReference) updateTableName(table string) *workflowReference {
w.ALL = field.NewAsterisk(table)
w.ID = field.NewInt64(table, "id")
w.ReferredID = field.NewInt64(table, "referred_id")
w.ReferringID = field.NewInt64(table, "referring_id")
w.ReferType = field.NewInt32(table, "refer_type")
w.ReferringBizType = field.NewInt32(table, "referring_biz_type")
w.CreatedAt = field.NewInt64(table, "created_at")
w.Status = field.NewInt32(table, "status")
w.DeletedAt = field.NewField(table, "deleted_at")
w.fillFieldMap()
return w
}
func (w *workflowReference) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowReference) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 8)
w.fieldMap["id"] = w.ID
w.fieldMap["referred_id"] = w.ReferredID
w.fieldMap["referring_id"] = w.ReferringID
w.fieldMap["refer_type"] = w.ReferType
w.fieldMap["referring_biz_type"] = w.ReferringBizType
w.fieldMap["created_at"] = w.CreatedAt
w.fieldMap["status"] = w.Status
w.fieldMap["deleted_at"] = w.DeletedAt
}
func (w workflowReference) clone(db *gorm.DB) workflowReference {
w.workflowReferenceDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowReference) replaceDB(db *gorm.DB) workflowReference {
w.workflowReferenceDo.ReplaceDB(db)
return w
}
type workflowReferenceDo struct{ gen.DO }
type IWorkflowReferenceDo interface {
gen.SubQuery
Debug() IWorkflowReferenceDo
WithContext(ctx context.Context) IWorkflowReferenceDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowReferenceDo
WriteDB() IWorkflowReferenceDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowReferenceDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowReferenceDo
Not(conds ...gen.Condition) IWorkflowReferenceDo
Or(conds ...gen.Condition) IWorkflowReferenceDo
Select(conds ...field.Expr) IWorkflowReferenceDo
Where(conds ...gen.Condition) IWorkflowReferenceDo
Order(conds ...field.Expr) IWorkflowReferenceDo
Distinct(cols ...field.Expr) IWorkflowReferenceDo
Omit(cols ...field.Expr) IWorkflowReferenceDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo
Group(cols ...field.Expr) IWorkflowReferenceDo
Having(conds ...gen.Condition) IWorkflowReferenceDo
Limit(limit int) IWorkflowReferenceDo
Offset(offset int) IWorkflowReferenceDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowReferenceDo
Unscoped() IWorkflowReferenceDo
Create(values ...*model.WorkflowReference) error
CreateInBatches(values []*model.WorkflowReference, batchSize int) error
Save(values ...*model.WorkflowReference) error
First() (*model.WorkflowReference, error)
Take() (*model.WorkflowReference, error)
Last() (*model.WorkflowReference, error)
Find() ([]*model.WorkflowReference, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowReference, err error)
FindInBatches(result *[]*model.WorkflowReference, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowReference) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowReferenceDo
Assign(attrs ...field.AssignExpr) IWorkflowReferenceDo
Joins(fields ...field.RelationField) IWorkflowReferenceDo
Preload(fields ...field.RelationField) IWorkflowReferenceDo
FirstOrInit() (*model.WorkflowReference, error)
FirstOrCreate() (*model.WorkflowReference, error)
FindByPage(offset int, limit int) (result []*model.WorkflowReference, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowReferenceDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowReferenceDo) Debug() IWorkflowReferenceDo {
return w.withDO(w.DO.Debug())
}
func (w workflowReferenceDo) WithContext(ctx context.Context) IWorkflowReferenceDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowReferenceDo) ReadDB() IWorkflowReferenceDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowReferenceDo) WriteDB() IWorkflowReferenceDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowReferenceDo) Session(config *gorm.Session) IWorkflowReferenceDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowReferenceDo) Clauses(conds ...clause.Expression) IWorkflowReferenceDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowReferenceDo) Returning(value interface{}, columns ...string) IWorkflowReferenceDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowReferenceDo) Not(conds ...gen.Condition) IWorkflowReferenceDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowReferenceDo) Or(conds ...gen.Condition) IWorkflowReferenceDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowReferenceDo) Select(conds ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowReferenceDo) Where(conds ...gen.Condition) IWorkflowReferenceDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowReferenceDo) Order(conds ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowReferenceDo) Distinct(cols ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowReferenceDo) Omit(cols ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowReferenceDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowReferenceDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowReferenceDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowReferenceDo) Group(cols ...field.Expr) IWorkflowReferenceDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowReferenceDo) Having(conds ...gen.Condition) IWorkflowReferenceDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowReferenceDo) Limit(limit int) IWorkflowReferenceDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowReferenceDo) Offset(offset int) IWorkflowReferenceDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowReferenceDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowReferenceDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowReferenceDo) Unscoped() IWorkflowReferenceDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowReferenceDo) Create(values ...*model.WorkflowReference) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowReferenceDo) CreateInBatches(values []*model.WorkflowReference, batchSize int) error {
return w.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 (w workflowReferenceDo) Save(values ...*model.WorkflowReference) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowReferenceDo) First() (*model.WorkflowReference, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowReference), nil
}
}
func (w workflowReferenceDo) Take() (*model.WorkflowReference, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowReference), nil
}
}
func (w workflowReferenceDo) Last() (*model.WorkflowReference, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowReference), nil
}
}
func (w workflowReferenceDo) Find() ([]*model.WorkflowReference, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowReference), err
}
func (w workflowReferenceDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowReference, err error) {
buf := make([]*model.WorkflowReference, 0, batchSize)
err = w.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 (w workflowReferenceDo) FindInBatches(result *[]*model.WorkflowReference, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowReferenceDo) Attrs(attrs ...field.AssignExpr) IWorkflowReferenceDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowReferenceDo) Assign(attrs ...field.AssignExpr) IWorkflowReferenceDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowReferenceDo) Joins(fields ...field.RelationField) IWorkflowReferenceDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowReferenceDo) Preload(fields ...field.RelationField) IWorkflowReferenceDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowReferenceDo) FirstOrInit() (*model.WorkflowReference, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowReference), nil
}
}
func (w workflowReferenceDo) FirstOrCreate() (*model.WorkflowReference, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowReference), nil
}
}
func (w workflowReferenceDo) FindByPage(offset int, limit int) (result []*model.WorkflowReference, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowReferenceDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowReferenceDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowReferenceDo) Delete(models ...*model.WorkflowReference) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowReferenceDo) withDO(do gen.Dao) *workflowReferenceDo {
w.DO = *do.(*gen.DO)
return w
}

View File

@@ -0,0 +1,405 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowSnapshot(db *gorm.DB, opts ...gen.DOOption) workflowSnapshot {
_workflowSnapshot := workflowSnapshot{}
_workflowSnapshot.workflowSnapshotDo.UseDB(db, opts...)
_workflowSnapshot.workflowSnapshotDo.UseModel(&model.WorkflowSnapshot{})
tableName := _workflowSnapshot.workflowSnapshotDo.TableName()
_workflowSnapshot.ALL = field.NewAsterisk(tableName)
_workflowSnapshot.WorkflowID = field.NewInt64(tableName, "workflow_id")
_workflowSnapshot.CommitID = field.NewString(tableName, "commit_id")
_workflowSnapshot.Canvas = field.NewString(tableName, "canvas")
_workflowSnapshot.InputParams = field.NewString(tableName, "input_params")
_workflowSnapshot.OutputParams = field.NewString(tableName, "output_params")
_workflowSnapshot.CreatedAt = field.NewInt64(tableName, "created_at")
_workflowSnapshot.ID = field.NewInt64(tableName, "id")
_workflowSnapshot.fillFieldMap()
return _workflowSnapshot
}
// workflowSnapshot snapshot for executed workflow draft
type workflowSnapshot struct {
workflowSnapshotDo
ALL field.Asterisk
WorkflowID field.Int64 // workflow id this snapshot belongs to
CommitID field.String // the commit id of the workflow draft
Canvas field.String // frontend schema for this snapshot
InputParams field.String // input parameter info
OutputParams field.String // output parameter info
CreatedAt field.Int64
ID field.Int64 // ID
fieldMap map[string]field.Expr
}
func (w workflowSnapshot) Table(newTableName string) *workflowSnapshot {
w.workflowSnapshotDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowSnapshot) As(alias string) *workflowSnapshot {
w.workflowSnapshotDo.DO = *(w.workflowSnapshotDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowSnapshot) updateTableName(table string) *workflowSnapshot {
w.ALL = field.NewAsterisk(table)
w.WorkflowID = field.NewInt64(table, "workflow_id")
w.CommitID = field.NewString(table, "commit_id")
w.Canvas = field.NewString(table, "canvas")
w.InputParams = field.NewString(table, "input_params")
w.OutputParams = field.NewString(table, "output_params")
w.CreatedAt = field.NewInt64(table, "created_at")
w.ID = field.NewInt64(table, "id")
w.fillFieldMap()
return w
}
func (w *workflowSnapshot) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowSnapshot) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 7)
w.fieldMap["workflow_id"] = w.WorkflowID
w.fieldMap["commit_id"] = w.CommitID
w.fieldMap["canvas"] = w.Canvas
w.fieldMap["input_params"] = w.InputParams
w.fieldMap["output_params"] = w.OutputParams
w.fieldMap["created_at"] = w.CreatedAt
w.fieldMap["id"] = w.ID
}
func (w workflowSnapshot) clone(db *gorm.DB) workflowSnapshot {
w.workflowSnapshotDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowSnapshot) replaceDB(db *gorm.DB) workflowSnapshot {
w.workflowSnapshotDo.ReplaceDB(db)
return w
}
type workflowSnapshotDo struct{ gen.DO }
type IWorkflowSnapshotDo interface {
gen.SubQuery
Debug() IWorkflowSnapshotDo
WithContext(ctx context.Context) IWorkflowSnapshotDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowSnapshotDo
WriteDB() IWorkflowSnapshotDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowSnapshotDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowSnapshotDo
Not(conds ...gen.Condition) IWorkflowSnapshotDo
Or(conds ...gen.Condition) IWorkflowSnapshotDo
Select(conds ...field.Expr) IWorkflowSnapshotDo
Where(conds ...gen.Condition) IWorkflowSnapshotDo
Order(conds ...field.Expr) IWorkflowSnapshotDo
Distinct(cols ...field.Expr) IWorkflowSnapshotDo
Omit(cols ...field.Expr) IWorkflowSnapshotDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo
Group(cols ...field.Expr) IWorkflowSnapshotDo
Having(conds ...gen.Condition) IWorkflowSnapshotDo
Limit(limit int) IWorkflowSnapshotDo
Offset(offset int) IWorkflowSnapshotDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowSnapshotDo
Unscoped() IWorkflowSnapshotDo
Create(values ...*model.WorkflowSnapshot) error
CreateInBatches(values []*model.WorkflowSnapshot, batchSize int) error
Save(values ...*model.WorkflowSnapshot) error
First() (*model.WorkflowSnapshot, error)
Take() (*model.WorkflowSnapshot, error)
Last() (*model.WorkflowSnapshot, error)
Find() ([]*model.WorkflowSnapshot, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowSnapshot, err error)
FindInBatches(result *[]*model.WorkflowSnapshot, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowSnapshot) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowSnapshotDo
Assign(attrs ...field.AssignExpr) IWorkflowSnapshotDo
Joins(fields ...field.RelationField) IWorkflowSnapshotDo
Preload(fields ...field.RelationField) IWorkflowSnapshotDo
FirstOrInit() (*model.WorkflowSnapshot, error)
FirstOrCreate() (*model.WorkflowSnapshot, error)
FindByPage(offset int, limit int) (result []*model.WorkflowSnapshot, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowSnapshotDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowSnapshotDo) Debug() IWorkflowSnapshotDo {
return w.withDO(w.DO.Debug())
}
func (w workflowSnapshotDo) WithContext(ctx context.Context) IWorkflowSnapshotDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowSnapshotDo) ReadDB() IWorkflowSnapshotDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowSnapshotDo) WriteDB() IWorkflowSnapshotDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowSnapshotDo) Session(config *gorm.Session) IWorkflowSnapshotDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowSnapshotDo) Clauses(conds ...clause.Expression) IWorkflowSnapshotDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowSnapshotDo) Returning(value interface{}, columns ...string) IWorkflowSnapshotDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowSnapshotDo) Not(conds ...gen.Condition) IWorkflowSnapshotDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowSnapshotDo) Or(conds ...gen.Condition) IWorkflowSnapshotDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowSnapshotDo) Select(conds ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowSnapshotDo) Where(conds ...gen.Condition) IWorkflowSnapshotDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowSnapshotDo) Order(conds ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowSnapshotDo) Distinct(cols ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowSnapshotDo) Omit(cols ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowSnapshotDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowSnapshotDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowSnapshotDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowSnapshotDo) Group(cols ...field.Expr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowSnapshotDo) Having(conds ...gen.Condition) IWorkflowSnapshotDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowSnapshotDo) Limit(limit int) IWorkflowSnapshotDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowSnapshotDo) Offset(offset int) IWorkflowSnapshotDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowSnapshotDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowSnapshotDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowSnapshotDo) Unscoped() IWorkflowSnapshotDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowSnapshotDo) Create(values ...*model.WorkflowSnapshot) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowSnapshotDo) CreateInBatches(values []*model.WorkflowSnapshot, batchSize int) error {
return w.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 (w workflowSnapshotDo) Save(values ...*model.WorkflowSnapshot) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowSnapshotDo) First() (*model.WorkflowSnapshot, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowSnapshot), nil
}
}
func (w workflowSnapshotDo) Take() (*model.WorkflowSnapshot, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowSnapshot), nil
}
}
func (w workflowSnapshotDo) Last() (*model.WorkflowSnapshot, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowSnapshot), nil
}
}
func (w workflowSnapshotDo) Find() ([]*model.WorkflowSnapshot, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowSnapshot), err
}
func (w workflowSnapshotDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowSnapshot, err error) {
buf := make([]*model.WorkflowSnapshot, 0, batchSize)
err = w.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 (w workflowSnapshotDo) FindInBatches(result *[]*model.WorkflowSnapshot, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowSnapshotDo) Attrs(attrs ...field.AssignExpr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowSnapshotDo) Assign(attrs ...field.AssignExpr) IWorkflowSnapshotDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowSnapshotDo) Joins(fields ...field.RelationField) IWorkflowSnapshotDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowSnapshotDo) Preload(fields ...field.RelationField) IWorkflowSnapshotDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowSnapshotDo) FirstOrInit() (*model.WorkflowSnapshot, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowSnapshot), nil
}
}
func (w workflowSnapshotDo) FirstOrCreate() (*model.WorkflowSnapshot, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowSnapshot), nil
}
}
func (w workflowSnapshotDo) FindByPage(offset int, limit int) (result []*model.WorkflowSnapshot, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowSnapshotDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowSnapshotDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowSnapshotDo) Delete(models ...*model.WorkflowSnapshot) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowSnapshotDo) withDO(do gen.Dao) *workflowSnapshotDo {
w.DO = *do.(*gen.DO)
return w
}

View File

@@ -0,0 +1,421 @@
// 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"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"gorm.io/gen"
"gorm.io/gen/field"
"gorm.io/plugin/dbresolver"
"github.com/coze-dev/coze-studio/backend/domain/workflow/internal/repo/dal/model"
)
func newWorkflowVersion(db *gorm.DB, opts ...gen.DOOption) workflowVersion {
_workflowVersion := workflowVersion{}
_workflowVersion.workflowVersionDo.UseDB(db, opts...)
_workflowVersion.workflowVersionDo.UseModel(&model.WorkflowVersion{})
tableName := _workflowVersion.workflowVersionDo.TableName()
_workflowVersion.ALL = field.NewAsterisk(tableName)
_workflowVersion.ID = field.NewInt64(tableName, "id")
_workflowVersion.WorkflowID = field.NewInt64(tableName, "workflow_id")
_workflowVersion.Version = field.NewString(tableName, "version")
_workflowVersion.VersionDescription = field.NewString(tableName, "version_description")
_workflowVersion.Canvas = field.NewString(tableName, "canvas")
_workflowVersion.InputParams = field.NewString(tableName, "input_params")
_workflowVersion.OutputParams = field.NewString(tableName, "output_params")
_workflowVersion.CreatorID = field.NewInt64(tableName, "creator_id")
_workflowVersion.CreatedAt = field.NewInt64(tableName, "created_at")
_workflowVersion.DeletedAt = field.NewField(tableName, "deleted_at")
_workflowVersion.CommitID = field.NewString(tableName, "commit_id")
_workflowVersion.fillFieldMap()
return _workflowVersion
}
// workflowVersion workflow 画布版本信息表,用于记录不同版本的画布信息
type workflowVersion struct {
workflowVersionDo
ALL field.Asterisk
ID field.Int64 // ID
WorkflowID field.Int64 // workflow id
Version field.String // 发布版本
VersionDescription field.String // 版本描述
Canvas field.String // 前端 schema
InputParams field.String
OutputParams field.String
CreatorID field.Int64 // 发布用户 ID
CreatedAt field.Int64 // 创建时间毫秒时间戳
DeletedAt field.Field // 删除毫秒时间戳
CommitID field.String // the commit id corresponding to this version
fieldMap map[string]field.Expr
}
func (w workflowVersion) Table(newTableName string) *workflowVersion {
w.workflowVersionDo.UseTable(newTableName)
return w.updateTableName(newTableName)
}
func (w workflowVersion) As(alias string) *workflowVersion {
w.workflowVersionDo.DO = *(w.workflowVersionDo.As(alias).(*gen.DO))
return w.updateTableName(alias)
}
func (w *workflowVersion) updateTableName(table string) *workflowVersion {
w.ALL = field.NewAsterisk(table)
w.ID = field.NewInt64(table, "id")
w.WorkflowID = field.NewInt64(table, "workflow_id")
w.Version = field.NewString(table, "version")
w.VersionDescription = field.NewString(table, "version_description")
w.Canvas = field.NewString(table, "canvas")
w.InputParams = field.NewString(table, "input_params")
w.OutputParams = field.NewString(table, "output_params")
w.CreatorID = field.NewInt64(table, "creator_id")
w.CreatedAt = field.NewInt64(table, "created_at")
w.DeletedAt = field.NewField(table, "deleted_at")
w.CommitID = field.NewString(table, "commit_id")
w.fillFieldMap()
return w
}
func (w *workflowVersion) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := w.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (w *workflowVersion) fillFieldMap() {
w.fieldMap = make(map[string]field.Expr, 11)
w.fieldMap["id"] = w.ID
w.fieldMap["workflow_id"] = w.WorkflowID
w.fieldMap["version"] = w.Version
w.fieldMap["version_description"] = w.VersionDescription
w.fieldMap["canvas"] = w.Canvas
w.fieldMap["input_params"] = w.InputParams
w.fieldMap["output_params"] = w.OutputParams
w.fieldMap["creator_id"] = w.CreatorID
w.fieldMap["created_at"] = w.CreatedAt
w.fieldMap["deleted_at"] = w.DeletedAt
w.fieldMap["commit_id"] = w.CommitID
}
func (w workflowVersion) clone(db *gorm.DB) workflowVersion {
w.workflowVersionDo.ReplaceConnPool(db.Statement.ConnPool)
return w
}
func (w workflowVersion) replaceDB(db *gorm.DB) workflowVersion {
w.workflowVersionDo.ReplaceDB(db)
return w
}
type workflowVersionDo struct{ gen.DO }
type IWorkflowVersionDo interface {
gen.SubQuery
Debug() IWorkflowVersionDo
WithContext(ctx context.Context) IWorkflowVersionDo
WithResult(fc func(tx gen.Dao)) gen.ResultInfo
ReplaceDB(db *gorm.DB)
ReadDB() IWorkflowVersionDo
WriteDB() IWorkflowVersionDo
As(alias string) gen.Dao
Session(config *gorm.Session) IWorkflowVersionDo
Columns(cols ...field.Expr) gen.Columns
Clauses(conds ...clause.Expression) IWorkflowVersionDo
Not(conds ...gen.Condition) IWorkflowVersionDo
Or(conds ...gen.Condition) IWorkflowVersionDo
Select(conds ...field.Expr) IWorkflowVersionDo
Where(conds ...gen.Condition) IWorkflowVersionDo
Order(conds ...field.Expr) IWorkflowVersionDo
Distinct(cols ...field.Expr) IWorkflowVersionDo
Omit(cols ...field.Expr) IWorkflowVersionDo
Join(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo
LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo
RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo
Group(cols ...field.Expr) IWorkflowVersionDo
Having(conds ...gen.Condition) IWorkflowVersionDo
Limit(limit int) IWorkflowVersionDo
Offset(offset int) IWorkflowVersionDo
Count() (count int64, err error)
Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowVersionDo
Unscoped() IWorkflowVersionDo
Create(values ...*model.WorkflowVersion) error
CreateInBatches(values []*model.WorkflowVersion, batchSize int) error
Save(values ...*model.WorkflowVersion) error
First() (*model.WorkflowVersion, error)
Take() (*model.WorkflowVersion, error)
Last() (*model.WorkflowVersion, error)
Find() ([]*model.WorkflowVersion, error)
FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowVersion, err error)
FindInBatches(result *[]*model.WorkflowVersion, batchSize int, fc func(tx gen.Dao, batch int) error) error
Pluck(column field.Expr, dest interface{}) error
Delete(...*model.WorkflowVersion) (info gen.ResultInfo, err error)
Update(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
Updates(value interface{}) (info gen.ResultInfo, err error)
UpdateColumn(column field.Expr, value interface{}) (info gen.ResultInfo, err error)
UpdateColumnSimple(columns ...field.AssignExpr) (info gen.ResultInfo, err error)
UpdateColumns(value interface{}) (info gen.ResultInfo, err error)
UpdateFrom(q gen.SubQuery) gen.Dao
Attrs(attrs ...field.AssignExpr) IWorkflowVersionDo
Assign(attrs ...field.AssignExpr) IWorkflowVersionDo
Joins(fields ...field.RelationField) IWorkflowVersionDo
Preload(fields ...field.RelationField) IWorkflowVersionDo
FirstOrInit() (*model.WorkflowVersion, error)
FirstOrCreate() (*model.WorkflowVersion, error)
FindByPage(offset int, limit int) (result []*model.WorkflowVersion, count int64, err error)
ScanByPage(result interface{}, offset int, limit int) (count int64, err error)
Scan(result interface{}) (err error)
Returning(value interface{}, columns ...string) IWorkflowVersionDo
UnderlyingDB() *gorm.DB
schema.Tabler
}
func (w workflowVersionDo) Debug() IWorkflowVersionDo {
return w.withDO(w.DO.Debug())
}
func (w workflowVersionDo) WithContext(ctx context.Context) IWorkflowVersionDo {
return w.withDO(w.DO.WithContext(ctx))
}
func (w workflowVersionDo) ReadDB() IWorkflowVersionDo {
return w.Clauses(dbresolver.Read)
}
func (w workflowVersionDo) WriteDB() IWorkflowVersionDo {
return w.Clauses(dbresolver.Write)
}
func (w workflowVersionDo) Session(config *gorm.Session) IWorkflowVersionDo {
return w.withDO(w.DO.Session(config))
}
func (w workflowVersionDo) Clauses(conds ...clause.Expression) IWorkflowVersionDo {
return w.withDO(w.DO.Clauses(conds...))
}
func (w workflowVersionDo) Returning(value interface{}, columns ...string) IWorkflowVersionDo {
return w.withDO(w.DO.Returning(value, columns...))
}
func (w workflowVersionDo) Not(conds ...gen.Condition) IWorkflowVersionDo {
return w.withDO(w.DO.Not(conds...))
}
func (w workflowVersionDo) Or(conds ...gen.Condition) IWorkflowVersionDo {
return w.withDO(w.DO.Or(conds...))
}
func (w workflowVersionDo) Select(conds ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Select(conds...))
}
func (w workflowVersionDo) Where(conds ...gen.Condition) IWorkflowVersionDo {
return w.withDO(w.DO.Where(conds...))
}
func (w workflowVersionDo) Order(conds ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Order(conds...))
}
func (w workflowVersionDo) Distinct(cols ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Distinct(cols...))
}
func (w workflowVersionDo) Omit(cols ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Omit(cols...))
}
func (w workflowVersionDo) Join(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Join(table, on...))
}
func (w workflowVersionDo) LeftJoin(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.LeftJoin(table, on...))
}
func (w workflowVersionDo) RightJoin(table schema.Tabler, on ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.RightJoin(table, on...))
}
func (w workflowVersionDo) Group(cols ...field.Expr) IWorkflowVersionDo {
return w.withDO(w.DO.Group(cols...))
}
func (w workflowVersionDo) Having(conds ...gen.Condition) IWorkflowVersionDo {
return w.withDO(w.DO.Having(conds...))
}
func (w workflowVersionDo) Limit(limit int) IWorkflowVersionDo {
return w.withDO(w.DO.Limit(limit))
}
func (w workflowVersionDo) Offset(offset int) IWorkflowVersionDo {
return w.withDO(w.DO.Offset(offset))
}
func (w workflowVersionDo) Scopes(funcs ...func(gen.Dao) gen.Dao) IWorkflowVersionDo {
return w.withDO(w.DO.Scopes(funcs...))
}
func (w workflowVersionDo) Unscoped() IWorkflowVersionDo {
return w.withDO(w.DO.Unscoped())
}
func (w workflowVersionDo) Create(values ...*model.WorkflowVersion) error {
if len(values) == 0 {
return nil
}
return w.DO.Create(values)
}
func (w workflowVersionDo) CreateInBatches(values []*model.WorkflowVersion, batchSize int) error {
return w.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 (w workflowVersionDo) Save(values ...*model.WorkflowVersion) error {
if len(values) == 0 {
return nil
}
return w.DO.Save(values)
}
func (w workflowVersionDo) First() (*model.WorkflowVersion, error) {
if result, err := w.DO.First(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowVersion), nil
}
}
func (w workflowVersionDo) Take() (*model.WorkflowVersion, error) {
if result, err := w.DO.Take(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowVersion), nil
}
}
func (w workflowVersionDo) Last() (*model.WorkflowVersion, error) {
if result, err := w.DO.Last(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowVersion), nil
}
}
func (w workflowVersionDo) Find() ([]*model.WorkflowVersion, error) {
result, err := w.DO.Find()
return result.([]*model.WorkflowVersion), err
}
func (w workflowVersionDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.WorkflowVersion, err error) {
buf := make([]*model.WorkflowVersion, 0, batchSize)
err = w.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 (w workflowVersionDo) FindInBatches(result *[]*model.WorkflowVersion, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return w.DO.FindInBatches(result, batchSize, fc)
}
func (w workflowVersionDo) Attrs(attrs ...field.AssignExpr) IWorkflowVersionDo {
return w.withDO(w.DO.Attrs(attrs...))
}
func (w workflowVersionDo) Assign(attrs ...field.AssignExpr) IWorkflowVersionDo {
return w.withDO(w.DO.Assign(attrs...))
}
func (w workflowVersionDo) Joins(fields ...field.RelationField) IWorkflowVersionDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Joins(_f))
}
return &w
}
func (w workflowVersionDo) Preload(fields ...field.RelationField) IWorkflowVersionDo {
for _, _f := range fields {
w = *w.withDO(w.DO.Preload(_f))
}
return &w
}
func (w workflowVersionDo) FirstOrInit() (*model.WorkflowVersion, error) {
if result, err := w.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowVersion), nil
}
}
func (w workflowVersionDo) FirstOrCreate() (*model.WorkflowVersion, error) {
if result, err := w.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*model.WorkflowVersion), nil
}
}
func (w workflowVersionDo) FindByPage(offset int, limit int) (result []*model.WorkflowVersion, count int64, err error) {
result, err = w.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 = w.Offset(-1).Limit(-1).Count()
return
}
func (w workflowVersionDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = w.Count()
if err != nil {
return
}
err = w.Offset(offset).Limit(limit).Scan(result)
return
}
func (w workflowVersionDo) Scan(result interface{}) (err error) {
return w.DO.Scan(result)
}
func (w workflowVersionDo) Delete(models ...*model.WorkflowVersion) (result gen.ResultInfo, err error) {
return w.DO.Delete(models)
}
func (w *workflowVersionDo) withDO(do gen.Dao) *workflowVersionDo {
w.DO = *do.(*gen.DO)
return w
}