daemon

package
v1.1.5 Latest Latest
Warning

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

 Go to latest Published: Jun 17, 2025 License: MIT Imports: 24 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Daemon 

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

Daemon manages multiple spammer instances with database persistence and graceful shutdown. It orchestrates the lifecycle of spammers including restoration from database, global configuration management, and coordinated shutdown with a 10-second timeout.

func NewDaemon 

func NewDaemon(parentCtx context.Context, logger logrus.FieldLogger, clientPool *spamoor.ClientPool, rootWallet *spamoor.RootWallet, txpool *spamoor.TxPool, db *db.Database) *Daemon

NewDaemon creates a new daemon instance with the provided components. It initializes a cancellable context, spammer map, and global configuration map. The daemon manages spammer instances and handles their lifecycle.

func (*Daemon) DeleteSpammer 

func (d *Daemon) DeleteSpammer(id int64) error

DeleteSpammer removes a spammer from both the daemon and database. If the spammer is running, it will be paused first before deletion. Returns an error if the spammer is not found or if database deletion fails.

func (*Daemon) ExportSpammers added in v1.1.5 

func (d *Daemon) ExportSpammers(spammerIDs ...int64) (string, error)

ExportSpammers exports the specified spammer IDs to YAML format. If no IDs are provided, exports all spammers. Returns the YAML string representation of the spammers.

func (*Daemon) GetAllSpammers 

func (d *Daemon) GetAllSpammers() []*Spammer

GetAllSpammers returns all spammer instances sorted by ID in descending order. This provides a snapshot of all active spammer instances managed by the daemon. The returned slice is safe to modify as it's a copy.

func (*Daemon) GetClientPool 

func (d *Daemon) GetClientPool() *spamoor.ClientPool

GetClientPool returns the RPC client pool used by all spammers. This provides access to the Ethereum client connections for transaction submission.

func (*Daemon) GetGlobalCfg added in v1.1.3 

func (d *Daemon) GetGlobalCfg() map[string]interface{}

GetGlobalCfg returns the global configuration map that contains shared configuration values accessible to all spammer instances.

func (*Daemon) GetRootWallet 

func (d *Daemon) GetRootWallet() *spamoor.RootWallet

GetRootWallet returns the root wallet used for funding spammer wallets. This provides access to the main wallet that distributes funds to child wallets.

func (*Daemon) GetSpammer 

func (d *Daemon) GetSpammer(id int64) *Spammer

GetSpammer retrieves a spammer instance by ID from the internal map. Returns nil if the spammer with the given ID does not exist. This method is thread-safe using read locks.

func (*Daemon) ImportSpammers added in v1.1.5 

func (d *Daemon) ImportSpammers(input string) (*ImportResult, error)

ImportSpammers imports spammers from YAML data, file path, or URL. Handles deduplication by checking name combinations and validates before importing. Returns validation results and the number of spammers imported.

func (*Daemon) ImportSpammersOnStartup added in v1.1.5 

func (d *Daemon) ImportSpammersOnStartup(source string, logger logrus.FieldLogger) error

ImportSpammersOnStartup imports spammers from a file or URL on startup. This method reuses the import functionality but starts the spammers after importing them, which is the key difference from regular imports. Only imports on first startup for safety.

func (*Daemon) InitializeMetrics added in v1.1.5 

func (d *Daemon) InitializeMetrics() error

func (*Daemon) NewMetricsCollector added in v1.1.5 

func (d *Daemon) NewMetricsCollector() *MetricsCollector

NewMetricsCollector creates and registers all metrics

func (*Daemon) NewSpammer 

func (d *Daemon) NewSpammer(scenarioName string, config string, name string, description string, startImmediately bool) (*Spammer, error)

NewSpammer creates a new spammer instance with the specified configuration. It validates the config, generates a unique ID (starting from 100), persists to database, and optionally starts execution immediately. The ID counter is stored in database state.

func (*Daemon) ReclaimSpammer added in v1.1.4 

func (d *Daemon) ReclaimSpammer(id int64) error

ReclaimSpammer reclaims funds from all wallets in the spammer's wallet pool. This transfers remaining ETH back to the root wallet for reuse. Returns an error if the spammer is not found or if fund reclamation fails.

