Browse Source

Update user.go

Dominic Renner 6 years ago
parent
commit
2d3de9740c
1 changed files with 29 additions and 13 deletions
  1. 29 13
      database/user.go

+ 29 - 13
database/user.go

@@ -34,19 +34,35 @@ type UserQuery struct {
 }
 
 func (uq *UserQuery) CreateTable(dbType string) error {
-	_, err := uq.db.Exec(`CREATE TABLE IF NOT EXISTS whatsapp_user (
-		mxid VARCHAR(255) PRIMARY KEY,
-		jid  VARCHAR(255)  UNIQUE,
-
-		management_room VARCHAR(255),
-
-		client_id    VARCHAR(255),
-		client_token VARCHAR(255),
-		server_token VARCHAR(255),
-		enc_key      bytea,
-		mac_key      bytea
-	)`)
-	return err
+	if strings.ToLower(dbType) == "postgres" {
+		_, err := uq.db.Exec(`CREATE TABLE IF NOT EXISTS whatsapp_user (
+			mxid VARCHAR(255) PRIMARY KEY,
+			jid  VARCHAR(255)  UNIQUE,
+
+			management_room VARCHAR(255),
+
+			client_id    VARCHAR(255),
+			client_token VARCHAR(255),
+			server_token VARCHAR(255),
+			enc_key      bytea,
+			mac_key      bytea
+		)`)
+		return err
+	} else {
+		_, err := uq.db.Exec(`CREATE TABLE IF NOT EXISTS whatsapp_user (
+			mxid VARCHAR(255) PRIMARY KEY,
+			jid  VARCHAR(255)  UNIQUE,
+
+			management_room VARCHAR(255),
+
+			client_id    VARCHAR(255),
+			client_token VARCHAR(255),
+			server_token VARCHAR(255),
+			enc_key      BLOB,
+			mac_key      BLOB
+		)`)
+		return err		
+	}
 }
 
 func (uq *UserQuery) New() *User {