db

Documentation

Index

• Constants
• func ReplaceColon(s string) string
• type Config
  • func (c *Config) Fomate2TgHTML() []byte
  • func (c *Config) Fomate2TgMdV2() []byte
• type DB
  • func New(dsn string) *DB
  • func (db *DB) DB() *gorm.DB
  • func (db *DB) DBType() dbTYPE
  • func (d *DB) FindImg(cfg FindConfig) ([]PhotoInfo, error)
  • func (d *DB) GetMaxCount(cfg MaxCountCfg) int64
• type FindConfig
• type MaxCountCfg
• type PhotoInfo
• type Subscribe
• type Token
• type UserDefaultCfg
• type UserInfo

Constants

const (
	T_SQLITE dbTYPE = iota
	T_POSTGRESQL
)

Functions

func ReplaceColon

func ReplaceColon(s string) string

Types

type Config

type Config struct {
	Tag               string  `json:"tag,omitempty" yaml:"tag,omitempty"`
	Mode              string  `json:"mode,omitempty" yaml:"mode,omitempty"`
	Num               int     `json:"num,omitempty" yaml:"num,omitempty"`
	Steps             int     `json:"steps,omitempty" yaml:"steps,omitempty"`
	Seed              uint32  `json:"seed,omitempty" yaml:"seed,omitempty"`
	CfgScale          int     `json:"scale,omitempty" yaml:"scale,omitempty"`
	Width             int     `json:"width" yaml:"width"`
	Height            int     `json:"height" yaml:"height"`
	Model             string  `json:"model,omitempty" yaml:"model,omitempty"`
	Uc                string  `json:"uc,omitempty" yaml:"uc,omitempty"`
	PrePhotoID        string  `json:"pre_photo_id,omitempty" yaml:"pre_photo_id,omitempty"`
	Strength          float64 `json:"strength,omitempty" yaml:"strength,omitempty"`
	ControlPhotoID    string  `json:"control_photo_id,omitempty" yaml:"control_photo_id,omitempty"`
	ControlPreprocess string  `json:"control_preprocess,omitempty" yaml:"control_preprocess,omitempty"`
	ControlProcess    string  `json:"control_process,omitempty" yaml:"control_process,omitempty"`
}

func (*Config) Fomate2TgHTML

func (c *Config) Fomate2TgHTML() []byte

func (*Config) Fomate2TgMdV2

func (c *Config) Fomate2TgMdV2() []byte

type DB

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

func New

func New(dsn string) *DB

func (*DB) DB

func (db *DB) DB() *gorm.DB

func (*DB) DBType

func (db *DB) DBType() dbTYPE

func (*DB) FindImg

func (d *DB) FindImg(cfg FindConfig) ([]PhotoInfo, error)

func (*DB) GetMaxCount

func (d *DB) GetMaxCount(cfg MaxCountCfg) int64

type FindConfig

type FindConfig struct {
	Deadline      time.Time
	Order         string
	Limit, Offset int64
	User_id       any
	Keywords      []string
	KeywordsRe    []string // Not support sqlite
	SelectKey     []string
}

type MaxCountCfg

type MaxCountCfg struct {
	Deadline   time.Time
	User_id    any
	Keywords   []string
	KeywordsRe []string // Not support sqlite
}

type PhotoInfo

type PhotoInfo struct {
	gorm.Model
	FileID  string `gorm:"unique;not null"`
	UserID  int64  `gorm:"not null"`
	UnShare bool   `gorm:"default:false"`
	Config  Config `gorm:"embedded"`
}

type Subscribe

type Subscribe struct {
	gorm.Model
	UserID     int64 `gorm:"unique;not null"`
	FreeAmount int   `gorm:"not null"`
	Deadline   time.Time
}

type Token

type Token struct {
	Token     string `gorm:"unique;not null"`
	ValidDate uint64 `gorm:"not null"`
}

type UserDefaultCfg

type UserDefaultCfg struct {
	UserDefaultUC     string
	UserDefaultMODE   string
	UserDefaultNumber int
	UserDefaultScale  int
	UserDefaultSteps  int
}

type UserInfo

type UserInfo struct {
	UserID     int64  `gorm:"unique;not null;primary_key"`
	Passwd     string `gorm:"not null"`
	Language   string `gorm:"default:en_us"`
	SharePhoto bool   `gorm:"default:true"`
	UserDefaultCfg
}