db

package
v0.1.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Apr 13, 2025 License: Apache-2.0 Imports: 35 Imported by: 0

Documentation

Index

Constants

View Source 
const DriverName = "sqlite3"

DriverName is the default driver name for SQLite.

Variables

View Source 
var (
	// ErrRecordNotFound - Record not found error
	ErrRecordNotFound = gorm.ErrRecordNotFound
	Client            *gorm.DB
)

Client - Database Client

Functions

func AddCertificate added in v0.0.2 

func AddCertificate(caType int, keyType string, commonName string, cert []byte, key []byte) error

AddCertificate add a certificate to the database

func AddContent 

func AddContent(content *clientpb.WebContent) (*models.WebsiteContent, error)

AddContent - Add content to website

func AddTask 

func AddTask(task *clientpb.Task) error

func CreateOperator 

func CreateOperator(name string, typ string, remoteAddr string) error

func DeleteAllCertificates 

func DeleteAllCertificates() error

DeleteAllCertificates

func DeleteArtifactByName added in v0.1.0 

func DeleteArtifactByName(artifactName string) error

func DeleteCertificate 

func DeleteCertificate(name string) error

DeleteCertificate

func DeleteContext added in v0.1.0 

func DeleteContext(contextID string) error

func DeletePipeline added in v0.1.0 

func DeletePipeline(name string) error

func DeleteProfileByName added in v0.1.0 

func DeleteProfileByName(profileName string) error

func DeleteWebsite added in v0.1.0 

func DeleteWebsite(name string) error

func DisablePipeline added in v0.0.3 

func DisablePipeline(pid string) error

func EnablePipeline added in v0.0.3 

func EnablePipeline(pid string) error

func FindAliveSessions 

func FindAliveSessions() ([]*models.Session, error)

func FindAllSessions 

func FindAllSessions() (*clientpb.Sessions, error)

func FindArtifact added in v0.1.0 

func FindArtifact(target *clientpb.Artifact) (*clientpb.Artifact, error)

FindArtifact

func FindContext added in v0.1.0 

func FindContext(taskID string) (*models.Context, error)

func FindPipeline added in v0.0.2 

func FindPipeline(name string) (*models.Pipeline, error)

func FindPipelineCert added in v0.0.3 

func FindPipelineCert(pipelineName, listenerID string) (string, string, error)

func FindSession 

func FindSession(sessionID string) (*models.Session, error)

func FindTaskAndMaxTasksID 

func FindTaskAndMaxTasksID(sessionID string) ([]*models.Task, uint32, error)

func FindWebContent added in v0.1.0 

func FindWebContent(id string) (*models.WebsiteContent, error)

WebContent by ID and path

func FindWebContentsByWebsite added in v0.1.0 

func FindWebContentsByWebsite(website string) ([]*models.WebsiteContent, error)

func FindWebsiteByName added in v0.1.0 

func FindWebsiteByName(name string) (*models.Pipeline, error)

FindWebsiteByName - Get website by name

func GetAllTask added in v0.0.3 

func GetAllTask() (*clientpb.Tasks, error)

func GetArtifactById added in v0.0.3 

func GetArtifactById(id uint32) (*models.Builder, error)

func GetArtifactByName added in v0.0.3 

func GetArtifactByName(name string) (*models.Builder, error)

func GetBuilderByModules added in v0.1.0 

func GetBuilderByModules(target string, modules []string) (*models.Builder, error)

func GetBuilderByProfileName added in v0.1.0 

func GetBuilderByProfileName(profileName string) (*clientpb.Builders, error)

func GetBuilderLogs added in v0.0.3 

func GetBuilderLogs(builderID uint32, limit int) (string, error)

func GetBuilders added in v0.1.0 

func GetBuilders() (*clientpb.Builders, error)

func GetContextByTask added in v0.1.0 

func GetContextByTask(taskID string) (*models.Context, error)

func GetContextFilesBySessionID added in v0.1.0 

func GetContextFilesBySessionID(sessionID string, fileTypes []string) ([]*models.Context, error)

func GetDownloadFiles added in v0.1.0 

func GetDownloadFiles(sid string) ([]*clientpb.File, error)

func GetProfile added in v0.0.3 

func GetProfile(name string) (*types.ProfileConfig, error)

func GetProfiles added in v0.0.3 

func GetProfiles() ([]models.Profile, error)

func GetTaskPB added in v0.0.3 

func GetTaskPB(taskID string) (*clientpb.Task, error)

func GetTasksByID added in v0.1.0 

func GetTasksByID(sessionID string) (*clientpb.Tasks, error)

func HasOperator added in v0.0.2 

func HasOperator(typ string) (bool, error)

func ListClients added in v0.0.2 

func ListClients() ([]*models.Operator, error)

func ListListeners 

func ListListeners() ([]models.Operator, error)

func ListPipelines added in v0.0.2 

func ListPipelines(listenerID string) ([]models.Pipeline, error)

func ListWebsite added in v0.0.3 

func ListWebsite(listenerID string) ([]*models.Pipeline, error)

func NewDBClient 

func NewDBClient() *gorm.DB

newDBClient - Initialize the db client

func NewProfile added in v0.0.3 

func NewProfile(profile *clientpb.Profile) error

generator

func Open 

func Open(dsn string) gorm.Dialector

func RemoveContent 

func RemoveContent(id string) error

RemoveContent - Remove content by ID

