store

package
v0.0.4 Latest Latest
Warning

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

 Go to latest Published: Dec 13, 2023 License: BSD-2-Clause Imports: 9 Imported by: 0

Documentation

Index

Constants

View Source 
const (
	StatusWaitingToStart Status = "Waiting to start"
	StatusJobQueued      Status = "Job queued"
	StatusProcessing     Status = "Processing"
	StatusComplete       Status = "Complete"
	StatusUnknown        Status = "Unknown"

	IsQuadrilateralGrid  = "0"
	NotQuadrilateralGrid = "1"
	TypeImagine          = "Imagine"
	TypeUpscale          = "Upscale"
	TypeUpscaleXTwo      = "UpscaleXTwo"
	TypeReroll           = "Reroll"
	TypeUpscaleXFour     = "UpscaleXFour"
	TypeVariation        = "Variation"
	TypeVariationStrong  = "VariationStrong"
	TypeVariationSubtle  = "VariationSubtle"
	TypeVariationRegion  = "VariationRegion"
	TypePanUp            = "PanUp"
	TypePanRight         = "PanRight"
	TypePanLeft          = "PanLeft"
	TypePanDown          = "PanDown"
	TypePicReader        = "PicReader"
	TypeZoomOutOneX      = "ZoomOutOneX"
	TypeZoomOutTwoX      = "ZoomOutTwoX"
	TypeUnknown          = "Unknown"

	TypeUpscaleKey         = "Upscale"
	TypeUpscaleXFourKey    = "UpscaleXFour"
	TypeUpscaleXTwoKey     = "UpscaleXTwo"
	TypeVariationKey       = "Variation"
	TypeVariationStrongKey = "VariationStrong"
	TypeVariationSubtleKey = "VariationSubtle"
	TypeVariationRegionKey = "VariationRegion"
	TypeRerollKey          = "Reroll"
	PanTypeUpKey           = "PanUp"
	PanTypeRightKey        = "PanRight"
	PanTypeLeftKey         = "PanLeft"
	PanTypeDownKey         = "PanDown"
	TypePicReaderKey       = "PicReader"
	TypeZoomOutOneXKey     = "ZoomOutOneX"
	TypeZoomOutTwoXKey     = "ZoomOutTwoX"

	PanParamUp    = "pan_up"
	PanParamRight = "pan_right"
	PanParamLeft  = "pan_left"
	PanParamDown  = "pan_down"

	DrmMidjourneyCaptcha               = `drm:midjourney_server`
	Expired              time.Duration = 3 * 24 * time.Hour

	ImageStart           = `\*\*(.*?)\*\* - <@\d+> \((.*?)\)`
	UpscaleRun           = `\*\*(.*?)\*\* - Image #(\d+) <@\d+>`
	UpscaleStart         = `Upscaling image #(\d+)`
	UpscaleMultipleStart = `\*\*(.*?)\*\* - Upscaled \((.*?)\) by <@\d+> \((.*?)\)`
	VariationRun         = `\*\*(.*?)\*\* - Variations by <@\d+> \((.*?)\)`
	VariationStart       = `Making variations for image #(\d+)`
)

Variables

This section is empty.

Functions

func CreateKey 

func CreateKey(items ...string) string

func GetKey 

func GetKey(s string) string

Types

type CompleteInfo 

type CompleteInfo struct {
	ID          string
	Attachments []map[string]any
}

type Config 

type Config struct {
	Redis Redis
}

type Error 

type Error struct {
	Code api.Codes
	Msg  string
}

func (Error) Error 

func (e Error) Error() string

type Key 

type Key string

type MetaData 

type MetaData struct {
	ID                  string `redis:"id"`
	Prompt              string `redis:"prompt"`
	Type                string `redis:"type"`
	Status              Status `redis:"status"`
	ProcessRate         string `redis:"process_rate"`
	Attachments         string `redis:"attachments"`
	StartTime           int64  `redis:"start_time"`
	CompleteTime        int64  `redis:"complete_time"`
	Webhook             string `redis:"webhook"`
	CompleteMessageID   string `redis:"complete_message_id"`
	Mode                string `redis:"mode"`
	RequestId           string `redis:"request_id"`
	IsQuadrilateralGrid string `redis:"is_quadrilateral_grid"`
	MemberId            string `redis:"member_id"`
}

func (*MetaData) GetImageURL 

func (md *MetaData) GetImageURL() (string, error)

type Redis 

type Redis struct {
	Address  string
	Password string
}

type Status 

type Status string

type Store 

type Store struct {
	*redis.Client
}

func NewStore 

func NewStore(config *Config) *Store

func (*Store) CheckPrompt 

func (s *Store) CheckPrompt(ctx context.Context, prompt string) error

func (*Store) GetID 

func (s *Store) GetID(ctx context.Context, key string) (string, error)

func (*Store) GetMetaData 

func (s *Store) GetMetaData(ctx context.Context, id string) (*MetaData, error)

func (*Store) GetType added in v0.0.4 

func (s *Store) GetType(ctx context.Context, key string) (isQuadrilateralGrid, mjType string, err error)

通过referenced_message_id反向查询对应的类型

func (*Store) SaveMeta 

func (s *Store) SaveMeta(
	ctx context.Context,
	key,
	id,
	typ,
	prompt string,
	status Status,
	start_time int64,
) error

func (*Store) SaveWebhook 

func (s *Store) SaveWebhook(ctx context.Context, id, webhook, reqId, memberId string) error

func (*Store) SaveWithComplete 

func (s *Store) SaveWithComplete(ctx context.Context, completeMessageID, quadrilateralGrid, prompt, key, mode, attachments string, webhookFunc func(*MetaData)) error

func (*Store) UpdateProcessRate 

func (s *Store) UpdateProcessRate(ctx context.Context, id, processRate string, attachments string, webhookFunc func(*MetaData)) error

type Type 

type Type string

Source Files

View all Source files

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.