#include "unpthread.h"

//线程信息
typedef struct {
    pthread_t    thread_tid;        //thread ID
    long        thread_count;    //connections handled    
} Thread;

//线程结构数组
Thread *tptr;                    //array of Thread structures; calloc‘ed

int listenfd, nthreads;
socklen_t addrlen;
pthread_mutex_t mlock = PTHREAD_MUTEX_INITIALIZER;

static void thread_make(int i);
static void sig_int(int signo)
{
    void pr_cpu_time(void);
    pr_cpu_time();
    exit(0);
}

int main(int argc, char *argv[])
{
    if (argc == 3) {
        listenfd = Tcp_listen(NULL, argv[1], &addrlen);
    } else if (argc == 4) {
        listenfd = Tcp_listen(argv[1], argv[2], &addrlen);
    } else {
        err_quit("Usage: a.out [ <host> ] <port#> <#threads>");
    }

    nthreads = atoi(argv[argc - 1]);
    tptr = (Thread*)Calloc(nthreads, sizeof(Thread));
    
    //预创建一定数目线程
    for (int i = 0; i < nthreads; ++i) {
        thread_make(i);
    }

    Signal(SIGINT, sig_int);

    //主线程什么也不做，创建的各个线程accept
    for (;;) {
        pause();
    }

    return 0;
}

static void *thread_main(void *arg)
{
    //第几个线程
    int tcnt = (int)arg;
    DPRINTF("thread %d starting\n", tcnt);

    void web_child(int);
    struct sockaddr *cliaddr = (struct sockaddr*)Malloc(addrlen);
    
    //accept，然后处理连接
    for (;;) {
        socklen_t clilen = addrlen;

        DPRINTF("thread %d lock\n", tcnt);
         //加锁
        Pthread_mutex_lock(&mlock);
        DPRINTF("thread %d start to accept\n", tcnt);
        int connfd = Accept(listenfd, cliaddr, &clilen);
        DPRINTF("thread %d accept a connection\n", tcnt);
        Pthread_mutex_unlock(&mlock);
        DPRINTF("thread %d unlock\n", tcnt);

        ++tptr[tcnt].thread_count;
        DPRINTF("thread %d accept %d connections\n", tcnt, tptr[tcnt].thread_count);
        web_child(connfd);
        Close(connfd);
    }

    return NULL;
}

static void thread_make(int i)
{
    Pthread_create(&tptr[i].thread_tid, NULL, &thread_main, (void*)i);
    return;
}