一、批量任務開發思路

1、開發目標

2、開發思路

1、前端提交發起請求100臺機器，那我要等待5分鐘，

問題就在於，這100臺不是同時執行完的，有可能我有10臺執行完了，執行完了不能讓他等五分鐘，讓他執行完了立刻就返回

2、返回一個什麽呢？

task_id的值，這個值是唯一值

3、觸發任務和拿取結果關聯起來？

你相當於第一次提交是要觸發這個任務，觸發完這個任務後，不想讓頁面卡主，所以阿賈克斯必須要立即返回

你每隔10秒鐘去後臺取次結果，你再去後臺取就是發的新的請求，

觸發完任務後給它返回一個task_id的值，這個值是唯一值，我來後端取的時候拿上task_id的值

4、問題是現在結果存在哪？

1、寫消息隊列，一個任務一個隊列，可能不合適，消息放到隊列取走就消失了，因為我想要結果永久保存

2、一個寫數據庫，每個線程，把任務執行完了，自己寫到數據庫，我前端直接從數據庫裏那

二、批量任務存儲表結構

1、引子

1、往後臺提交一條任務，就往數據庫裏寫一條記錄，就把這個記錄的id返回

2、真正執行這條命令的主機從數據庫拿詳細的任務

3、一條大任務分出來很多子結果，所以你還要有一張表示存任務結果的

2、批量任務記錄表

class Task(models.Model):
    """批量任務記錄表"""
    user = models.ForeignKey("UserProfile")
    task_type_choices = ((0,‘cmd‘),(1,‘file_transfer‘))
    task_type = models.SmallIntegerField(choices=task_type_choices)
    content = models.TextField(verbose_name="任務內容")
    #hosts = models.ManyToManyField("BindHost")
    date  = models.DateTimeField(auto_now_add=True)

1. 那個用戶執行的要不要存？
2. 任務類型是批量執行命令還是批量分發文件
3. 執行任務的類型要不要存？
4. 這個任務有沒有執行完，是用過什麽來確定的？

3、子任務表

class TaskLogDetail(models.Model):
    task = models.ForeignKey("Task")
    bind_host = models.ForeignKey("BindHost")
    result = models.TextField()

    status_choices = ((0,‘success‘),(1,‘failed‘),(2,‘init‘))
    status = models.SmallIntegerField(choices=status_choices)

    start_date = models.DateTimeField(auto_now_add=True)
    end_date = models.DateTimeField(blank=True,null=True)


    def __str__(self):
        return "%s %s" %(self.bind_host,self.status)

1. 寫清楚自己關聯是哪個大的任務
2. 執行每個任務的時候，要不要寫清楚自己是那臺主機
3. 每個子任務執行是否成功是不是要存起來

4、admin註冊

admin.site.register(models.Task,TaskAdmin)
admin.site.register(models.TaskLogDetail,TaskLogDetailAdmin)

三、批量任務前端頁面開發

1、url

url(r‘^multitask/cmd/$‘, views.multitask_cmd,name="multitask_cmd"),

2、views

@login_required
def multitask_cmd(request):

    return render(request,"multitask_cmd.html")

@login_required
def multitask_result(request):
    task_id = request.GET.get(‘task_id‘)
    task_obj = models.Task.objects.get(id=task_id)
    task_log_results = list(task_obj.tasklogdetail_set.values(‘id‘, ‘result‘,‘status‘,‘start_date‘,‘end_date‘))

    return  HttpResponse(json.dumps(task_log_results,default=json_date_handler))

3、multitask_cmd.html

{% extends ‘index.html‘ %}


{% block page-title %}主機管理|批量命令{% endblock %}

{% block page-content %}
    {% csrf_token %}

<div class="row">

    {% include ‘multitask_host_list_component.html‘ %}

    <div class="col-lg-8">
        <div class="panel">
            <div class="panel-heading">
                <h3 class="panel-title">命令操作</h3>
            </div>
            <div class="panel-body">
                <textarea id="cmd_text" class="form-control"></textarea>
                <input type="button" id=‘post_task_btn‘  onclick="PostTask(this,‘cmd‘)" class="btn btn-success pull-right" value="執行命令">
            </div>
        </div>
        <div class="panel">
            <div class="panel-heading">
                <h3 class="panel-title">任務結果</h3>
            </div>
            <div class="panel-body">
                <ul id="task_result_container"></ul>
            </div>
        </div>
    </div>

</div>


{% include ‘multitask_js_component.html‘ %}

{% endblock %}

4、multitask_host_list_component.html

    <div class="col-lg-4">
        <div class="panel">
            <div class="panel-heading">
                <h3 class="panel-title">主機列表</h3>
            </div>
            <div class="panel-body">
                    <div class="list-group bord-no">
                        <a onclick="HostListDisplayToggle(this)" class="list-group-item " href="#">
                            <input type="checkbox" onclick="SelectGroup(this)">
                            未分組主機
                            <span class="badge badge-primary">{{ request.user.bind_hosts.count }}</span>
                        </a>
                        <ol class="hide">
                            {% for bind_host in request.user.bind_hosts.all %}
                                <li><input type="checkbox"  select_host="true" value="{{ bind_host.id }}">{{ bind_host.host.hostname }}({{ bind_host.host.ip_addr }})@{{ bind_host.remote_user.username}}</li>
                            {% endfor %}
                        </ol>



                        {% for host_group in request.user.host_groups.select_related %}

                            <a onclick="HostListDisplayToggle(this)" class="list-group-item " href="#">
                                <input type="checkbox" onclick="SelectGroup(this)">
                                {{ host_group.name }}
                                <span class="badge badge-primary">{{ host_group.bind_hosts.count }}</span>
                            </a>
                            <ol class="hide">
                                {% for bind_host in host_group.bind_hosts.all %}
                                    <li><input type="checkbox"  select_host="true" value="{{ bind_host.id }}">{{ bind_host.host.hostname }}({{ bind_host.host.ip_addr }})@{{ bind_host.remote_user.username}}</li>
                                {% endfor %}
                            </ol>


                        {% endfor %}
                    </div>
            </div>
        </div>

    </div>

四、前端測試截圖

1、基本框架截圖

2、全選反選功能截圖

3、命令操作功能截圖

4、整個框架完成截圖

5、控制臺批量執行命令截圖

主機管理+堡壘機系統開發：批量任務開發思路（十一）