78 lines
1.3 KiB
Go
78 lines
1.3 KiB
Go
|
package community
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/status-im/status-go/services/wallet/thirdparty"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DataDB struct {
|
||
|
|
db *sql.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewDataDB(sqlDb *sql.DB) *DataDB {
|
||
|
|
return &DataDB{
|
||
|
|
db: sqlDb,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const communityDataColumns = "id, name, color, image"
|
||
|
|
const selectCommunityDataColumns = "name, color, image"
|
||
|
|
|
||
|
|
func (o *DataDB) SetCommunityInfo(id string, c thirdparty.CommunityInfo) (err error) {
|
||
|
|
tx, err := o.db.Begin()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
if err == nil {
|
||
|
|
err = tx.Commit()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
_ = tx.Rollback()
|
||
|
|
}()
|
||
|
|
|
||
|
|
update, err := tx.Prepare(fmt.Sprintf(`INSERT OR REPLACE INTO community_data_cache (%s)
|
||
|
|
VALUES (?, ?, ?, ?)`, communityDataColumns))
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = update.Exec(
|
||
|
|
id,
|
||
|
|
c.CommunityName,
|
||
|
|
c.CommunityColor,
|
||
|
|
c.CommunityImage,
|
||
|
|
)
|
||
|
|
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *DataDB) GetCommunityInfo(id string) (*thirdparty.CommunityInfo, error) {
|
||
|
|
var ret thirdparty.CommunityInfo
|
||
|
|
|
||
|
|
getData, err := o.db.Prepare(fmt.Sprintf(`SELECT %s
|
||
|
|
FROM community_data_cache
|
||
|
|
WHERE id=?`, selectCommunityDataColumns))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
row := getData.QueryRow(id)
|
||
|
|
|
||
|
|
err = row.Scan(
|
||
|
|
&ret.CommunityName,
|
||
|
|
&ret.CommunityColor,
|
||
|
|
&ret.CommunityImage,
|
||
|
|
)
|
||
|
|
|
||
|
|
if err == sql.ErrNoRows {
|
||
|
|
return nil, nil
|
||
|
|
} else if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return &ret, nil
|
||
|
|
}
|