friends.c

#include <string.h>
#include "server.h"
#include "users.h"
#include "server_configs.h"
#include "friends.h"
#include "mlog.h"

int get_friends_by_user_from_mysql(struct server *sv, struct user *usr)
{
	struct mysql_config *config = sv->mysql_config;
	MYSQL_RES *result = NULL;
	MYSQL_ROW row = NULL;
	int ret = 0, num_of_rows = 0, offset = 0;
	char query[100], *tmp = NULL;

	snprintf(query, sizeof(query),"select username, friend from "
		"friends where username = '%s'", usr->username);
	if(sv->dump) mlog("%s\n", query);
	ret = mysql_real_query_result(config, query, &result);
	if(ret < 0 || result == NULL) {
		mlog("mysql_query: %s\n", mysql_err_str(config));
		return ret;
	}

	num_of_rows = mysql_num_rows(result);
	usr->friends.buffer = malloc(num_of_rows * (SERVER_USERNAME_LENS + 1));
	if(usr->friends.buffer == NULL) {
		perror("malloc");
		ret = -1;
		goto out;
	}
	memset(usr->friends.buffer, 0, num_of_rows * SERVER_USERNAME_LENS); 
	usr->friends.friends_map = hashmap_new();
	if(usr->friends.friends_map == NULL) {
		printf("malloc friends_map failed.\n");
		ret = -1;
		goto out;
	}

	usr->friends.is_inited = 1;
	while(row = mysql_fetch_row(result)) {
		tmp = usr->friends.buffer + (SERVER_USERNAME_LENS + 1) * offset;
		strncpy(tmp, row[1], SERVER_USERNAME_LENS);
		if(sv->dump) mlog("username:%s, friend:%s\n", row[0], row[1]);

		ret = hashmap_put(usr->friends.friends_map, tmp, "1");
		if(ret != MAP_OK) {
			mlog("Error: The user %s friend %s is misake.\n", row[0], row[1]);
			free(usr->friends.buffer);
			usr->friends.buffer = NULL;
			ret = -1;
			goto out;
		}
		offset++;
	}
	usr->friends.num_of_friends = num_of_rows;

out:
	mysql_free_result(result);
	return ret;
}

int get_friends_by_user(struct server *sv, struct user *usr, struct friends **friends)
{
	int ret = 0;
	if(!usr->friends.is_inited) {
		ret = get_friends_by_user_from_mysql(sv, usr);
	}
	*friends = &usr->friends;
	return ret;
}

int insert_friend_to_mysql(struct server *sv, struct user *usr, struct user *friend)
{
	struct mysql_config *config = sv->mysql_config;
	int ret = 0;
	char query[SERVER_MAX_PACKETS];
	struct user *tmp = NULL;

	snprintf(query, sizeof(query),"insert into friends(username, friend)"\
		" values ('%s', '%s')", usr->username, friend->username);
	if(sv->dump) mlog("%s\n", query);
	ret = mysql_real_query_affected(config, query);
	if(ret < 0) {
		mlog("mysql_query: %s\n", mysql_err_str(config));
		return ret;
	}

	if(ret != 1) {
		mlog("affected rows is not one.\n");
		return -1;
	}

	return ret;
}

int make_friend(struct server *sv, struct user *usr, struct user *friend)
{
	int ret = 0;
	ret = insert_friend_to_mysql(sv, usr, friend);
	if(ret < 0) {
		goto out;
	}
	free_friends(usr);

out:
	return ret;
}

int free_friends(struct user *usr)
{
	usr->friends.num_of_friends = 0;
	usr->friends.is_inited = 0;
	if(usr->friends.buffer != NULL) {
		free(usr->friends.buffer);
		hashmap_free(usr->friends.friends_map);
	}
	return 0;
}

int friends_put(struct friends *friends)
{
	return 0;
}