store

Documentation

Index

• Constants
• func CreateKey(items ...string) string
• func GetKey(s string) string
• type CompleteInfo
• type Config
• type Error
  • func (e Error) Error() string
• type Key
• type MetaData
  • func (md *MetaData) GetImageURL() (string, error)
• type Redis
• type Status
• type Store
  • func NewStore(config *Config) *Store
  • func (s *Store) CheckPrompt(ctx context.Context, prompt string) error
  • func (s *Store) GetID(ctx context.Context, prompt string) (string, error)
  • func (s *Store) GetMetaData(ctx context.Context, id string) (*MetaData, error)
  • func (s *Store) SaveMeta(ctx context.Context, id, prompt string, status Status, typ Type, ...) error
  • func (s *Store) SaveWebhook(ctx context.Context, id, webhook, reqId, memberId string) error
  • func (s *Store) SaveWithComplete(ctx context.Context, completeMessageID, prompt, mode, attachments string, ...) error
  • func (s *Store) UpdateProcessRate(ctx context.Context, id, processRate string, attachments string, ...) error
• type Type

Constants

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

	TypeImagine          Type          = "Imagine"
	TypeUpscale          Type          = "Upscale"
	TypeUnknown          Type          = "Unknown"
	DrmMidjourneyCaptcha               = `drm:midjourney_server`
	Expired              time.Duration = 3 * time.Hour
)

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              Type   `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"`
	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, prompt string) (string, error)

func (*Store) GetMetaData

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

func (*Store) SaveMeta

func (s *Store) SaveMeta(
	ctx context.Context,
	id,
	prompt string,
	status Status,
	typ Type,
	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, prompt, 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