func (*Daemon) Run 

func (d *Daemon) Run() (bool, error)

Run initializes the daemon by restoring spammers from the database. Returns true if this is the first launch, false if spammers were restored. Marks the first launch state in the database to track initialization status.

func (*Daemon) SetGlobalCfg added in v1.1.3 

func (d *Daemon) SetGlobalCfg(name string, value interface{})

SetGlobalCfg sets a global configuration value that can be accessed by all spammers. This allows sharing configuration data across multiple spammer instances.

func (*Daemon) Shutdown 

func (d *Daemon) Shutdown()

Shutdown performs a graceful shutdown of the daemon and all running spammers. It cancels the context to stop all spammers, waits up to 10 seconds for them to finish, and closes the database connection. This ensures clean resource cleanup.

func (*Daemon) TrackSpammerStatusChange added in v1.1.5 

func (d *Daemon) TrackSpammerStatusChange(spammerID int64, running bool)

TrackSpammerStatusChange updates the spammer running status metric

func (*Daemon) TrackTransactionFailure added in v1.1.5 

func (d *Daemon) TrackTransactionFailure(spammerID int64)

TrackTransactionFailure records a failed transaction for metrics

func (*Daemon) TrackTransactionSent added in v1.1.5 

func (d *Daemon) TrackTransactionSent(spammerID int64)

TrackTransactionSent records a successful transaction send for metrics

func (*Daemon) UpdateSpammer 

func (d *Daemon) UpdateSpammer(id int64, name string, description string, config string) error

UpdateSpammer modifies the name, description, and configuration of an existing spammer. The configuration is validated by attempting to unmarshal it into SpammerConfig. Returns an error if the spammer is not found, config is invalid, or database update fails.

type ExportSpammerConfig added in v1.1.5 

type ExportSpammerConfig struct {
	Scenario    string                 `yaml:"scenario"`
	Name        string                 `yaml:"name"`
	Description string                 `yaml:"description"`
	Config      map[string]interface{} `yaml:"config"`
}

ExportSpammerConfig represents a spammer configuration for export/import. This uses the same format as StartupSpammerConfig to maintain compatibility.

type ImportItem added in v1.1.5 

type ImportItem struct {
	// Spammer configuration fields
	Scenario    string                 `yaml:"scenario,omitempty"`
	Name        string                 `yaml:"name,omitempty"`
	Description string                 `yaml:"description,omitempty"`
	Config      map[string]interface{} `yaml:"config,omitempty"`

	// Include directive
	Include string `yaml:"include,omitempty"`
}

ImportItem represents either a spammer config or an include directive

type ImportResult added in v1.1.5 

type ImportResult struct {
	ImportedCount int                     `json:"imported_count"`
	Validation    *ImportValidationResult `json:"validation"`
	Imported      []ImportedSpammerInfo   `json:"imported"`
	Errors        []string                `json:"errors"`
	Warnings      []string                `json:"warnings"`
	Message       string                  `json:"message"`
}

ImportResult contains the results of an import operation

type ImportValidationResult added in v1.1.5 

type ImportValidationResult struct {
	TotalSpammers    int                     `json:"total_spammers"`
	ValidSpammers    int                     `json:"valid_spammers"`
	Duplicates       []string                `json:"duplicates"`
	InvalidScenarios []string                `json:"invalid_scenarios"`
	Spammers         []SpammerValidationInfo `json:"spammers"`
}

ImportValidationResult contains validation results for import data

type ImportedSpammerInfo added in v1.1.5 

type ImportedSpammerInfo struct {
	ID          int64  `json:"id"`
	Name        string `json:"name"`
	Scenario    string `json:"scenario"`
	Description string `json:"description"`
}

type MetricsCollector added in v1.1.5 

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

MetricsCollector holds all Prometheus metrics for the spamoor daemon

func (*MetricsCollector) IncrementTransactionFailures added in v1.1.5 

func (mc *MetricsCollector) IncrementTransactionFailures(spammerID int64, spammerName, scenario string)

IncrementTransactionFailures increments the failed transaction counter

func (*MetricsCollector) IncrementTransactionsSent added in v1.1.5 

