Fix some build errors

This commit is contained in:
Abhishek Kumar 2020-01-25 18:15:39 +05:30
parent cebef9bf09
commit 5f15826c2f
6 changed files with 35 additions and 10 deletions

View file

@ -70,7 +70,7 @@ type GetCreatorIDForCanonicalAliasRequest struct {
} }
// GetCreatorIDForCanonicalAliasResponse is a response to GetCreatorIDForCanonicalAlias // GetCreatorIDForCanonicalAliasResponse is a response to GetCreatorIDForCanonicalAlias
type GetCreatorIDForAliasResponse struct { type GetCreatorIDForCanonicalAliasResponse struct {
// The user ID of the canonical alias creator // The user ID of the canonical alias creator
UserID string `json:"user_id"` UserID string `json:"user_id"`
} }
@ -102,7 +102,7 @@ type RoomserverCanonicalAliasAPI interface {
GetCreatorIDForCanonicalAlias( GetCreatorIDForCanonicalAlias(
ctx context.Context, ctx context.Context,
req *GetCreatorIDForCanonicalAliasRequest, req *GetCreatorIDForCanonicalAliasRequest,
response *GetCreatorIDForCanonicalAliasAliasResponse, response *GetCreatorIDForCanonicalAliasResponse,
) error ) error
} }

View file

@ -17,15 +17,12 @@ package canonical_alias
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"fmt"
"net/http" "net/http"
"time"
appserviceAPI "github.com/matrix-org/dendrite/appservice/api" appserviceAPI "github.com/matrix-org/dendrite/appservice/api"
"github.com/matrix-org/dendrite/common" "github.com/matrix-org/dendrite/common"
"github.com/matrix-org/dendrite/common/config" "github.com/matrix-org/dendrite/common/config"
roomserverAPI "github.com/matrix-org/dendrite/roomserver/api" roomserverAPI "github.com/matrix-org/dendrite/roomserver/api"
"github.com/matrix-org/gomatrixserverlib"
"github.com/matrix-org/util" "github.com/matrix-org/util"
) )
@ -58,7 +55,7 @@ type RoomserverCanonicalAliasAPI struct {
} }
// SetRoomCanonicalAlias implements alias.RoomserverCanonicalAliasAPI // SetRoomCanonicalAlias implements alias.RoomserverCanonicalAliasAPI
func (r *RoomserverCanonicalAliasAPI) SetRoomAlias( func (r *RoomserverCanonicalAliasAPI) SetRoomCanonicalAlias(
ctx context.Context, ctx context.Context,
request *roomserverAPI.SetRoomCanonicalAliasRequest, request *roomserverAPI.SetRoomCanonicalAliasRequest,
response *roomserverAPI.SetRoomCanonicalAliasResponse, response *roomserverAPI.SetRoomCanonicalAliasResponse,
@ -66,7 +63,7 @@ func (r *RoomserverCanonicalAliasAPI) SetRoomAlias(
// SPEC: Room with `m.room.canonical_alias` with empty alias field should be // SPEC: Room with `m.room.canonical_alias` with empty alias field should be
// treated same as room without a canonical alias. // treated same as room without a canonical alias.
if request.CanonicalAlias == "" { if request.CanonicalAlias == "" {
return r.db.RemoveCanonicalAlias(ctx, request.RoomID) return r.DB.RemoveCanonicalAlias(ctx, request.RoomID)
} }
roomID, err := r.DB.GetRoomIDForAlias(ctx, request.CanonicalAlias) roomID, err := r.DB.GetRoomIDForAlias(ctx, request.CanonicalAlias)

View file

@ -22,6 +22,7 @@ import (
asQuery "github.com/matrix-org/dendrite/appservice/query" asQuery "github.com/matrix-org/dendrite/appservice/query"
"github.com/matrix-org/dendrite/common/basecomponent" "github.com/matrix-org/dendrite/common/basecomponent"
"github.com/matrix-org/dendrite/roomserver/alias" "github.com/matrix-org/dendrite/roomserver/alias"
"github.com/matrix-org/dendrite/roomserver/canonical_alias"
"github.com/matrix-org/dendrite/roomserver/input" "github.com/matrix-org/dendrite/roomserver/input"
"github.com/matrix-org/dendrite/roomserver/query" "github.com/matrix-org/dendrite/roomserver/query"
"github.com/matrix-org/dendrite/roomserver/storage" "github.com/matrix-org/dendrite/roomserver/storage"

View file

@ -111,8 +111,8 @@ func (s *roomCanonicalAliasStatements) selectCreatorIDFromCanonicalAlias(
} }
func (s *roomCanonicalAliasStatements) deleteRoomCanonicalAlias( func (s *roomCanonicalAliasStatements) deleteRoomCanonicalAlias(
ctx context.Context, canonical_alias string, ctx context.Context, roomID string,
) (err error) { ) (err error) {
_, err = s.deleteRoomCanonicalAliasStmt.ExecContext(ctx, canonical_alias) _, err = s.deleteRoomCanonicalAliasStmt.ExecContext(ctx, roomID)
return return
} }

View file

@ -29,7 +29,7 @@ type statements struct {
stateBlockStatements stateBlockStatements
previousEventStatements previousEventStatements
roomAliasesStatements roomAliasesStatements
roomCanonicalStatements roomCanonicalAliasStatements
inviteStatements inviteStatements
membershipStatements membershipStatements
transactionStatements transactionStatements

View file

@ -468,6 +468,33 @@ func (d *Database) RemoveRoomAlias(ctx context.Context, alias string) error {
return d.statements.deleteRoomAlias(ctx, alias) return d.statements.deleteRoomAlias(ctx, alias)
} }
// SetRoomCanonicalAlias implements alias.RoomserverCanonicalAliasAPIDB
func (d *Database) SetRoomCanonicalAlias(ctx context.Context, canonical_alias string, roomID string, creatorUserID string) error {
return d.statements.insertRoomCanonicalAlias(ctx, canonical_alias, roomID, creatorUserID)
}
// GetRoomIDForCanonicalAlias implements alias.RoomserverCanonicalAliasAPIDB
func (d *Database) GetRoomIDForCanonicalAlias(ctx context.Context, canonical_alias string) (string, error) {
return d.statements.selectRoomIDFromCanonicalAlias(ctx, canonical_alias)
}
// GetCanonicalAliasForRoomID implements alias.RoomserverCanonicalAliasAPIDB
func (d *Database) GetCanonicalAliasForRoomID(ctx context.Context, roomID string) (string, error) {
return d.statements.selectCanonicalAliasFromRoomID(ctx, roomID)
}
// GetCreatorIDForCanonicalAlias implements alias.RoomserverCanonicalAliasAPIDB
func (d *Database) GetCreatorIDForCanonicalAlias(
ctx context.Context, canonical_alias string,
) (string, error) {
return d.statements.selectCreatorIDFromCanonicalAlias(ctx, canonical_alias)
}
// RemoveRoomCanonicalAlias implements alias.RoomserverCanonicalAliasAPIDB
func (d *Database) RemoveRoomCanonicalAlias(ctx context.Context, canonical_alias string) error {
return d.statements.deleteRoomCanonicalAlias(ctx, canonical_alias)
}
// StateEntriesForTuples implements state.RoomStateDatabase // StateEntriesForTuples implements state.RoomStateDatabase
func (d *Database) StateEntriesForTuples( func (d *Database) StateEntriesForTuples(
ctx context.Context, ctx context.Context,