Add: put tag on file

This commit is contained in:
2021-12-12 17:09:16 +08:00
parent f71544caab
commit 2c802ca807
6 changed files with 234 additions and 3 deletions

View File

@@ -0,0 +1,25 @@
package database
func (database *Database) PutTagOnFile(tagID, fileID, userID int64) error {
_, err := database.stmt.putTagOnFile.Exec(tagID, fileID, userID)
return err
}
func (database *Database) GetTagsOnFile(fileID int64) ([]*Tag, error) {
rows, err := database.stmt.getTagsOnFile.Query(fileID)
if err != nil {
return nil, err
}
defer rows.Close()
tags := make([]*Tag, 0)
for rows.Next() {
tag := &Tag{}
err = rows.Scan(&tag.ID, &tag.Name, &tag.Description, &tag.CreatedByUserId)
if err != nil {
return nil, err
}
tags = append(tags, tag)
}
return tags, nil
}

View File

@@ -190,6 +190,14 @@ JOIN users ON tags.created_by_user_id = users.id
var updateTagQuery = `UPDATE tags SET name = ?, description = ? WHERE id = ?;`
var putTagOnFileQuery = `INSERT INTO file_has_tag (tag_id, file_id, user_id) VALUES (?, ?, ?);`
var getTagsOnFileQuery = `SELECT
tags.id, tags.name, tags.description, tags.created_by_user_id
FROM file_has_tag
JOIN tags ON file_has_tag.tag_id = tags.id
WHERE file_has_tag.file_id = ?;`
type Stmt struct {
initFilesTable *sql.Stmt
initFoldersTable *sql.Stmt
@@ -226,6 +234,8 @@ type Stmt struct {
getTag *sql.Stmt
getTags *sql.Stmt
updateTag *sql.Stmt
putTagOnFile *sql.Stmt
getTagsOnFile *sql.Stmt
}
func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
@@ -506,5 +516,17 @@ func NewPreparedStatement(sqlConn *sql.DB) (*Stmt, error) {
return nil, err
}
// init putTagOnFile
stmt.putTagOnFile, err = sqlConn.Prepare(putTagOnFileQuery)
if err != nil {
return nil, err
}
// init getTagsOnFile
stmt.getTagsOnFile, err = sqlConn.Prepare(getTagsOnFileQuery)
if err != nil {
return nil, err
}
return stmt, err
}