func (mc *MetricsCollector) IncrementTransactionsSent(spammerID int64, spammerName, scenario string)

IncrementTransactionsSent increments the sent transaction counter

func (*MetricsCollector) SetSpammerRunning added in v1.1.5 

func (mc *MetricsCollector) SetSpammerRunning(spammerID int64, spammerName, scenario string, running bool)

SetSpammerRunning sets the running status for a spammer

type Spammer 

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

Spammer represents a single scenario execution instance with database persistence. It manages the lifecycle of a scenario including wallet pool creation, configuration loading, and execution monitoring with panic recovery.

func (*Spammer) GetConfig 

func (s *Spammer) GetConfig() string

GetConfig returns the YAML configuration string used by this spammer.

func (*Spammer) GetCreatedAt 

func (s *Spammer) GetCreatedAt() int64

GetCreatedAt returns the Unix timestamp when this spammer was created.

func (*Spammer) GetDescription 

func (s *Spammer) GetDescription() string

GetDescription returns the description text of the spammer instance.

func (*Spammer) GetID 

func (s *Spammer) GetID() int64

GetID returns the unique identifier of the spammer instance.

func (*Spammer) GetLogScope 

func (s *Spammer) GetLogScope() *logscope.LogScope

GetLogScope returns the buffered log scope for this spammer. This provides access to captured log messages for debugging and monitoring.

func (*Spammer) GetName 

func (s *Spammer) GetName() string

GetName returns the human-readable name of the spammer instance.

func (*Spammer) GetScenario 

func (s *Spammer) GetScenario() string

GetScenario returns the name of the scenario being executed by this spammer.

func (*Spammer) GetStatus 

func (s *Spammer) GetStatus() int

GetStatus returns the current execution status of the spammer as an integer. Use SpammerStatus constants to interpret the returned value.

func (*Spammer) GetWalletPool 

func (s *Spammer) GetWalletPool() *spamoor.WalletPool

GetWalletPool returns the wallet pool used by this spammer for transaction submission. Returns nil if the spammer has not been started or the wallet pool is not initialized.

func (*Spammer) Pause 

func (s *Spammer) Pause() error

Pause stops the running scenario by canceling its context. It waits up to 10 seconds for the scenario to stop gracefully. Returns an error if the scenario is not running or fails to stop within the timeout.

func (*Spammer) Start 

func (s *Spammer) Start() error

Start begins execution of the spammer's scenario in a separate goroutine. It updates the status to running in the database and launches the scenario runner. Returns an error if the database update fails.

func (*Spammer) TrackTransactionResult added in v1.1.5 

func (s *Spammer) TrackTransactionResult(err error)

TrackTransactionResult records transaction success or failure for metrics

type SpammerConfig 

type SpammerConfig struct {
	Seed           string       `yaml:"seed"`
	RefillAmount   *uint256.Int `yaml:"refill_amount"`
	RefillBalance  *uint256.Int `yaml:"refill_balance"`
	RefillInterval uint64       `yaml:"refill_interval"`
	WalletCount    int          `yaml:"wallet_count"`
}

SpammerConfig defines the wallet configuration for a spammer instance. This includes the seed for deterministic wallet generation and funding parameters.

type SpammerStatus 

type SpammerStatus int

SpammerStatus represents the execution state of a spammer instance. It tracks whether the spammer is paused, running, finished, or failed. 

const (
	// SpammerStatusPaused indicates the spammer is stopped and can be resumed
	SpammerStatusPaused SpammerStatus = iota
	// SpammerStatusRunning indicates the spammer is actively executing its scenario
	SpammerStatusRunning
	// SpammerStatusFinished indicates the spammer completed its scenario successfully
	SpammerStatusFinished
	// SpammerStatusFailed indicates the spammer encountered an error and stopped
	SpammerStatusFailed
)

type SpammerValidationInfo added in v1.1.5 

type SpammerValidationInfo struct {
	Name        string   `json:"name"`
	Scenario    string   `json:"scenario"`
	Description string   `json:"description"`
	Valid       bool     `json:"valid"`
	Issues      []string `json:"issues"`
}

SpammerValidationInfo contains validation info for a single spammer

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.