mirror of
				https://github.com/openimsdk/open-im-server.git
				synced 2025-11-04 11:22:10 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			67 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Bash
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			67 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			Bash
		
	
	
		
			Executable File
		
	
	
	
	
# Copyright © 2023 OpenIM. All rights reserved.
 | 
						|
#
 | 
						|
# Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
# you may not use this file except in compliance with the License.
 | 
						|
# You may obtain a copy of the License at
 | 
						|
#
 | 
						|
#     http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
#
 | 
						|
# Unless required by applicable law or agreed to in writing, software
 | 
						|
# distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
# See the License for the specific language governing permissions and
 | 
						|
# limitations under the License.
 | 
						|
mongosh <<EOF
 | 
						|
var maxRetries = 300;
 | 
						|
var connected = false;
 | 
						|
var rootUsername = '$MONGO_INITDB_ROOT_USERNAME';
 | 
						|
var rootPassword = '$MONGO_INITDB_ROOT_PASSWORD';
 | 
						|
var dbName = '$MONGO_INITDB_DATABASE';
 | 
						|
var openimUsername = '$MONGO_OPENIM_USERNAME';
 | 
						|
var openimPassword = '$MONGO_OPENIM_PASSWORD';
 | 
						|
 | 
						|
while (!connected && maxRetries > 0) {
 | 
						|
    try {
 | 
						|
        db = connect('mongodb://127.0.0.1:27017/admin');
 | 
						|
        var authResult = db.auth(rootUsername, rootPassword);
 | 
						|
        if (authResult) {
 | 
						|
            print('Authentication successful for root user: ' + rootUsername);
 | 
						|
            connected = true;
 | 
						|
        } else {
 | 
						|
            print('Authentication failed for root user: ' + rootUsername + ' with password: ' + rootPassword);
 | 
						|
            quit(1);
 | 
						|
        }
 | 
						|
    } catch (e) {
 | 
						|
        maxRetries--;
 | 
						|
        print('Connection failed, retrying... Remaining attempts: ' + maxRetries);
 | 
						|
        sleep(1000); // Sleep for 1 second
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
if (connected) {
 | 
						|
    db = db.getSiblingDB(dbName);
 | 
						|
    var createUserResult = db.createUser({
 | 
						|
        user: openimUsername,
 | 
						|
        pwd: openimPassword,
 | 
						|
        roles: [{
 | 
						|
            role: 'readWrite',
 | 
						|
            db: dbName
 | 
						|
        }]
 | 
						|
    });
 | 
						|
 | 
						|
    if (createUserResult.ok == 1) {
 | 
						|
        print('User creation successful. User: ' + openimUsername + ', Database: ' + dbName);
 | 
						|
    } else {
 | 
						|
        print('User creation failed for user: ' + openimUsername + ' in database: ' + dbName);
 | 
						|
        quit(1);
 | 
						|
    }
 | 
						|
} else {
 | 
						|
    print('Failed to connect to MongoDB after 300 retries.');
 | 
						|
    quit(1);
 | 
						|
}
 | 
						|
EOF
 | 
						|
 | 
						|
 | 
						|
 | 
						|
 |