63 lines
1 KiB
Go
63 lines
1 KiB
Go
|
package collection
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"git.nutfactory.org/hoernschen/ActivityPub/utils/database"
|
||
|
|
)
|
||
|
|
|
||
|
|
func CreateCollectionObject(collectionId string, objectId string) (err error) {
|
||
|
|
sqlStmt := fmt.Sprintf(`INSERT INTO collectionObject
|
||
|
|
(collectionId, objectId)
|
||
|
|
VALUES
|
||
|
|
(?, ?)`)
|
||
|
|
|
||
|
|
tx, err := database.DB.Begin()
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
stmt, err := tx.Prepare(sqlStmt)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
defer stmt.Close()
|
||
|
|
|
||
|
|
_, err = stmt.Exec(
|
||
|
|
collectionId,
|
||
|
|
objectId,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
tx.Rollback()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
tx.Commit()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadCollectionObjects(collectionId string) (collectionObjects []string, err error) {
|
||
|
|
queryStmt := fmt.Sprintf(`SELECT objectId
|
||
|
|
FROM collectionObject
|
||
|
|
WHERE collectionId = '%s'`, collectionId)
|
||
|
|
|
||
|
|
rows, err := database.DB.Query(queryStmt)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
for rows.Next() {
|
||
|
|
var collectionObject string
|
||
|
|
err = rows.Scan(
|
||
|
|
&collectionObject,
|
||
|
|
)
|
||
|
|
if err != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
collectionObjects = append(collectionObjects, collectionObject)
|
||
|
|
}
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|