db

package
v0.0.0-...-4e43b7b Latest Latest
Warning

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

Go to latest
Published: Jun 9, 2023 License: MIT Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func MakeSomeErr

func MakeSomeErr() error

Types

type DB

type DB interface {
	ConnectDB(host string, port int, user, password, dbname string) (*Database, error)
}

type DBConfig

type DBConfig struct {
	Host        string
	Port        int
	User        string
	Password    string
	Dbname      string
	Sslmode     string
	Sslrootcert string
	Sslkey      string
	Sslsert     string
}

func GetDBConfig

func GetDBConfig() (*DBConfig, error)

type Database

type Database struct {
	Conn *sql.DB
}

func ConnectDB

func ConnectDB(config *DBConfig) (*Database, error)

func (*Database) AddNewCosineSimilarity

func (db *Database) AddNewCosineSimilarity(workerId, jobId string, score []float64, rowId, columnId []string) error

func (*Database) AddNewJob

func (db *Database) AddNewJob(jobId string, keyword string, owner string) error

func (*Database) AddNewTfIdfScore

func (db *Database) AddNewTfIdfScore(workerId, articleId string, wordId int64, vocabId []int, score []float64) error

func (*Database) AddNewVocabList

func (db *Database) AddNewVocabList(vocabList []string, workerId string, jobId string) (int64, error)

func (*Database) AddNewWorker

func (db *Database) AddNewWorker(workerId string, jobId string, keyword string) error

func (*Database) DeleteJob

func (db *Database) DeleteJob(id string) error

func (*Database) DeleteWorker

func (db *Database) DeleteWorker(workerId string) error

func (*Database) GetAllJob

func (db *Database) GetAllJob() ([]*models.Job, error)

func (*Database) GetAllReRunJob

func (db *Database) GetAllReRunJob() ([]string, error)

func (*Database) GetAllWorker

func (db *Database) GetAllWorker(jobId string) ([]*models.Worker, error)

func (*Database) GetArticlesByJobID

func (db *Database) GetArticlesByJobID(jobID string) ([]*models.Article, error)

func (Database) GetArticlesByOneDay

func (db Database) GetArticlesByOneDay(startDate, endDate time.Time, workerId string) ([]*models.SimplifiedArticle, error)

func (*Database) GetArticlesByWorkerID

func (db *Database) GetArticlesByWorkerID(workerID string) ([]*models.Article, error)

func (*Database) GetCosineSimilarity

func (db *Database) GetCosineSimilarity(workerId string) (*cosine_similarity.CosineSimilarity, error)

func (*Database) GetCountByDay

func (db *Database) GetCountByDay(jobId string) ([]*models.ArticleCount, error)

func (*Database) GetCountByHour

func (db *Database) GetCountByHour(jobId string) ([]*models.ArticleCount, error)

func (*Database) GetJob

func (db *Database) GetJob(jobId string) (*models.Job, error)

func (*Database) GetJobIsReRun

func (db *Database) GetJobIsReRun(id string) (bool, error)

func (Database) GetPreprocessedTextByWorkerID

func (db Database) GetPreprocessedTextByWorkerID(workerID string) ([]*models.SimplifiedArticle, error)

func (*Database) GetSimilarity

func (db *Database) GetSimilarity(workerId string) error

func (*Database) GetTfidfScore

func (db *Database) GetTfidfScore(workerId string) ([]*tfidf.Tfidf, error)

func (*Database) GetVocabList

func (db *Database) GetVocabList(workerId string) (*vocab.Vocab, error)

func (*Database) GetWorkerCreateTime

func (db *Database) GetWorkerCreateTime(workerId string) error

func (*Database) RemoveDuplicateArticle

func (db *Database) RemoveDuplicateArticle(jobID string) error

func (*Database) UpdateJob

func (db *Database) UpdateJob(id string, status string) error

func (*Database) UpdateScheduleJob

func (db *Database) UpdateScheduleJob(id string, isSchedule bool) error

func (*Database) UpdateWorker

func (db *Database) UpdateWorker(workerId string, status string) error

func (*Database) WriteTweetToArticle

func (db *Database) WriteTweetToArticle(tweet models.TweetArticle) error

type DuplicateContent

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

func NewDuplicateContent

func NewDuplicateContent(message string) *DuplicateContent

func (*DuplicateContent) Error

func (e *DuplicateContent) Error() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL