Шрифт:
Интервал:
Закладка:
STDMETHODIMP_(ULONG) CoBook::AddRef()
{
return ++m_refCount;
}
STDMETHODIMP_(ULONG) CoBook::Release()
{
if (--m_refCount == 0)
{
delete this;
return 0;
}
else
return m_refCount;
}
STDMETHODIMP CoBook::Querylnterface(REFIID riid, void** pIFace)
{
if (riid == IID_IUnknown)
{
*pIFace = (IUnknown*)this;
}
else if (riid == IID_IPub)
}
*pIFace = (IPub*)this;
}
else if (riid == IID_IBook)
{
*pIFace = (IBook*)this;
}
else
{
*pIFace = NULL; return E_NOINTERFACE;
}
((IUnknown*)(*pIFace))->AddRef();
return S OK;
}
//IPub&IBook
STDMETHODIMP CoBook::SetTitle (BSTR bstrTitle)
{
SysReAllocString(&m_bstrTitle, bstrTitle);
return S_OK;
}
STDMETHODIMP CoBook::SetYear(int nYear)
{
m_nYear = nYear;
return S_OK;
}
STDMETHODIMP CoBook::SetAuthor(BSTR bstrAuthor)
{
SysReAllocString(&m_bstrAuthor, bstrAuthor);
return S_OK;
}
STDMETHODIMP CoBook::Getlnfo(BSTR *pbstrInfo)
{
char* pszTitle = NULL;
char* pszAuthor = NULL;
char* pszText = NULL;
int nAuthorLength, nTitleLength;
nAuthorLength = SysStringLen(m_bstrAuthor);
nTitleLength = SysStringLen(m_bstrTitle);
pszAuthor = (char*)malloc(2*nAuthorLength);
pszTitle = (char*)malloc(2*nTitleLength);
pszText = (char*)malloc(2*nAuthorLength + 2*nTitleLength + 50);
wcstombs(pszTitle, m_bstrTitle, 2*nTitleLength);
wcstombs(pszAuthor, m_bstrAuthor, 2*nAuthorLength);
sprintf(pszText,
"BooknnAuthor: %snTitle: %snYear: %d", pszAuthor, pszTitle, m_nYear);
*pbstrInfo = SysAllocStringLen(NULL, 2*strlen(pszText));
mbstowcs (*pbstrInfo, pszText, 2*strlen (pszText));
free(pszAuthor);
free(pszTitle);
free(pszText);
return S_OK;
}
В конструкторе CoBook () задается начальное нулевое значение для счетчика числа ссылок на данный объект и увеличивается на 1 глобальный счетчик объектов. Глобальный счетчик объектов используется при принятии решения о возможности удаления из памяти сервера, возможно содержащего не только класс CоBоок, но и какие-то другие классы. Кроме того, члены класса m_bstrTitie и m_bstrAuthor получают в качестве значения пустую строку. Префикс L перед строкой константой говорит о том, что в этой строке на один символ отводится два байта. Для ее преобразования к типу BSTR и размещения в памяти используется функция SysAllocString ().
В деструкторе CоBооk () при уничтожении объекта на 1 уменьшается счетчик объектов и освобождается память, выделенная ранее для заголовка публикации и имени автора. Для освобождения памяти, выделенной под BSTR-строку, используется функция SysFreeString ().
Реализация методов AddRef () и Release () тривиальна и полность соответствует предписанной семантике этих методов.
Метод Query inter face () обеспечивает переход между любыми двумя интерфейсами, реализованными в коклассе совоок. Заметим, что среди этих интерфейсов не только IBook, но и все интерфейсы, от которых прямо или косвенно IBook порожден — IPub и IUnknown. Если запрашиваемый интерфейс не найден, возвращается E_NOINTERFACE. В противном случае вызывается метод AddRef () интерфейса IUnknown, увеличивающий счетчик ссылок на данный объект. Вызвать функцию Release () при освобождении ссылки, полученной в результате вызова QueryInterface () обязан уже клиент.
Методы SetTitle (), SetYear () и SetAuthor () позволяют задать соответственно название, год издания и имя автора книги — нового объекта CоBооk. При передачи в качестве входного аргумента BSTR-строки происходит переразмещение в памяти соответствующей BSTR-строки — члена класса. Для этого используется функция SysReAllocstring ().
Метод GetInfо () возвращает информацию о книге в строке типа BSTR. Вся информация о книге хранится в экземпляре класса CоBооk в виде BSTR-cтрок и целого числа. В процессе формирования результирующей строки все BSTR-строки преобразуются в ANSI-строки. Для выполнения преобразования BSTR-строки в ANSI-строку применяется функция wcstombs (). Первый параметр задает буфер для ANSI-строки, второй — BSTR-строку, третий — размер буфера. Функция SysStringien () позволяет определить длину BSTR-строки в символах. Размер буфера для соответствующей ANSI-строки выбирается в два раза превосходящим длину BSTR-строки, т. к. некоторые Unicode-символы представляются двумя ANSI-символами. Далее в буфер pszText помещается вся выводимая информация, выделяется память под возвращаемую BSTR-строку *pbstrInfо и вызывается функция преобразования ANSI-строки pszText в BSTR-строку *pbstrInfо-mbstowcs().
Для выделения памяти под возвращаемую строку используется функция SysAlloSstringLen (). Первый параметр задает BSTR-строку, первые символы которой, в количестве равном второму параметру, заносятся в новую строку. Задав в качестве первого параметра null мы получаем неинициализированную BSTR-строку длины, заданной вторым параметром.
Последнее действие — освобождение памяти, выделенной под буферы.
Далее определяется и реализуется класс CоJournal, порожденный от интерфейса IJournal. Так как интерфейс IJournal был сам порожден от интерфейса IPub, а последний — от стандартного интерфейса IUnknown, то класс CоJournal должен реализовать чисто виртуальные методы всех этих интерфейсов. Данный класс мало отличается от класса CоBооk, в связи с чем комментарии минимальны.
//////////////////////////////////////////////////
// СоJournal.h: заголовочный файл для класс СоJournal.
//////////////////////////////////////////////////
#ifndef _CoJournal_
#define _CoJournal_
#include "IJournal.h"
#include "iid.h"
class CoJournal:
public IJournal
{
public:
CoJournal ();
virtual ~CoJournal();
// IUnknown
STDMETHODIMP Querylnterface(REFIID riid, void** pIFace);
STDMETHODIMP_(ULONG) AddRef();
STDMETHODIMP_(ULONG) Release();
// IPub
STDMETHOD(SetTitle)(BSTR bstrTitle);
STDMETHOD(SetYear)(int nYear);
STDMETHOD(Getlnfo)(BSTR* pbstrlnfo);
// IJournal
STDMETHOD(SetNumber)(int nNumber);
private:
ULONG m_refCount; // Счетчик ссылок
BSTR m_bstrTitle; // Название публикации
int m_nYear; // Год публикации
int m nNumber; // Номер журнала
};
#endif
И теперь реализация класса CоJournal
//////////////////////////////////////////////////
// СоJournal.cpp: реализация класса CoJournal.
//////////////////////////////////////////////////
#include <stdio.h>
#include "CoJournal.h"
#include <stdio.h>
#include "CoJournal.h"
extern ULONG g_objCount; // Счетчик числа объектов
//////////////////////////////////////////////////
// Конструктор и деструктор
//////////////////////////////////////////////////
CoJournal::CoJournal()
{
m_refCount = 0;
++g_obj Count;
m_bstrTitle = SysAllocString(L"");
}
CoJournal::~CoJournal()
{
-- g_objCount;
if(m_b strTitle)
SysFreeString(m_bstrTitle);
}
// IUnknown