builder_pg

Documentation

Index

• Constants
• type BuilderPG
  • func NewBuilderPG(cfg config.Config) BuilderPG
  • func (b BuilderPG) ImportData() error
  • func (b BuilderPG) ResetData()
• type Title
• type WriteCounter
  • func NewWriteCounter(total int) *WriteCounter
  • func (wc *WriteCounter) Write(p []byte) (int, error)

Constants

const BatchSize = 100_000

Types

type BuilderPG

type BuilderPG struct {
	config.Config
	DB     *sql.DB
	GormDB *gorm.DB
}

func NewBuilderPG

func NewBuilderPG(cfg config.Config) BuilderPG

func (BuilderPG) ImportData

func (b BuilderPG) ImportData() error

func (BuilderPG) ResetData

func (b BuilderPG) ResetData()

type Title

type Title struct {
	ID        int
	Name      string
	YearStart sql.NullInt32
	YearEnd   sql.NullInt32
	Language  string
	DOI       string
}

type WriteCounter

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

WriteCounter counts the number of bytes written to it. It implements to the io.Writer interface and we can pass this into io.TeeReader() which will report progress on each write cycle.

func NewWriteCounter

func NewWriteCounter(total int) *WriteCounter

NewWriteCounter is a constructor for WriteCounter type.

func (*WriteCounter) Write

func (wc *WriteCounter) Write(p []byte) (int, error)

Write writes more bytes to WriteCounter and sets the progress for progress bar.