func RemoveSession added in v0.1.0 

func RemoveSession(sessionID string) error

Basic Session OP

func SaveArtifact added in v0.0.3 

func SaveArtifact(name, artifactType, platform, arch, stage, source string) (*models.Builder, error)

func SaveArtifactFromGenerate added in v0.0.3 

func SaveArtifactFromGenerate(req *clientpb.Generate) (*models.Builder, error)

func SaveArtifactFromID added in v0.1.0 

func SaveArtifactFromID(req *clientpb.Generate, ID uint32, resource string) (*models.Builder, error)

func SaveBuilderFromAction added in v0.1.0 

func SaveBuilderFromAction(inputs map[string]string, req *clientpb.Generate) (*models.Builder, error)

func SaveCertificate 

func SaveCertificate(certificate *models.Certificate) error

func SaveContext added in v0.1.0 

func SaveContext(ctx *clientpb.Context) (*models.Context, error)

func SavePipeline added in v0.1.0 

func SavePipeline(pipeline *models.Pipeline) (*models.Pipeline, error)

func Session 

func Session() *gorm.DB

Session - Database session

func UpdateBuilderLog added in v0.0.3 

func UpdateBuilderLog(name string, logEntry string)

func UpdateBuilderPath added in v0.1.0 

func UpdateBuilderPath(builder *models.Builder) error

func UpdateBuilderSrdi added in v0.1.0 

func UpdateBuilderSrdi(builder *models.Builder) error

func UpdateDownloadTotal added in v0.0.3 

func UpdateDownloadTotal(task *clientpb.Task, total int) error

func UpdateGeneratorConfig added in v0.0.3 

func UpdateGeneratorConfig(req *clientpb.Generate, path string, config *types.ProfileConfig) (string, error)

UpdateGeneratorConfig - Update the generator config

func UpdateProfileRaw added in v0.1.0 

func UpdateProfileRaw(profileName string, raw []byte) error

func UpdateSession 

func UpdateSession(sessionID, note, group string) error

func UpdateSessionTimer added in v0.1.0 

func UpdateSessionTimer(sessionID string, interval uint64, jitter float64) error

func UpdateTask 

func UpdateTask(task *clientpb.Task) error

func UpdateTaskCur added in v0.1.0 

func UpdateTaskCur(cur int, taskID string) error

func UpdateTaskDescription added in v0.0.3 

func UpdateTaskDescription(taskID, Description string) error

Types

type ContextQuery added in v0.1.0 

type ContextQuery struct {
	// contains filtered or unexported fields
}

ContextQuery 用于构建Context查询的结构体

func NewContextQuery added in v0.1.0 

func NewContextQuery() *ContextQuery

NewContextQuery 创建新的Context查询构建器

func (*ContextQuery) ByNonce added in v0.1.0 

func (q *ContextQuery) ByNonce(nonce string) *ContextQuery

ByNonce 按Nonce查询

func (*ContextQuery) ByPipeline added in v0.1.0 

func (q *ContextQuery) ByPipeline(pipelineID string) *ContextQuery

ByPipeline 按Pipeline ID查询

func (*ContextQuery) BySession added in v0.1.0 

func (q *ContextQuery) BySession(sessionID string) *ContextQuery

BySession 按会话ID查询

func (*ContextQuery) ByTask added in v0.1.0 

func (q *ContextQuery) ByTask(taskID string) *ContextQuery

ByTask 按任务ID查询

func (*ContextQuery) ByType added in v0.1.0 

func (q *ContextQuery) ByType(typ string) *ContextQuery

ByType 按类型查询

func (*ContextQuery) Find added in v0.1.0 

func (q *ContextQuery) Find() ([]*models.Context, error)

Find 执行查询并返回结果

func (*ContextQuery) First added in v0.1.0 

func (q *ContextQuery) First() (*models.Context, error)

First 查询单个结果

type Dialector 

type Dialector struct {
	DriverName string
	DSN        string
	Conn       gorm.ConnPool
}

func (Dialector) BindVarTo 

func (dialector Dialector) BindVarTo(writer clause.Writer, stmt *gorm.Statement, v interface{})

func (Dialector) ClauseBuilders 

func (dialector Dialector) ClauseBuilders() map[string]clause.ClauseBuilder

func (Dialector) DataTypeOf 

func (dialector Dialector) DataTypeOf(field *schema.Field) string

func (Dialector) DefaultValueOf 

func (dialector Dialector) DefaultValueOf(field *schema.Field) clause.Expression

func (Dialector) Explain 

func (dialector Dialector) Explain(sql string, vars ...interface{}) string

func (Dialector) Initialize 

func (dialector Dialector) Initialize(db *gorm.DB) (err error)

func (Dialector) Migrator 

func (dialector Dialector) Migrator(db *gorm.DB) gorm.Migrator

func (Dialector) Name 

func (dialector Dialector) Name() string

func (Dialector) QuoteTo 

func (dialector Dialector) QuoteTo(writer clause.Writer, str string)

func (Dialector) RollbackTo 

func (dialectopr Dialector) RollbackTo(tx *gorm.DB, name string) error

func (Dialector) SavePoint 

func (dialectopr Dialector) SavePoint(tx *gorm.DB, name string) error

type Migrator 

type Migrator struct {
	migrator.Migrator
}

Source Files

View all Source files

Directories

Path Synopsis

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.