0305571cb3
Fixed the build tags for the alt database engines. Tweaked the quotes on the PgSQL adapter. Still not functional. Removed the SQLite build tag as it's unlikely that it'll ever be implemented.
115 lines
3.5 KiB
Go
115 lines
3.5 KiB
Go
// +build mssql
|
|
|
|
/*
|
|
*
|
|
* Gosora MSSQL Interface
|
|
* Copyright Azareal 2016 - 2019
|
|
*
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
"net/url"
|
|
|
|
"./common"
|
|
"./query_gen/lib"
|
|
_ "github.com/denisenkom/go-mssqldb"
|
|
)
|
|
|
|
var dbInstance string = ""
|
|
|
|
func init() {
|
|
dbAdapter = "mssql"
|
|
_initDatabase = initMSSQL
|
|
}
|
|
|
|
func initMSSQL() (err error) {
|
|
// TODO: Move this bit to the query gen lib
|
|
query := url.Values{}
|
|
query.Add("database", common.DbConfig.Dbname)
|
|
u := &url.URL{
|
|
Scheme: "sqlserver",
|
|
User: url.UserPassword(common.DbConfig.Username, common.DbConfig.Password),
|
|
Host: common.DbConfig.Host + ":" + common.DbConfig.Port,
|
|
Path: dbInstance,
|
|
RawQuery: query.Encode(),
|
|
}
|
|
db, err = sql.Open("mssql", u.String())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Make sure that the connection is alive
|
|
err = db.Ping()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Set the number of max open connections
|
|
db.SetMaxOpenConns(64)
|
|
db.SetMaxIdleConns(32)
|
|
|
|
// Only hold connections open for five seconds to avoid accumulating a large number of stale connections
|
|
//db.SetConnMaxLifetime(5 * time.Second)
|
|
|
|
// Build the generated prepared statements, we are going to slowly move the queries over to the query generator rather than writing them all by hand, this'll make it easier for us to implement database adapters for other databases like PostgreSQL, MSSQL, SQlite, etc.
|
|
err = _gen_mssql()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Ready the query builder
|
|
qgen.Builder.SetConn(db)
|
|
err = qgen.Builder.SetAdapter("mssql")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
setter, ok := qgen.Builder.GetAdapter().(qgen.SetPrimaryKeys)
|
|
if ok {
|
|
setter.SetPrimaryKeys(dbTablePrimaryKeys)
|
|
}
|
|
|
|
// TODO: Is there a less noisy way of doing this for tests?
|
|
/*log.Print("Preparing getActivityFeedByWatcher statement.")
|
|
stmts.getActivityFeedByWatcherStmt, err = db.Prepare("SELECT activity_stream_matches.asid, activity_stream.actor, activity_stream.targetUser, activity_stream.event, activity_stream.elementType, activity_stream.elementID FROM [activity_stream_matches] INNER JOIN [activity_stream] ON activity_stream_matches.asid = activity_stream.asid AND activity_stream_matches.watcher != activity_stream.actor WHERE [watcher] = ? ORDER BY activity_stream.asid DESC OFFSET 0 ROWS FETCH NEXT 8 ROWS ONLY")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Print("Preparing getActivityCountByWatcher statement.")
|
|
stmts.getActivityCountByWatcherStmt, err = db.Prepare("SELECT count(*) FROM [activity_stream_matches] INNER JOIN [activity_stream] ON activity_stream_matches.asid = activity_stream.asid AND activity_stream_matches.watcher != activity_stream.actor WHERE [watcher] = ?")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Print("Preparing todaysPostCount statement.")
|
|
stmts.todaysPostCountStmt, err = db.Prepare("select count(*) from replies where createdAt >= DATEADD(DAY, -1, GETUTCDATE())")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Print("Preparing todaysTopicCount statement.")
|
|
stmts.todaysTopicCountStmt, err = db.Prepare("select count(*) from topics where createdAt >= DATEADD(DAY, -1, GETUTCDATE())")
|
|
if err != nil {
|
|
return err
|
|
}*/
|
|
|
|
log.Print("Preparing todaysTopicCountByForum statement.")
|
|
// TODO: Stop hard-coding this query
|
|
stmts.todaysTopicCountByForum, err = db.Prepare("select count(*) from topics where createdAt >= DATEADD(DAY, -1, GETUTCDATE()) and parentID = ?")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
/*log.Print("Preparing todaysNewUserCount statement.")
|
|
stmts.todaysNewUserCountStmt, err = db.Prepare("select count(*) from users where createdAt >= DATEADD(DAY, -1, GETUTCDATE())")
|
|
if err != nil {
|
|
return err
|
|
}*/
|
|
|
|
return nil
|
|
}
|